Initial subscription commit.

Implement changes in graphql/graphql-js@8a589f6

Author: jhgg

graphql/core/execution/base.py

@@ -133,8 +133,17 @@ def get_operation_root_type(schema, operation):
 
         return mutation_type
 
+    elif op == 'subscription':
+        subscription_type = schema.get_subscription_type()
+
+        if not subscription_type:
+            raise GraphQLError(
+                'Schema is not configured for subscriptions',
+                [operation]
+            )
+
     raise GraphQLError(
-        'Can only execute queries and mutations',
+        'Can only execute queries, mutations and subscriptions',
         [operation]
     )
 

graphql/core/type/introspection.py

@@ -39,6 +39,12 @@
             type=__Type,
             resolver=lambda schema, *_: schema.get_mutation_type(),
         )),
+        ('subscriptionType', GraphQLField(
+            description='If this server support subscription, the type '
+                        'that subscription operations will be rooted at.',
+            type=__Type,
+            resolver=lambda schema, *_: schema.get_subscription_type(),
+        )),
         ('directives', GraphQLField(
             description='A list of all directives supported by this server.',
             type=GraphQLNonNull(GraphQLList(GraphQLNonNull(__Directive))),

graphql/core/type/schema.py

@@ -24,16 +24,21 @@ class GraphQLSchema(object):
             mutation=MyAppMutationRootType
         )
     """
-    __slots__ = '_query', '_mutation', '_type_map', '_directives'
+    __slots__ = '_query', '_mutation', '_subscription', '_type_map', '_directives',
 
-    def __init__(self, query, mutation=None):
+    def __init__(self, query, mutation=None, subscription=None):
         assert isinstance(query, GraphQLObjectType), 'Schema query must be Object Type but got: {}.'.format(query)
         if mutation:
             assert isinstance(mutation, GraphQLObjectType), \
                 'Schema mutation must be Object Type but got: {}.'.format(mutation)
 
+        if subscription:
+            assert isinstance(subscription, GraphQLObjectType), \
+                'Schema subscription must be Object Type but got: {}.'.format(subscription)
+
         self._query = query
         self._mutation = mutation
+        self._subscription = subscription
         self._type_map = self._build_type_map()
         self._directives = None
 
@@ -48,6 +53,9 @@ def get_query_type(self):
     def get_mutation_type(self):
         return self._mutation
 
+    def get_subscription_type(self):
+        return self._subscription
+
     def get_type_map(self):
         return self._type_map
 
@@ -72,7 +80,8 @@ def get_directive(self, name):
 
     def _build_type_map(self):
         type_map = OrderedDict()
-        for type in (self.get_query_type(), self.get_mutation_type(), IntrospectionSchema):
+        types = (self.get_query_type(), self.get_mutation_type(), self.get_subscription_type(), IntrospectionSchema)
+        for type in types:
             type_map = type_map_reducer(type_map, type)
 
         return type_map

graphql/core/utils/build_ast_schema.py

@@ -45,7 +45,7 @@ def _get_inner_type_name(type_ast):
 _none = lambda *_: None
 
 
-def build_ast_schema(document, query_type_name, mutation_type_name=None):
+def build_ast_schema(document, query_type_name, mutation_type_name=None, subscription_type_name=None):
     assert isinstance(document, ast.Document), 'must pass in Document ast.'
     assert query_type_name, 'must pass in query type'
 
@@ -58,6 +58,9 @@ def build_ast_schema(document, query_type_name, mutation_type_name=None):
     if mutation_type_name and mutation_type_name not in ast_map:
         raise Exception('Specified mutation type {} not found in document.'.format(mutation_type_name))
 
+    if subscription_type_name and subscription_type_name not in ast_map:
+        raise Exception('Specified subscription type {} not found in document.'.format(subscription_type_name))
+
     inner_type_map = OrderedDict([
         ('String', GraphQLString),
         ('Int', GraphQLInt),
@@ -175,4 +178,7 @@ def make_schema_def(definition):
     if mutation_type_name:
         schema_kwargs['mutation'] = produce_type_def(ast_map[mutation_type_name])
 
+    if subscription_type_name:
+        schema_kwargs['subscription'] = produce_type_def(ast_map[subscription_type_name])
+
     return GraphQLSchema(**schema_kwargs)

graphql/core/utils/build_client_schema.py

@@ -204,5 +204,7 @@ def build_input_value_def_map(input_value_introspection, argument_type):
 
     query_type = get_type(schema_introspection['queryType'])
     mutation_type = get_type(schema_introspection['mutationType']) if schema_introspection.get('mutationType') else None
+    subscription_type = get_type(schema_introspection['subscriptionType']) if \
+        schema_introspection.get('subscriptionType') else None
 
-    return GraphQLSchema(query_type, mutation_type)
+    return GraphQLSchema(query=query_type, mutation=mutation_type, subscription=subscription_type)

graphql/core/utils/introspection_query.py

@@ -3,6 +3,7 @@
     __schema {
         queryType { name }
         mutationType { name }
+        subscriptionType { name }
         types {
             ...FullType
         }

tests/core_type/test_introspection.py

@@ -53,6 +53,7 @@ def test_executes_an_introspection_query():
                                      'onFragment': True,
                                      'onOperation': False}],
                      'mutationType': None,
+                     'subscriptionType': None,
                      'queryType': {'name': u'QueryRoot'},
                      'types': [{'description': None,
                                 'enumValues': None,
@@ -111,6 +112,14 @@ def test_executes_an_introspection_query():
                                                'type': {'kind': 'OBJECT',
                                                         'name': u'__Type',
                                                         'ofType': None}},
+                                              {'args': [],
+                                               'deprecationReason': None,
+                                               'description': u'If this server support subscription, the type that subscription operations will be rooted at.',
+                                               'isDeprecated': False,
+                                               'name': u'subscriptionType',
+                                               'type': {'kind': 'OBJECT',
+                                                        'name': u'__Type',
+                                                        'ofType': None}},
                                               {'args': [],
                                                'deprecationReason': None,
                                                'description': u'A list of all directives supported by this server.',
@@ -744,9 +753,14 @@ def test_exposes_descriptions_on_types_and_fields():
             },
             {
                 'name': 'mutationType',
-                'description': 'If this server supports mutation, the type that ' +
+                'description': 'If this server supports mutation, the type that '
                                'mutation operations will be rooted at.'
             },
+            {
+                'name': 'subscriptionType',
+                'description': 'If this server support subscription, the type '
+                               'that subscription operations will be rooted at.'
+            },
             {
                 'name': 'directives',
                 'description': 'A list of all directives supported by this server.'

tests/core_utils/test_schema_printer.py

@@ -499,6 +499,7 @@ def test_prints_introspection_schema():
   types: [__Type!]!
   queryType: __Type!
   mutationType: __Type
+  subscriptionType: __Type
   directives: [__Directive!]!
 }