graphql support non async resolvers

Author: jell-o-fishi

examples/graphql/client_graphql.py

@@ -25,13 +25,28 @@ async def main(server_port: int):
             transport=RSocketTransport(client),
         )
 
+        await query_schema(graphql)
+
         await greeting(graphql)
         await subscription(graphql)
 
         await set_message(graphql, "updated message")
         await verify_message(graphql, "updated message")
 
 
+async def query_schema(graphql: Client):
+    response = await graphql.execute_async(
+        gql("""{
+        __schema {
+        types {
+        name
+    }
+    }
+    }"""),
+        get_execution_result=True)
+    print(response.data)
+
+
 async def subscription(graphql: Client):
     async for response in graphql.subscribe_async(
             document=gql("""

rsocket/graphql/server_helper.py

@@ -5,6 +5,7 @@
 
 from graphql import execute, parse, GraphQLSchema, ExecutionResult
 from graphql import subscribe
+from graphql.pyutils import is_awaitable
 
 from rsocket.frame_helpers import str_to_bytes
 from rsocket.helpers import create_future
@@ -25,12 +26,11 @@ def graphql_handler(schema: GraphQLSchema, route: str,
     async def graphql_query(payload: Payload):
         document, params = parse_payload(payload, json_deserialize)
 
-        execution_result = await execute(
-            schema,
-            document,
-            variable_values=params.variables,
-            operation_name=params.operation_name
-        )
+        execution_result = execute(schema, document, variable_values=params.variables,
+                                   operation_name=params.operation_name)
+
+        if is_awaitable(execution_result):
+            execution_result = await execution_result
 
         rsocket_payload = graphql_to_rsocket_payload(execution_result, json_serialize)
 
@@ -41,11 +41,12 @@ async def graphql_subscription(payload: Payload):
         async def generator() -> AsyncGenerator[Tuple[Payload, bool], None]:
             document, params = parse_payload(payload, json_deserialize)
 
-            async for execution_result in await subscribe(
-                    schema,
-                    document,
-                    operation_name=params.operation_name
-            ):
+            subscription = subscribe(schema, document, operation_name=params.operation_name)
+
+            if is_awaitable(subscription):
+                subscription = await subscription
+
+            async for execution_result in subscription:
                 rsocket_payload = graphql_to_rsocket_payload(execution_result, json_serialize)
                 yield rsocket_payload, False