123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184 |
- 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); }
- import { SYMBOL_ASYNC_ITERATOR } from "../polyfills/symbols.mjs";
- import inspect from "../jsutils/inspect.mjs";
- import { addPath, pathToArray } from "../jsutils/Path.mjs";
- import { GraphQLError } from "../error/GraphQLError.mjs";
- import { locatedError } from "../error/locatedError.mjs";
- import { assertValidExecutionArguments, buildExecutionContext, buildResolveInfo, collectFields, execute, getFieldDef, resolveFieldValueOrError } from "../execution/execute.mjs";
- import { getOperationRootType } from "../utilities/getOperationRootType.mjs";
- import mapAsyncIterator from "./mapAsyncIterator.mjs";
- export 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) {
- 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 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) ? mapAsyncIterator(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.
- */
- export 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.
- 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 = 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 = getOperationRootType(schema, exeContext.operation);
- var fields = 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 = getFieldDef(schema, type, fieldName);
- if (!fieldDef) {
- throw new 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 = addPath(undefined, responseName, type.name);
- var info = 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 = 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: [locatedError(eventStream, fieldNodes, 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(inspect(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 ? 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[SYMBOL_ASYNC_ITERATOR] === 'function';
- }
|