graphql-modules 2.4.0 → 2.4.1-alpha-20241223144441-674927c3a29a8c57bbdad7eb31576cb88178f815

package/application/context.d.ts

@@ -3,7 +3,10 @@ import { ResolvedProvider } from '../di/resolution';
 import type { InternalAppContext, ModulesMap } from './application';
 export type ExecutionContextBuilder<TContext extends {
     [key: string]: any;
-} = {}> = (context: TContext) => {
+} = {}> = (context: TContext) => ExecutionContextEnv & {
+    runWithContext<TReturn = any>(cb: (env: ExecutionContextEnv) => TReturn): TReturn;
+};
+export type ExecutionContextEnv = {
     context: InternalAppContext;
     ɵdestroy(): void;
     ɵinjector: Injector;

package/index.js

@@ -4,6 +4,7 @@ Object.defineProperty(exports, '__esModule', { value: true });
 
 const schema = require('@graphql-tools/schema');
 const graphql = require('graphql');
+const async_hooks = require('async_hooks');
 const wrap = require('@graphql-tools/wrap');
 const ramda = require('ramda');
 
@@ -854,6 +855,7 @@ function duplicatedGlobalTokenError(provider, modules) {
  */
 const CONTEXT = new InjectionToken('context');
 
+const alc = new async_hooks.AsyncLocalStorage();
 function createContextBuilder({ appInjector, modulesMap, appLevelOperationProviders, singletonGlobalProvidersMap, operationGlobalProvidersMap, }) {
     // This is very critical. It creates an execution context.
     // It has to run on every operation.
@@ -882,11 +884,14 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             },
         });
         appInjector.setExecutionContextGetter(function executionContextGetter() {
-            return appContext;
+            var _a;
+            return ((_a = alc.getStore()) === null || _a === void 0 ? void 0 : _a.getApplicationContext()) || appContext;
         });
         function createModuleExecutionContextGetter(moduleId) {
             return function moduleExecutionContextGetter() {
-                return getModuleContext(moduleId, context);
+                var _a;
+                return (((_a = alc.getStore()) === null || _a === void 0 ? void 0 : _a.getModuleContext(moduleId)) ||
+                    getModuleContext(moduleId, context));
             };
         }
         modulesMap.forEach((mod, moduleId) => {
@@ -956,7 +961,7 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
                 return getModuleContext(moduleId, sharedContext).injector;
             },
         });
