fix(proxy-state-tree): fix memory leak

Author: christianalfoni

packages/node_modules/proxy-state-tree/src/MutationTree.ts

@@ -79,6 +79,13 @@ export class MutationTree<T extends object> implements IMutationTree<T> {
   dispose() {
     this.isTracking = false
     this.mutationCallbacks.length = 0
+
+    if (this.proxifier !== this.master.proxifier) {
+      this.proxifier.dispose()
+    }
+
+    this.proxifier = this.master.proxifier
+
     return this
   }
 }

packages/node_modules/proxy-state-tree/src/TrackStateTree.ts

@@ -85,6 +85,12 @@ export class TrackStateTree<T extends object> implements ITrackStateTree<T> {
     this.pathDependencies.clear()
     this.callback = null
 
+    if (this.proxifier !== this.master.proxifier) {
+      this.proxifier.dispose()
+    }
+
+    this.proxifier = this.master.proxifier
+
     if (this.master.currentTree === this) {
       this.master.currentTree = null
     }

packages/node_modules/proxy-state-tree/src/index.ts

@@ -124,12 +124,6 @@ export class ProxyStateTree<T extends object> implements IProxyStateTree<T> {
       this.cache.mutationTree.push(tree.dispose())
     } else if (tree instanceof TrackStateTree) {
       this.cache.trackStateTree.push(tree.dispose())
-
-      if (tree.proxifier !== this.proxifier) {
-        tree.proxifier.dispose()
-      }
-
-      tree.proxifier = this.proxifier
     }
   }
   onMutation(callback: IMutationCallback) {