Django - 如何检查服务器是在 ASGI 还是在 WSGI 模式下运行？

Question

我们在 WSGI 模式下运行相同的 django 项目来处理 HTTP 请求，并在 ASGI 模式下处理 WebSockets。对于 WSGI 模式，我们使用 gunicorn3 服务器：

gunicorn3 --pythonpath . -b 0.0.0.0:8000 chat_bot.wsgi:application

对于 ASGI 模式，我们使用 daphne 服务器：

daphne --root-path . -b 0.0.0.0 -p 8001 chat_bot.asgi:application

如何以编程方式检测当前正在运行的模式是 GreenUnicorn+WSGI 还是 Daphne+ASGI？

Answer 1

一种可能性：

在您的 wsgi.py 文件中，您可以将环境变量设置为您不会在其他任何地方设置的值：

os.environ.setdefault('SERVER_GATEWAY_INTERFACE', 'Web')

然后在asgi.py 内部将其设置为不同的变量：

os.environ.setdefault('SERVER_GATEWAY_INTERFACE', 'Asynchronous')

然后在代码的其他部分，只需检查环境变量：

if os.environ.get('SERVER_GATEWAY_INTERFACE') == 'Web':
    # WSGI, do something
else:
    # ASGI, do something else