fix: memory leak issues

antvis · Nov 3, 2023 · 322451b · 322451b
1 parent b92a23e
commit 322451b
Show file tree

Hide file tree

Showing 8 changed files with 56 additions and 13 deletions.
diff --git a/packages/x6-common/src/common/basecoat.ts b/packages/x6-common/src/common/basecoat.ts
@@ -3,7 +3,15 @@ import { EventArgs } from '../event/types'
 import { ObjectExt } from '../object'
 import { Disposable } from './disposable'
 
-export class Basecoat<A extends EventArgs = any> extends Events<A> {}
+export class Basecoat<A extends EventArgs = any>
+  extends Events<A>
+  implements Disposable
+{
+  @Disposable.dispose()
+  dispose() {
+    this.off()
+  }
+}
 
 export interface Basecoat extends Disposable {}
 

diff --git a/packages/x6/src/graph/graph.ts b/packages/x6/src/graph/graph.ts
@@ -1262,6 +1262,7 @@ export class Graph extends Basecoat<EventArgs> {
     const aboutToChangePlugins = this.getPlugins(postPlugins)
     aboutToChangePlugins?.forEach((plugin) => {
       plugin.dispose()
+      this.installedPlugins.delete(plugin)
     })
     return this
   }
@@ -1273,11 +1274,9 @@ export class Graph extends Basecoat<EventArgs> {
   @Basecoat.dispose()
   dispose(clean = true) {
     if (clean) {
-      this.clearCells()
+      this.model.dispose()
     }
 
-    this.off()
-
     this.css.dispose()
     this.defs.dispose()
     this.grid.dispose()

diff --git a/packages/x6/src/model/collection.ts b/packages/x6/src/model/collection.ts
@@ -313,6 +313,11 @@ export class Collection extends Basecoat<Collection.EventArgs> {
     this.cells = []
     this.map = {}
   }
+
+  @Collection.dispose()
+  dispose() {
+    this.reset([])
+  }
 }
 
 export namespace Collection {

diff --git a/packages/x6/src/model/model.ts b/packages/x6/src/model/model.ts
@@ -1323,6 +1323,11 @@ export class Model extends Basecoat<Model.EventArgs> {
   }
 
   // #endregion
+
+  @Model.dispose()
+  dispose() {
+    this.collection.dispose()
+  }
 }
 
 export namespace Model {

diff --git a/packages/x6/src/renderer/scheduler.ts b/packages/x6/src/renderer/scheduler.ts
@@ -524,6 +524,11 @@ export class Scheduler extends Disposable {
   @Disposable.dispose()
   dispose() {
     this.stopListening()
+    // clear views
+    Object.keys(this.views).forEach((id) => {
+      this.views[id].view.dispose()
+    })
+    this.views = {}
   }
 }
 export namespace Scheduler {

diff --git a/packages/x6/src/shape/html.ts b/packages/x6/src/shape/html.ts
@@ -17,15 +17,17 @@ export namespace HTML {
   export class View extends NodeView<HTML> {
     protected init() {
       super.init()
-      this.cell.on('change:*', ({ key }) => {
-        const content = shapeMaps[this.cell.shape]
-        if (content) {
-          const { effect } = content
-          if (!effect || effect.includes(key)) {
-            this.renderHTMLComponent()
-          }
+      this.cell.on('change:*', this.onCellChangeAny, this)
+    }
+
+    protected onCellChangeAny({ key }: Cell.EventArgs['change:*']) {
+      const content = shapeMaps[this.cell.shape]
+      if (content) {
+        const { effect } = content
+        if (!effect || effect.includes(key)) {
+          this.renderHTMLComponent()
         }
-      })
+      }
     }
 
     confirmUpdate(flag: number) {
@@ -58,6 +60,11 @@ export namespace HTML {
         }
       }
     }
+
+    @View.dispose()
+    dispose() {
+      this.cell.off('change:*', this.onCellChangeAny, this)
+    }
   }
 
   export namespace View {

diff --git a/packages/x6/src/view/cell.ts b/packages/x6/src/view/cell.ts
@@ -251,7 +251,11 @@ export class CellView<
   }
 
   protected setup() {
-    this.cell.on('changed', ({ options }) => this.onAttrsChange(options))
+    this.cell.on('changed', this.onCellChanged, this)
+  }
+
+  protected onCellChanged({ options }: Cell.EventArgs['changed']) {
+    this.onAttrsChange(options)
   }
 
   protected onAttrsChange(options: Cell.MutateOptions) {
@@ -739,6 +743,11 @@ export class CellView<
     view.onMouseEnter(e as Dom.MouseEnterEvent)
   }
 
+  @CellView.dispose()
+  dispose() {
+    this.cell.off('changed', this.onCellChanged, this)
+  }
+
   // #endregion
 }
 

diff --git a/packages/x6/src/view/view.ts b/packages/x6/src/view/view.ts
@@ -362,6 +362,11 @@ export abstract class View<A extends EventArgs = any> extends Basecoat<A> {
   normalizeEvent<T extends Dom.EventObject>(evt: T) {
     return View.normalizeEvent(evt)
   }
+
+  @View.dispose()
+  dispose() {
+    this.remove()
+  }
 }
 
 export namespace View {