fix(module:message): fix message remove error when no container (#5123)

close #5121
NG-ZORRO · Apr 24, 2020 · 1eca795 · 1eca795
1 parent 82618c2
commit 1eca795
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/components/message/base.ts b/components/message/base.ts
@@ -24,10 +24,12 @@ export abstract class NzMNService {
   constructor(protected nzSingletonService: NzSingletonService, protected overlay: Overlay, private injector: Injector) {}
 
   remove(id?: string): void {
-    if (id) {
-      this.container.remove(id);
-    } else {
-      this.container.removeAll();
+    if (this.container) {
+      if (id) {
+        this.container.remove(id);
+      } else {
+        this.container.removeAll();
+      }
     }
   }