UNPKG

@graphql-tools/executor-graphql-ws

Version:

A set of utils for faster development of GraphQL tools

43 lines (42 loc) 1.6 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.buildGraphQLWSExecutor = void 0; const tslib_1 = require("tslib"); const utils_1 = require("@graphql-tools/utils"); const graphql_1 = require("graphql"); const graphql_ws_1 = require("graphql-ws"); const isomorphic_ws_1 = tslib_1.__importDefault(require("isomorphic-ws")); function isClient(client) { return 'subscribe' in client; } function buildGraphQLWSExecutor(clientOptionsOrClient) { let graphqlWSClient; if (isClient(clientOptionsOrClient)) { graphqlWSClient = clientOptionsOrClient; } else { graphqlWSClient = (0, graphql_ws_1.createClient)({ webSocketImpl: isomorphic_ws_1.default, lazy: true, ...clientOptionsOrClient, }); if (clientOptionsOrClient.onClient) { clientOptionsOrClient.onClient(graphqlWSClient); } } return function GraphQLWSExecutor(executionRequest) { const { document, variables, operationName, extensions, operationType = (0, utils_1.getOperationASTFromRequest)(executionRequest).operation, } = executionRequest; const query = (0, graphql_1.print)(document); const iterableIterator = graphqlWSClient.iterate({ query, variables, operationName, extensions, }); if (operationType === 'subscription') { return iterableIterator; } return iterableIterator.next().then(({ value }) => value); }; } exports.buildGraphQLWSExecutor = buildGraphQLWSExecutor;