feat: support unregister

exuanbo · Oct 16, 2024 · 4e4fbfe · 4e4fbfe
1 parent b5e0260
commit 4e4fbfe
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/container.ts b/src/container.ts
@@ -77,6 +77,11 @@ export class Container {
     return this;
   }
 
+  unregister<Value>(token: InjectionToken<Value>): this {
+    this.registry.delete(token);
+    return this;
+  }
+
   resolve<Values extends unknown[]>(...tokens: InjectionTokens<Values>): Values[number];
   resolve<Value>(...tokens: InjectionToken<Value>[]): Value {
     for (const token of tokens) {
@@ -130,7 +135,7 @@ export class Container {
   private resolveScopedInstance<T>(registration: Registration<T>, instantiate: () => T): T {
     const context = useInjectionContext();
 
-    if (!context || context.container != this) {
+    if (!context || context.container !== this) {
       return withInjectionContext({
         container: this,
         resolution: {

diff --git a/src/registry.ts b/src/registry.ts
@@ -26,6 +26,10 @@ export class Registry {
     this.map.clear();
   }
 
+  delete<T>(token: InjectionToken<T>): void {
+    this.map.delete(token);
+  }
+
   get<T>(token: InjectionToken<T>): Registration<T> | undefined {
     return (
       internals.get(token)
@@ -72,7 +76,7 @@ export class Registry {
     );
   }
 
-  set<T>(token: InjectionToken<T>, registration: Registration<T>): this {
+  set<T>(token: InjectionToken<T>, registration: Registration<T>): void {
     assert(!internals.has(token), ErrorMessage.ReservedToken, token.name);
     let registrations = this.map.get(token);
     if (!registrations) {
@@ -84,7 +88,6 @@ export class Registry {
         provider !== registration.provider)) {
       registrations.push(registration);
     }
-    return this;
   }
 
   values(): IterableIterator<Registration[]> {