-        return {
+        const env = {
             ɵdestroy: once(() => {
                 providersToDestroy.forEach(([injector, keyId]) => {
                     // If provider was instantiated
@@ -970,6 +975,19 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             ɵinjector: operationAppInjector,
             context: sharedContext,
         };
+        return {
+            ...env,
+            runWithContext(cb) {
+                return alc.run({
+                    getApplicationContext() {
+                        return appContext;
+                    },
+                    getModuleContext(moduleId) {
+                        return getModuleContext(moduleId, context);
+                    },
+                }, () => cb(env));
+            },
+        };
     };
     return contextBuilder;
 }
@@ -979,31 +997,34 @@ function executionCreator({ contextBuilder, }) {
         // Custom or original execute function
         const executeFn = (options === null || options === void 0 ? void 0 : options.execute) || graphql.execute;
         return (argsOrSchema, document, rootValue, contextValue, variableValues, operationName, fieldResolver, typeResolver) => {
-            var _a;
-            // Create an execution context
-            const { context, ɵdestroy: destroy } = (_a = options === null || options === void 0 ? void 0 : options.controller) !== null && _a !== void 0 ? _a : contextBuilder(isNotSchema(argsOrSchema)
+            function perform({ context, ɵdestroy: destroy, }) {
+                const executionArgs = isNotSchema(argsOrSchema)
+                    ? {
+                        ...argsOrSchema,
+                        contextValue: context,
+                    }
+                    : {
+                        schema: argsOrSchema,
+                        document: document,
+                        rootValue,
+                        contextValue: context,
+                        variableValues,
+                        operationName,
+                        fieldResolver,
+                        typeResolver,
+                    };
+                // It's important to wrap the executeFn within a promise
+                // so we can easily control the end of execution (with finally)
+                return Promise.resolve()
+                    .then(() => executeFn(executionArgs))
+                    .finally(destroy);
+            }
+            if (options === null || options === void 0 ? void 0 : options.controller) {
+                return perform(options.controller);
+            }
+            return contextBuilder(isNotSchema(argsOrSchema)
                 ? argsOrSchema.contextValue
-                : contextValue);
-            const executionArgs = isNotSchema(argsOrSchema)
-                ? {
-                    ...argsOrSchema,
-                    contextValue: context,
-                }
-                : {
-                    schema: argsOrSchema,
-                    document: document,
-                    rootValue,
-                    contextValue: context,
-                    variableValues,
-                    operationName,
-                    fieldResolver,
-                    typeResolver,
-                };
-            // It's important to wrap the executeFn within a promise
-            // so we can easily control the end of execution (with finally)
-            return Promise.resolve()
-                .then(() => executeFn(executionArgs))
-                .finally(destroy);
+                : contextValue).runWithContext(perform);
         };
     };
     return createExecution;
@@ -1014,43 +1035,46 @@ function subscriptionCreator({ contextBuilder, }) {
         // Custom or original subscribe function
         const subscribeFn = (options === null || options === void 0 ? void 0 : options.subscribe) || graphql.subscribe;
         return (argsOrSchema, document, rootValue, contextValue, variableValues, operationName, fieldResolver, subscribeFieldResolver) => {
-            var _a;
-            // Create an subscription context
-            const { context, ɵdestroy: destroy } = (_a = options === null || options === void 0 ? void 0 : options.controller) !== null && _a !== void 0 ? _a : contextBuilder(isNotSchema(argsOrSchema)
+            function perform({ context, ɵdestroy: destroy, }) {
+                const subscriptionArgs = isNotSchema(argsOrSchema)
+                    ? {
+                        ...argsOrSchema,
+                        contextValue: context,
+                    }
+                    : {
+                        schema: argsOrSchema,
+                        document: document,
+                        rootValue,
+                        contextValue: context,
+                        variableValues,
+                        operationName,
+                        fieldResolver,
+                        subscribeFieldResolver,
+                    };
+                let isIterable = false;
+                // It's important to wrap the subscribeFn within a promise
+                // so we can easily control the end of subscription (with finally)
+                return Promise.resolve()
+                    .then(() => subscribeFn(subscriptionArgs))
+                    .then((sub) => {
+                    if (isAsyncIterable(sub)) {
+                        isIterable = true;
+                        return tapAsyncIterator(sub, destroy);
+                    }
+                    return sub;
+                })
+                    .finally(() => {
+                    if (!isIterable) {
+                        destroy();
+                    }
+                });
+            }
+            if (options === null || options === void 0 ? void 0 : options.controller) {
+                return perform(options.controller);
+            }
+            return contextBuilder(isNotSchema(argsOrSchema)
                 ? argsOrSchema.contextValue
-                : contextValue);
-            const subscriptionArgs = isNotSchema(argsOrSchema)
-                ? {
-                    ...argsOrSchema,
-                    contextValue: context,
-                }
-                : {
-                    schema: argsOrSchema,
-                    document: document,
-                    rootValue,
-                    contextValue: context,
-                    variableValues,
-                    operationName,
-                    fieldResolver,
-                    subscribeFieldResolver,
-                };
-            let isIterable = false;
-            // It's important to wrap the subscribeFn within a promise
-            // so we can easily control the end of subscription (with finally)
-            return Promise.resolve()
-                .then(() => subscribeFn(subscriptionArgs))
-                .then((sub) => {
-                if (isAsyncIterable(sub)) {
-                    isIterable = true;
-                    return tapAsyncIterator(sub, destroy);
-                }
-                return sub;
-            })
-                .finally(() => {
-                if (!isIterable) {
-                    destroy();
-                }
-            });
+                : contextValue).runWithContext(perform);
         };
     };
     return createSubscription;
@@ -1075,10 +1099,9 @@ function apolloSchemaCreator({ createSubscription, contextBuilder, schema, }) {
     const createApolloSchema = () => {
         const sessions = {};
         const subscription = createSubscription();
-        function getSession(ctx) {
+        function getSession(ctx, { context, ɵdestroy: destroy }) {
             if (!ctx[CONTEXT_ID]) {
                 ctx[CONTEXT_ID] = uniqueId((id) => !sessions[id]);
-                const { context, ɵdestroy: destroy } = contextBuilder(ctx);
                 sessions[ctx[CONTEXT_ID]] = {
                     count: 0,
                     session: {
@@ -1110,20 +1133,22 @@ function apolloSchemaCreator({ createSubscription, contextBuilder, schema, }) {
                         operationName: input.operationName,
                     });
                 }
-                // Create an execution context
-                const { context, destroy } = getSession(input.context);
-                // It's important to wrap the executeFn within a promise
-                // so we can easily control the end of execution (with finally)
-                return Promise.resolve()
-                    .then(() => graphql.execute({
-                    schema,
-                    document: input.document,
-                    contextValue: context,
-                    variableValues: input.variables,
-                    rootValue: input.rootValue,
-                    operationName: input.operationName,
-                }))
-                    .finally(destroy);
+                // Create an execution context and run within it
+                return contextBuilder(input.context).runWithContext((env) => {
+                    const { context, destroy } = getSession(input.context, env);
+                    // It's important to wrap the executeFn within a promise
+                    // so we can easily control the end of execution (with finally)
+                    return Promise.resolve()
+                        .then(() => graphql.execute({
+                        schema,
+                        document: input.document,
+                        contextValue: context,
+                        variableValues: input.variables,
+                        rootValue: input.rootValue,
+                        operationName: input.operationName,
+                    }))
+                        .finally(destroy);
+                });
             },
         });
     };

package/index.mjs

@@ -1,5 +1,6 @@
 import { makeExecutableSchema } from '@graphql-tools/schema';
 import { GraphQLSchema, execute as execute$1, subscribe, visit, Kind, GraphQLScalarType, concatAST, defaultFieldResolver, parse } from 'graphql';
+import { AsyncLocalStorage } from 'async_hooks';
 import { wrapSchema } from '@graphql-tools/wrap';
 import { mergeDeepWith } from 'ramda';
 
@@ -851,6 +852,7 @@ function duplicatedGlobalTokenError(provider, modules) {
  */
 const CONTEXT = new InjectionToken('context');
 
+const alc = new AsyncLocalStorage();
 function createContextBuilder({ appInjector, modulesMap, appLevelOperationProviders, singletonGlobalProvidersMap, operationGlobalProvidersMap, }) {
     // This is very critical. It creates an execution context.
     // It has to run on every operation.
@@ -879,11 +881,14 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             },
         });
         appInjector.setExecutionContextGetter(function executionContextGetter() {
-            return appContext;
+            var _a;
+            return ((_a = alc.getStore()) === null || _a === void 0 ? void 0 : _a.getApplicationContext()) || appContext;
         });
         function createModuleExecutionContextGetter(moduleId) {
             return function moduleExecutionContextGetter() {
-                return getModuleContext(moduleId, context);
+                var _a;
+                return (((_a = alc.getStore()) === null || _a === void 0 ? void 0 : _a.getModuleContext(moduleId)) ||
+                    getModuleContext(moduleId, context));
             };
         }
         modulesMap.forEach((mod, moduleId) => {
@@ -953,7 +958,7 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
                 return getModuleContext(moduleId, sharedContext).injector;
             },
         });
-        return {
+        const env = {
             ɵdestroy: once(() => {
                 providersToDestroy.forEach(([injector, keyId]) => {
                     // If provider was instantiated
@@ -967,6 +972,19 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             ɵinjector: operationAppInjector,
             context: sharedContext,
         };
+        return {
+            ...env,
+            runWithContext(cb) {
+                return alc.run({
+                    getApplicationContext() {
+                        return appContext;
+                    },
+                    getModuleContext(moduleId) {
+                        return getModuleContext(moduleId, context);
+                    },
+                }, () => cb(env));
+            },
+        };
     };
     return contextBuilder;
 }
@@ -976,31 +994,34 @@ function executionCreator({ contextBuilder, }) {
         // Custom or original execute function
         const executeFn = (options === null || options === void 0 ? void 0 : options.execute) || execute$1;
         return (argsOrSchema, document, rootValue, contextValue, variableValues, operationName, fieldResolver, typeResolver) => {
-            var _a;
-            // Create an execution context
-            const { context, ɵdestroy: destroy } = (_a = options === null || options === void 0 ? void 0 : options.controller) !== null && _a !== void 0 ? _a : contextBuilder(isNotSchema(argsOrSchema)
+            function perform({ context, ɵdestroy: destroy, }) {
+                const executionArgs = isNotSchema(argsOrSchema)
+                    ? {
+                        ...argsOrSchema,
+                        contextValue: context,
+                    }
+                    : {
+                        schema: argsOrSchema,
+                        document: document,
+                        rootValue,
+                        contextValue: context,
+                        variableValues,
+                        operationName,
+                        fieldResolver,
+                        typeResolver,
+                    };
+                // It's important to wrap the executeFn within a promise
+                // so we can easily control the end of execution (with finally)
+                return Promise.resolve()
+                    .then(() => executeFn(executionArgs))
+                    .finally(destroy);
+            }
+            if (options === null || options === void 0 ? void 0 : options.controller) {
+                return perform(options.controller);
+            }
+            return contextBuilder(isNotSchema(argsOrSchema)
                 ? argsOrSchema.contextValue
-                : contextValue);
-            const executionArgs = isNotSchema(argsOrSchema)
-                ? {
-                    ...argsOrSchema,
-                    contextValue: context,
-                }
-                : {
-                    schema: argsOrSchema,
-                    document: document,
-                    rootValue,
-                    contextValue: context,
-                    variableValues,
-                    operationName,
-                    fieldResolver,
-                    typeResolver,
-                };
-            // It's important to wrap the executeFn within a promise
-            // so we can easily control the end of execution (with finally)
-            return Promise.resolve()
-                .then(() => executeFn(executionArgs))
-                .finally(destroy);
+                : contextValue).runWithContext(perform);
         };
     };
     return createExecution;
@@ -1011,43 +1032,46 @@ function subscriptionCreator({ contextBuilder, }) {
         // Custom or original subscribe function
         const subscribeFn = (options === null || options === void 0 ? void 0 : options.subscribe) || subscribe;
         return (argsOrSchema, document, rootValue, contextValue, variableValues, operationName, fieldResolver, subscribeFieldResolver) => {
-            var _a;
-            // Create an subscription context
-            const { context, ɵdestroy: destroy } = (_a = options === null || options === void 0 ? void 0 : options.controller) !== null && _a !== void 0 ? _a : contextBuilder(isNotSchema(argsOrSchema)
+            function perform({ context, ɵdestroy: destroy, }) {
+                const subscriptionArgs = isNotSchema(argsOrSchema)
+                    ? {
+                        ...argsOrSchema,
+                        contextValue: context,
+                    }
+                    : {
+                        schema: argsOrSchema,
+                        document: document,
+                        rootValue,
+                        contextValue: context,
+                        variableValues,
+                        operationName,
+                        fieldResolver,
+                        subscribeFieldResolver,
+                    };
+                let isIterable = false;
+                // It's important to wrap the subscribeFn within a promise
+                // so we can easily control the end of subscription (with finally)
+                return Promise.resolve()
+                    .then(() => subscribeFn(subscriptionArgs))
+                    .then((sub) => {
+                    if (isAsyncIterable(sub)) {
+                        isIterable = true;
+                        return tapAsyncIterator(sub, destroy);
+                    }
+                    return sub;
+                })
+                    .finally(() => {
+                    if (!isIterable) {
+                        destroy();
+                    }
+                });
+            }
+            if (options === null || options === void 0 ? void 0 : options.controller) {
+                return perform(options.controller);
+            }
+            return contextBuilder(isNotSchema(argsOrSchema)
                 ? argsOrSchema.contextValue
-                : contextValue);
-            const subscriptionArgs = isNotSchema(argsOrSchema)
-                ? {
-                    ...argsOrSchema,
-                    contextValue: context,
-                }
-                : {
-                    schema: argsOrSchema,
-                    document: document,
-                    rootValue,
-                    contextValue: context,
-                    variableValues,
-                    operationName,
-                    fieldResolver,
-                    subscribeFieldResolver,
-                };
-            let isIterable = false;
-            // It's important to wrap the subscribeFn within a promise
-            // so we can easily control the end of subscription (with finally)
-            return Promise.resolve()
-                .then(() => subscribeFn(subscriptionArgs))
-                .then((sub) => {
-                if (isAsyncIterable(sub)) {
-                    isIterable = true;
-                    return tapAsyncIterator(sub, destroy);
-                }
-                return sub;
-            })
-                .finally(() => {
-                if (!isIterable) {
-                    destroy();
-                }
-            });
+                : contextValue).runWithContext(perform);
         };
     };
     return createSubscription;
@@ -1072,10 +1096,9 @@ function apolloSchemaCreator({ createSubscription, contextBuilder, schema, }) {
     const createApolloSchema = () => {
         const sessions = {};
         const subscription = createSubscription();
-        function getSession(ctx) {
+        function getSession(ctx, { context, ɵdestroy: destroy }) {
             if (!ctx[CONTEXT_ID]) {
                 ctx[CONTEXT_ID] = uniqueId((id) => !sessions[id]);
-                const { context, ɵdestroy: destroy } = contextBuilder(ctx);
                 sessions[ctx[CONTEXT_ID]] = {
                     count: 0,
                     session: {
@@ -1107,20 +1130,22 @@ function apolloSchemaCreator({ createSubscription, contextBuilder, schema, }) {
                         operationName: input.operationName,
                     });
                 }
-                // Create an execution context
-                const { context, destroy } = getSession(input.context);
-                // It's important to wrap the executeFn within a promise
-                // so we can easily control the end of execution (with finally)
-                return Promise.resolve()
-                    .then(() => execute$1({
-                    schema,
-                    document: input.document,
-                    contextValue: context,
-                    variableValues: input.variables,
-                    rootValue: input.rootValue,
-                    operationName: input.operationName,
-                }))
-                    .finally(destroy);
+                // Create an execution context and run within it
+                return contextBuilder(input.context).runWithContext((env) => {
+                    const { context, destroy } = getSession(input.context, env);
+                    // It's important to wrap the executeFn within a promise
+                    // so we can easily control the end of execution (with finally)
+                    return Promise.resolve()
+                        .then(() => execute$1({
+                        schema,
+                        document: input.document,
+                        contextValue: context,
+                        variableValues: input.variables,
+                        rootValue: input.rootValue,
+                        operationName: input.operationName,
+                    }))
+                        .finally(destroy);
+                });
             },
         });
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphql-modules",
-  "version": "2.4.0",
+  "version": "2.4.1-alpha-20241223144441-674927c3a29a8c57bbdad7eb31576cb88178f815",
   "description": "Create reusable, maintainable, testable and extendable GraphQL modules",
   "sideEffects": false,
   "peerDependencies": {