From 731c4de2171e3361003903ca08425f1a276c0c51 Mon Sep 17 00:00:00 2001 From: Kah Keng Tay Date: Tue, 27 Oct 2020 15:21:04 -0700 Subject: [PATCH] Register operation with connection context for gevent subscription --- graphql_ws/gevent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/graphql_ws/gevent.py b/graphql_ws/gevent.py index aadbe64..3089f8c 100644 --- a/graphql_ws/gevent.py +++ b/graphql_ws/gevent.py @@ -73,13 +73,14 @@ def on_start(self, connection_context, op_id, params): connection_context.request_context, params) assert isinstance(execution_result, Observable), \ "A subscription must return an observable" - execution_result.subscribe(SubscriptionObserver( + disposable = execution_result.subscribe(SubscriptionObserver( connection_context, op_id, self.send_execution_result, self.send_error, self.on_close )) + connection_context.register_operation(op_id, disposable) except Exception as e: self.send_error(connection_context, op_id, str(e))