graphql-modules 3.1.0-alpha-20260116100501-7f63ca7bf295181555d86c57addbead6abe7cce8 → 3.1.0-alpha-20260116100600-27f3a402a486f5305eddcf90a40aed7f0c67071b

package/application/async-context.d.ts

@@ -0,0 +1,6 @@
+export interface AsyncContext {
+    getApplicationContext(): GraphQLModules.AppContext;
+    getModuleContext(moduleId: string): GraphQLModules.ModuleContext;
+}
+export declare function getAsyncContext(): AsyncContext | undefined;
+export declare function runWithAsyncContext<R, TArgs extends any[]>(asyncContext: AsyncContext, callback: (...args: TArgs) => R, ...args: TArgs): R;

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,9 +4,14 @@ Object.defineProperty(exports, '__esModule', { value: true });
 
 const schema = require('@graphql-tools/schema');
 const graphql = require('graphql');
+const module$1 = require('module');
 const wrap = require('@graphql-tools/wrap');
 const ramda = require('ramda');
 
+function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; }
+
+const module__default = /*#__PURE__*/_interopDefaultLegacy(module$1);
+
 const ERROR_ORIGINAL_ERROR = 'diOriginalError';
 function getOriginalError(error) {
     return error[ERROR_ORIGINAL_ERROR];
@@ -838,6 +843,23 @@ function duplicatedGlobalTokenError(provider, modules) {
     ].join(' '));
 }
 
+let alc;
+if (typeof process !== 'undefined') {
+    // probably nodejs runtime
+    const require = module__default["default"].createRequire('file:///' /** path is not relevant since we're only loading a builtin */);
+    const hooks = require('async_hooks');
+    alc = new hooks.AsyncLocalStorage();
+}
+function getAsyncContext() {
+    return alc === null || alc === void 0 ? void 0 : alc.getStore();
+}
+function runWithAsyncContext(asyncContext, callback, ...args) {
+    if (!alc) {
+        return callback(...args);
+    }
+    return alc.run(asyncContext, callback, ...args);
+}
+
 /**
  * @api
  * `CONTEXT` is an InjectionToken representing the provided `GraphQLModules.GlobalContext`
@@ -883,11 +905,14 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             },
         });
         appInjector.setExecutionContextGetter(function executionContextGetter() {
-            return appContext;
+            var _a;
+            return ((_a = getAsyncContext()) === null || _a === void 0 ? void 0 : _a.getApplicationContext()) || appContext;
         });
         function createModuleExecutionContextGetter(moduleId) {
             return function moduleExecutionContextGetter() {
-                return getModuleContext(moduleId, context);
+                var _a;
+                return (((_a = getAsyncContext()) === null || _a === void 0 ? void 0 : _a.getModuleContext(moduleId)) ||
+                    getModuleContext(moduleId, context));
             };
         }
         modulesMap.forEach((mod, moduleId) => {
@@ -957,7 +982,7 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
                 return getModuleContext(moduleId, sharedContext).injector;
             },
         });
-        return {
+        const env = {
             ɵdestroy: once(() => {
                 providersToDestroy.forEach(([injector, keyId]) => {
                     // If provider was instantiated
@@ -971,6 +996,19 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             ɵinjector: operationAppInjector,
             context: sharedContext,
         };
+        return {
+            ...env,
+            runWithContext(cb) {
+                return runWithAsyncContext({
+                    getApplicationContext() {
+                        return appContext;
+                    },
+                    getModuleContext(moduleId) {
+                        return getModuleContext(moduleId, context);
+                    },
+                }, cb, env);
+            },
+        };
     };
     return contextBuilder;
 }
@@ -980,31 +1018,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;
@@ -1015,43 +1056,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;
@@ -1076,10 +1120,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: {
@@ -1111,20 +1154,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, concatAST, defaultFieldResolver, GraphQLScalarType, parse } from 'graphql';
+import module from 'module';
 import { wrapSchema } from '@graphql-tools/wrap';
 import { mergeDeepWith } from 'ramda';
 
@@ -834,6 +835,23 @@ function duplicatedGlobalTokenError(provider, modules) {
     ].join(' '));
 }
 
+let alc;
+if (typeof process !== 'undefined') {
+    // probably nodejs runtime
+    const require = module.createRequire('file:///' /** path is not relevant since we're only loading a builtin */);
+    const hooks = require('async_hooks');
+    alc = new hooks.AsyncLocalStorage();
+}
+function getAsyncContext() {
+    return alc === null || alc === void 0 ? void 0 : alc.getStore();
+}
+function runWithAsyncContext(asyncContext, callback, ...args) {
+    if (!alc) {
+        return callback(...args);
+    }
+    return alc.run(asyncContext, callback, ...args);
+}
+
 /**
  * @api
  * `CONTEXT` is an InjectionToken representing the provided `GraphQLModules.GlobalContext`
@@ -879,11 +897,14 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             },
         });
         appInjector.setExecutionContextGetter(function executionContextGetter() {
-            return appContext;
+            var _a;
+            return ((_a = getAsyncContext()) === null || _a === void 0 ? void 0 : _a.getApplicationContext()) || appContext;
         });
         function createModuleExecutionContextGetter(moduleId) {
             return function moduleExecutionContextGetter() {
-                return getModuleContext(moduleId, context);
+                var _a;
+                return (((_a = getAsyncContext()) === null || _a === void 0 ? void 0 : _a.getModuleContext(moduleId)) ||
+                    getModuleContext(moduleId, context));
             };
         }
         modulesMap.forEach((mod, moduleId) => {
@@ -953,7 +974,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 +988,19 @@ function createContextBuilder({ appInjector, modulesMap, appLevelOperationProvid
             ɵinjector: operationAppInjector,
             context: sharedContext,
         };
+        return {
+            ...env,
+            runWithContext(cb) {
+                return runWithAsyncContext({
+                    getApplicationContext() {
+                        return appContext;
+                    },
+                    getModuleContext(moduleId) {
+                        return getModuleContext(moduleId, context);
+                    },
+                }, cb, env);
+            },
+        };
     };
     return contextBuilder;
 }
@@ -976,31 +1010,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 +1048,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 +1112,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 +1146,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": "3.1.0-alpha-20260116100501-7f63ca7bf295181555d86c57addbead6abe7cce8",
+  "version": "3.1.0-alpha-20260116100600-27f3a402a486f5305eddcf90a40aed7f0c67071b",
   "description": "Create reusable, maintainable, testable and extendable GraphQL modules",
   "sideEffects": false,
   "peerDependencies": {