@graphql-tools/executor-graphql-ws
Version:
A set of utils for faster development of GraphQL tools
43 lines (42 loc) • 1.6 kB
JavaScript
;
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;