123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.subscribe = subscribe;
- exports.createSourceEventStream = createSourceEventStream;
- var _symbols = require("../polyfills/symbols");
- var _inspect = _interopRequireDefault(require("../jsutils/inspect"));
- var _Path = require("../jsutils/Path");
- var _GraphQLError = require("../error/GraphQLError");
- var _locatedError = require("../error/locatedError");
- var _execute = require("../execution/execute");
- var _getOperationRootType = require("../utilities/getOperationRootType");
- var _mapAsyncIterator = _interopRequireDefault(require("./mapAsyncIterator"));
- function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
- function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
- function subscribe(argsOrSchema, document, rootValue, contextValue, variableValues, operationName, fieldResolver, subscribeFieldResolver) {
- /* eslint-enable no-redeclare */
- // Extract arguments from object args if provided.
- return arguments.length === 1 ? subscribeImpl(argsOrSchema) : subscribeImpl({
- schema: argsOrSchema,
- document: document,
- rootValue: rootValue,
- contextValue: contextValue,
- variableValues: variableValues,
- operationName: operationName,
- fieldResolver: fieldResolver,
- subscribeFieldResolver: subscribeFieldResolver
- });
- }
- /**
- * This function checks if the error is a GraphQLError. If it is, report it as
- * an ExecutionResult, containing only errors and no data. Otherwise treat the
- * error as a system-class error and re-throw it.
- */
- function reportGraphQLError(error) {
- if (error instanceof _GraphQLError.GraphQLError) {
- return {
- errors: [error]
- };
- }
- throw error;
- }
- function subscribeImpl(args) {
- var schema = args.schema,
- document = args.document,
- rootValue = args.rootValue,
- contextValue = args.contextValue,
- variableValues = args.variableValues,
- operationName = args.operationName,
- fieldResolver = args.fieldResolver,
- subscribeFieldResolver = args.subscribeFieldResolver;
- var sourcePromise = createSourceEventStream(schema, document, rootValue, contextValue, variableValues, operationName, subscribeFieldResolver); // For each payload yielded from a subscription, map it over the normal
- // GraphQL `execute` function, with `payload` as the rootValue.
- // This implements the "MapSourceToResponseEvent" algorithm described in
- // the GraphQL specification. The `execute` function provides the
- // "ExecuteSubscriptionEvent" algorithm, as it is nearly identical to the
- // "ExecuteQuery" algorithm, for which `execute` is also used.
- var mapSourceToResponse = function mapSourceToResponse(payload) {
- return (0, _execute.execute)({
- schema: schema,
- document: document,
- rootValue: payload,
- contextValue: contextValue,
- variableValues: variableValues,
- operationName: operationName,
- fieldResolver: fieldResolver
- });
- }; // Resolve the Source Stream, then map every source value to a
- // ExecutionResult value as described above.
- return sourcePromise.then(function (resultOrStream) {
- return (// Note: Flow can't refine isAsyncIterable, so explicit casts are used.
- isAsyncIterable(resultOrStream) ? (0, _mapAsyncIterator.default)(resultOrStream, mapSourceToResponse, reportGraphQLError) : resultOrStream
- );
- });
- }
- /**
- * Implements the "CreateSourceEventStream" algorithm described in the
- * GraphQL specification, resolving the subscription source event stream.
- *
- * Returns a Promise which resolves to either an AsyncIterable (if successful)
- * or an ExecutionResult (error). The promise will be rejected if the schema or
- * other arguments to this function are invalid, or if the resolved event stream
- * is not an async iterable.
- *
- * If the client-provided arguments to this function do not result in a
- * compliant subscription, a GraphQL Response (ExecutionResult) with
- * descriptive errors and no data will be returned.
- *
- * If the the source stream could not be created due to faulty subscription
- * resolver logic or underlying systems, the promise will resolve to a single
- * ExecutionResult containing `errors` and no `data`.
- *
- * If the operation succeeded, the promise resolves to the AsyncIterable for the
- * event stream returned by the resolver.
- *
- * A Source Event Stream represents a sequence of events, each of which triggers
- * a GraphQL execution for that event.
- *
- * This may be useful when hosting the stateful subscription service in a
- * different process or machine than the stateless GraphQL execution engine,
- * or otherwise separating these two steps. For more on this, see the
- * "Supporting Subscriptions at Scale" information in the GraphQL specification.
- */
- function createSourceEventStream(schema, document, rootValue, contextValue, variableValues, operationName, fieldResolver) {
- // If arguments are missing or incorrectly typed, this is an internal
- // developer mistake which should throw an early error.
- (0, _execute.assertValidExecutionArguments)(schema, document, variableValues);
- try {
- var _fieldDef$subscribe;
- // If a valid context cannot be created due to incorrect arguments,
- // this will throw an error.
- var exeContext = (0, _execute.buildExecutionContext)(schema, document, rootValue, contextValue, variableValues, operationName, fieldResolver); // Return early errors if execution context failed.
- if (Array.isArray(exeContext)) {
- return Promise.resolve({
- errors: exeContext
- });
- }
- var type = (0, _getOperationRootType.getOperationRootType)(schema, exeContext.operation);
- var fields = (0, _execute.collectFields)(exeContext, type, exeContext.operation.selectionSet, Object.create(null), Object.create(null));
- var responseNames = Object.keys(fields);
- var responseName = responseNames[0];
- var fieldNodes = fields[responseName];
- var fieldNode = fieldNodes[0];
- var fieldName = fieldNode.name.value;
- var fieldDef = (0, _execute.getFieldDef)(schema, type, fieldName);
- if (!fieldDef) {
- throw new _GraphQLError.GraphQLError("The subscription field \"".concat(fieldName, "\" is not defined."), fieldNodes);
- } // Call the `subscribe()` resolver or the default resolver to produce an
- // AsyncIterable yielding raw payloads.
- var resolveFn = (_fieldDef$subscribe = fieldDef.subscribe) !== null && _fieldDef$subscribe !== void 0 ? _fieldDef$subscribe : exeContext.fieldResolver;
- var path = (0, _Path.addPath)(undefined, responseName, type.name);
- var info = (0, _execute.buildResolveInfo)(exeContext, fieldDef, fieldNodes, type, path); // resolveFieldValueOrError implements the "ResolveFieldEventStream"
- // algorithm from GraphQL specification. It differs from
- // "ResolveFieldValue" due to providing a different `resolveFn`.
- var result = (0, _execute.resolveFieldValueOrError)(exeContext, fieldDef, fieldNodes, resolveFn, rootValue, info); // Coerce to Promise for easier error handling and consistent return type.
- return Promise.resolve(result).then(function (eventStream) {
- // If eventStream is an Error, rethrow a located error.
- if (eventStream instanceof Error) {
- return {
- errors: [(0, _locatedError.locatedError)(eventStream, fieldNodes, (0, _Path.pathToArray)(path))]
- };
- } // Assert field returned an event stream, otherwise yield an error.
- if (isAsyncIterable(eventStream)) {
- // Note: isAsyncIterable above ensures this will be correct.
- return eventStream;
- }
- throw new Error('Subscription field must return Async Iterable. ' + "Received: ".concat((0, _inspect.default)(eventStream), "."));
- });
- } catch (error) {
- // As with reportGraphQLError above, if the error is a GraphQLError, report
- // it as an ExecutionResult; otherwise treat it as a system-class error and
- // re-throw it.
- return error instanceof _GraphQLError.GraphQLError ? Promise.resolve({
- errors: [error]
- }) : Promise.reject(error);
- }
- }
- /**
- * Returns true if the provided object implements the AsyncIterator protocol via
- * either implementing a `Symbol.asyncIterator` or `"@@asyncIterator"` method.
- */
- function isAsyncIterable(maybeAsyncIterable) {
- if (maybeAsyncIterable == null || _typeof(maybeAsyncIterable) !== 'object') {
- return false;
- }
- return typeof maybeAsyncIterable[_symbols.SYMBOL_ASYNC_ITERATOR] === 'function';
- }
|