fix(cdk): DirectiveStylesService fix ssr memory leak (#6288)

taiga-family · Dec 22, 2023 · 799e7d6 · 799e7d6
1 parent e8ac4ea
commit 799e7d6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/projects/cdk/services/directive-styles.service.ts b/projects/cdk/services/directive-styles.service.ts
@@ -1,9 +1,11 @@
 import {
     ComponentFactoryResolver,
+    ComponentRef,
     Inject,
     Injectable,
     INJECTOR,
     Injector,
+    OnDestroy,
     Type,
 } from '@angular/core';
 
@@ -13,8 +15,8 @@ import {
 @Injectable({
     providedIn: 'root',
 })
-export class TuiDirectiveStylesService {
-    private readonly map = new Map<Type<unknown>, unknown>();
+export class TuiDirectiveStylesService implements OnDestroy {
+    private readonly map = new Map<Type<unknown>, ComponentRef<unknown>>();
 
     constructor(
         @Inject(ComponentFactoryResolver)
@@ -30,4 +32,8 @@ export class TuiDirectiveStylesService {
             );
         }
     }
+
+    ngOnDestroy(): void {
+        this.map.forEach(value => value.destroy());
+    }
 }