add createPubSub method

moleculerjs · Aug 30, 2020 · 491c7e6 · 491c7e6
1 parent c463180
commit 491c7e6
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/service.js b/src/service.js
@@ -15,7 +15,7 @@ const GraphQL = require("graphql");
 const { PubSub, withFilter } = require("graphql-subscriptions");
 const hash = require("object-hash");
 
-module.exports = function (mixinOptions) {
+module.exports = function(mixinOptions) {
 	mixinOptions = _.defaultsDeep(mixinOptions, {
 		routeOptions: {
 			path: "/graphql",
@@ -594,6 +594,16 @@ module.exports = function (mixinOptions) {
 				return makeExecutableSchema(schemaDef);
 			},
 
+			/**
+			 * Create PubSub instance.
+			 */
+			createPubSub() {
+				return new PubSub();
+			},
+
+			/**
+			 * Prepare GraphQL schemas based on Moleculer services.
+			 */
 			prepareGraphQLSchema() {
 				// Schema is up-to-date
 				if (!this.shouldUpdateGraphqlSchema && this.graphqlHandler) {
@@ -606,7 +616,7 @@ module.exports = function (mixinOptions) {
 				);
 
 				try {
-					this.pubsub = new PubSub();
+					this.pubsub = this.createPubSub();
 					const services = this.broker.registry.getServiceList({ withActions: true });
 					const schema = this.generateGraphQLSchema(services);