fix: fallback DataLoaderOptions

TriPSs · Apr 25, 2023 · 0ff87bb · 0ff87bb
1 parent 3f5d8bf
commit 0ff87bb
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/packages/query-graphql/src/pipes/inject-data-loader-config.pipe.ts b/packages/query-graphql/src/pipes/inject-data-loader-config.pipe.ts
@@ -1,4 +1,5 @@
 import { Inject, Injectable, PipeTransform } from '@nestjs/common'
+import { ModuleRef } from '@nestjs/core'
 import { Options } from 'dataloader'
 
 export const dataLoaderOptionsToken = () => 'DATALOADER_OPTIONS'
@@ -7,12 +8,17 @@ export type DataLoaderOptions = Options<any, any, any>
 
 @Injectable()
 export class InjectDataLoaderConfigPipe implements PipeTransform {
-  // inject any dependency
+  private readonly options: DataLoaderOptions = {}
+
   constructor(
-    @Inject(dataLoaderOptionsToken())
-    private readonly options: DataLoaderOptions
+    @Inject(ModuleRef)
+    private moduleRef: ModuleRef
   ) {
-    //
+    try {
+      this.options = this.moduleRef.get(dataLoaderOptionsToken(), { strict: false })
+    } catch (error) {
+      //
+    }
   }
 
   transform() {