Setup ASGI config

See also: #17
snu-quiqcl · Nov 22, 2024 · 8fa55de · 8fa55de
1 parent 6074335
commit 8fa55de
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/wlm_server/wlm_server/asgi.py b/wlm_server/wlm_server/asgi.py
@@ -10,7 +10,19 @@
 import os
 
 from django.core.asgi import get_asgi_application
+from channels.auth import AuthMiddlewareStack
+from channels.routing import ProtocolTypeRouter, URLRouter
+from channels.security.websocket import AllowedHostsOriginValidator
 
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'wlm_server.settings')
 
-application = get_asgi_application()
+application = ProtocolTypeRouter(
+    {
+        'http': get_asgi_application(),
+        'websocket': AllowedHostsOriginValidator(
+            AuthMiddlewareStack(
+                URLRouter([])
+            )
+        ),
+    }
+)