add: context in generateDataLoader

krislefeber · Feb 7, 2020 · 808f52b · krislefeber · Feb 7, 2020 · 808f52b
1 parent a2f4b96
commit 808f52b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/index.ts b/index.ts
@@ -18,8 +18,9 @@ import { Observable } from 'rxjs';
 export interface NestDataLoader<ID, Type> {
   /**
    * Should return a new instance of dataloader each time
+   * @params ctx: the execution context
    */
-  generateDataLoader(): DataLoader<ID, Type>;
+  generateDataLoader(ctx: any): DataLoader<ID, Type>;
 }
 
 /**
@@ -49,7 +50,7 @@ export class DataLoaderInterceptor implements NestInterceptor {
           try {
             ctx[type] = this.moduleRef
               .get<NestDataLoader<any, any>>(type, { strict: false })
-              .generateDataLoader();
+              .generateDataLoader(ctx);
           } catch (e) {
             throw new InternalServerErrorException(`The loader ${type} is not provided`);
           }