perf(proxy-state-tree): single proxifier in production

Author: christianalfoni

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

@@ -104,6 +104,11 @@ export class Proxifier {
 
     return null
   }
+  getMutationTree() {
+    return (
+      this.tree.master.mutationTree || (this.tree as ITrackMutationTree<any>)
+    )
+  }
   private createArrayProxy(value, path) {
     var proxifier = this
 
@@ -137,7 +142,8 @@ export class Proxifier {
             ) {
               proxifier.ensureMutationTrackingIsEnabled(nestedPath)
               return (...args) => {
-                const mutationTree = proxifier.tree as ITrackMutationTree<any>
+                const mutationTree = proxifier.getMutationTree()
+
                 mutationTree.addMutation({
                   method: String(prop),
                   path: path,
@@ -160,7 +166,7 @@ export class Proxifier {
             proxifier.ensureMutationTrackingIsEnabled(nestedPath)
             proxifier.ensureValueDosntExistInStateTreeElsewhere(value)
 
-            const mutationTree = proxifier.tree as ITrackMutationTree<any>
+            const mutationTree = proxifier.getMutationTree()
 
             mutationTree.addMutation({
               method: 'set',
@@ -235,7 +241,7 @@ export class Proxifier {
                 objectChangePath = path
               }
 
-              const mutationTree = proxifier.tree as ITrackMutationTree<any>
+              const mutationTree = proxifier.getMutationTree()
 
               mutationTree.addMutation(
                 {
@@ -264,7 +270,8 @@ export class Proxifier {
                 objectChangePath = path
               }
 
-              const mutationTree = proxifier.tree as ITrackMutationTree<any>
+              const mutationTree = proxifier.getMutationTree()
+
               mutationTree.addMutation(
                 {
                   method: 'unset',

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

@@ -14,10 +14,10 @@ export class TrackMutationTree<T extends object>
   state: T
   proxifier: IProxifier<T>
   isTracking: boolean = false
-  constructor(master: IProxyStateTree<T>) {
+  constructor(master: IProxyStateTree<T>, proxifier?: IProxifier<T>) {
     this.isTracking = true
     this.master = master
-    this.proxifier = new Proxifier(this)
+    this.proxifier = proxifier || new Proxifier(this)
     this.state = this.proxifier.proxify(master.sourceState, '')
   }
   addMutation(mutation: IMutation, objectChangePath: string) {

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

@@ -40,6 +40,7 @@ export class ProxyStateTree<T extends object> implements IProxyStateTree<T> {
   currentFlushId: number = 0
   currentTree: TTree
   previousTree: TTree
+  mutationTree: ITrackMutationTree<T>
   proxifier: IProxifier<T>
   master: ProxyStateTree<T>
   mutations: IMutation[] = []
@@ -83,8 +84,8 @@ export class ProxyStateTree<T extends object> implements IProxyStateTree<T> {
   }
   getMutationTree(): ITrackMutationTree<T> {
     if (IS_PRODUCTION) {
-      return ((this as any).__cachedProductionTree =
-        (this as any).__cachedProductionTree || new TrackMutationTree(this))
+      return (this.mutationTree =
+        this.mutationTree || new TrackMutationTree(this, this.proxifier))
     }
 
     const tree =

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

@@ -89,5 +89,6 @@ export interface IProxyStateTree<T extends object> {
   proxifier: IProxifier<T>
   currentTree: TTree
   previousTree: TTree
+  mutationTree: ITrackMutationTree<T>
   currentFlushId: number
 }