support request scope (#13)

* support request scope

* fix tabbing

* change the format back

* get the format right

* passing contextId and specify return type

Co-authored-by: Tara Zou <[email protected]>

Author: tarazou9

index.ts

@@ -6,7 +6,7 @@ import {
   InternalServerErrorException,
   NestInterceptor,
 } from '@nestjs/common';
-import { APP_INTERCEPTOR, ModuleRef } from '@nestjs/core';
+import { APP_INTERCEPTOR, ModuleRef, ContextIdFactory } from '@nestjs/core';
 import { GqlExecutionContext } from '@nestjs/graphql';
 import DataLoader from 'dataloader';
 import { Observable } from 'rxjs';
@@ -35,7 +35,6 @@ const NEST_LOADER_CONTEXT_KEY: string = 'NEST_LOADER_CONTEXT_KEY';
 @Injectable()
 export class DataLoaderInterceptor implements NestInterceptor {
   constructor(private readonly moduleRef: ModuleRef) {}
-
   /**
    * @inheritdoc
    */
@@ -44,14 +43,15 @@ export class DataLoaderInterceptor implements NestInterceptor {
     const ctx: any = graphqlExecutionContext.getContext();
 
     if (ctx[NEST_LOADER_CONTEXT_KEY] === undefined) {
-      ctx[NEST_LOADER_CONTEXT_KEY] = (type: string): NestDataLoader<any, any> => {
+      ctx[NEST_LOADER_CONTEXT_KEY] = async (type: string) : Promise<NestDataLoader<any, any>> => {
         if (ctx[type] === undefined) {
-          try {
-            ctx[type] = this.moduleRef
-              .get<NestDataLoader<any, any>>(type, { strict: false })
+          try {           
+            const contextId = ContextIdFactory.getByRequest(ctx.req);         
+            ctx[type] = (await this.moduleRef
+              .resolve<NestDataLoader<any, any>>(type, contextId, { strict: false }))
               .generateDataLoader();
           } catch (e) {
-            throw new InternalServerErrorException(`The loader ${type} is not provided`);
+            throw new InternalServerErrorException(`The loader ${type} is not provided` + e);
           }
         }
 
@@ -65,12 +65,12 @@ export class DataLoaderInterceptor implements NestInterceptor {
 /**
  * The decorator to be used within your graphql method.
  */
-export const Loader = createParamDecorator((data: string, [_, __, ctx]) => {
+export const Loader = createParamDecorator(async (data: string, [_, __, ctx]) => {
   if (ctx[NEST_LOADER_CONTEXT_KEY] === undefined) {
     throw new InternalServerErrorException(`
-            You should provide interceptor ${DataLoaderInterceptor.name} globaly with ${APP_INTERCEPTOR}
+            You should provide interceptor ${DataLoaderInterceptor.name} globally with ${APP_INTERCEPTOR}
           `);
   }
 
-  return ctx[NEST_LOADER_CONTEXT_KEY](data);
+  return await ctx[NEST_LOADER_CONTEXT_KEY](data);
 });