emeraldjava
diff --git a/‎src/main/java/graphql/schema/idl/SchemaGenerator.java‎
Lines changed: 6 additions & 0 deletions b/‎src/main/java/graphql/schema/idl/SchemaGenerator.java‎
Lines changed: 6 additions & 0 deletions
@@ -158,9 +158,11 @@ private GraphQLSchema makeExecutableSchemaImpl(BuildContext buildCtx) {
         @SuppressWarnings("OptionalGetWithoutIsPresent")
         OperationTypeDefinition queryOp = operationTypes.stream().filter(op -> "query".equals(op.getName())).findFirst().get();
         Optional<OperationTypeDefinition> mutationOp = operationTypes.stream().filter(op -> "mutation".equals(op.getName())).findFirst();
+        Optional<OperationTypeDefinition> subscriptionOp = operationTypes.stream().filter(op -> "subscription".equals(op.getName())).findFirst();
 
         GraphQLObjectType query = buildOperation(buildCtx, queryOp);
         GraphQLObjectType mutation;
+        GraphQLObjectType subscription;
 
         GraphQLSchema.Builder schemaBuilder = GraphQLSchema
                 .newSchema()
@@ -170,6 +172,10 @@ private GraphQLSchema makeExecutableSchemaImpl(BuildContext buildCtx) {
             mutation = buildOperation(buildCtx, mutationOp.get());
             schemaBuilder.mutation(mutation);
         }
+        if (subscriptionOp.isPresent()) {
+            subscription = buildOperation(buildCtx, subscriptionOp.get());
+            schemaBuilder.subscription(subscription);
+        }
         return schemaBuilder.build();
     }