diff --git a/client/coral-framework/services/subscriptions.js b/client/coral-framework/services/subscriptions.js index f514ea18b..818a6fb33 100644 --- a/client/coral-framework/services/subscriptions.js +++ b/client/coral-framework/services/subscriptions.js @@ -1,14 +1,16 @@ import {print} from 'graphql-tag/printer'; // quick way to add the subscribe and unsubscribe functions to the network interface -const addGraphQLSubscriptions = (networkInterface, wsClient) => Object.assign(networkInterface, { - subscribe: (request, handler) => wsClient.subscribe({ - query: print(request.query), - variables: request.variables, - }, handler), - unsubscribe: (id) => { - wsClient.unsubscribe(id); - }, -}); +const addGraphQLSubscriptions = (networkInterface, wsClient) => { + return Object.assign(networkInterface, { + subscribe: (request, handler) => wsClient.subscribe({ + query: print(request.query), + variables: request.variables, + }, handler), + unsubscribe: (id) => { + wsClient.unsubscribe(id); + }, + }); +}; export default addGraphQLSubscriptions;