fix: registry

exuanbo · Oct 15, 2024 · f849531 · f849531
1 parent 43b7949
commit f849531
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/src/registry.ts b/src/registry.ts
@@ -29,16 +29,28 @@ export class Registry {
   get<T>(token: InjectionToken<T>): Registration<T> | undefined {
     return (
       internals.get(token)
-      || this.map.get(token)
-      || this.parent?.map.get(token)
+      || this.getRecursive(token)
+    );
+  }
+
+  private getRecursive<T>(token: InjectionToken<T>): Registration<T> | undefined {
+    return (
+      this.map.get(token)
+      || this.parent?.getRecursive(token)
     );
   }
 
   has(token: InjectionToken): boolean {
     return Boolean(
       internals.has(token)
-      || this.map.has(token)
-      || this.parent?.map.has(token),
+      || this.hasRecursive(token),
+    );
+  }
+
+  private hasRecursive(token: InjectionToken): boolean | undefined {
+    return (
+      this.map.has(token)
+      || this.parent?.hasRecursive(token)
     );
   }