upgrade-ws

main.py

@@ -21,9 +21,9 @@ from services.stat.viewed import ViewedStorage
 from services.zine.gittask import GitTask
 from settings import DEV_SERVER_PID_FILE_NAME, SENTRY_DSN
 # from sse.transport import GraphQLSSEHandler
-# from services.inbox.presence import on_connect, on_disconnect
-from services.inbox.sse import sse_messages
-
+from services.inbox.presence import on_connect, on_disconnect
+# from services.inbox.sse import sse_messages
+from ariadne.asgi.handlers import GraphQLTransportWSHandler
 
 import_module("resolvers")
 schema = make_executable_schema(load_schema_from_path("schema.graphql"), resolvers)  # type: ignore
@@ -66,7 +66,7 @@ async def shutdown():
 
 
 routes = [
-    Route("/messages", endpoint=sse_messages),
+    # Route("/messages", endpoint=sse_messages),
     Route("/oauth/{provider}", endpoint=oauth_login),
     Route("/oauth-authorize", endpoint=oauth_authorize),
     Route("/confirm/{token}", endpoint=confirm_email_handler)
@@ -82,10 +82,10 @@ app = Starlette(
 app.mount("/", GraphQL(
     schema,
     debug=True,
-    # websocket_handler=GraphQLTransportWSHandler(
-    #    on_connect=on_connect,
-    #    on_disconnect=on_disconnect
-    # )
+    websocket_handler=GraphQLTransportWSHandler(
+        on_connect=on_connect,
+        on_disconnect=on_disconnect
+    )
 ))
 
 dev_app = app = Starlette(
@@ -98,8 +98,8 @@ dev_app = app = Starlette(
 dev_app.mount("/", GraphQL(
     schema,
     debug=True,
-    # websocket_handler=GraphQLTransportWSHandler(
-    #    on_connect=on_connect,
-    #    on_disconnect=on_disconnect
-    # )
+    websocket_handler=GraphQLTransportWSHandler(
+        on_connect=on_connect,
+        on_disconnect=on_disconnect
+    )
 ))