From da3393a1262bffbd9f6a91ddb2bf84bf080276a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=8A=9B=E7=9A=93?= <lihao.ylh@alibaba-inc.com>
Date: Wed, 12 May 2021 14:05:04 +0800
Subject: [PATCH] =?UTF-8?q?refactor(perf):=20=E5=9C=A8=20iframe=20?=
 =?UTF-8?q?=E5=8D=B8=E8=BD=BD=E5=89=8D,=20=E6=8A=8A=E7=9B=B8=E5=85=B3?=
 =?UTF-8?q?=E5=AF=B9=E8=B1=A1=E9=83=BD=E9=94=80=E6=AF=81=E6=8E=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../react-simulator-renderer/src/index.ts     |  1 -
 .../react-simulator-renderer/src/renderer.ts  | 33 ++++++++++++++-----
 2 files changed, 24 insertions(+), 10 deletions(-)

diff --git a/packages/react-simulator-renderer/src/index.ts b/packages/react-simulator-renderer/src/index.ts
index b88a94c05..0be54766f 100644
--- a/packages/react-simulator-renderer/src/index.ts
+++ b/packages/react-simulator-renderer/src/index.ts
@@ -6,7 +6,6 @@ if (typeof window !== 'undefined') {
 
 window.addEventListener('beforeunload', () => {
   (window as any).LCSimulatorHost = null;
-  // @ts-ignore
   renderer.dispose?.();
   (window as any).SimulatorRenderer = null;
   (window as any).ReactDOM.unmountComponentAtNode(document.getElementById('app'));
diff --git a/packages/react-simulator-renderer/src/renderer.ts b/packages/react-simulator-renderer/src/renderer.ts
index ede0b9131..e36974b4e 100644
--- a/packages/react-simulator-renderer/src/renderer.ts
+++ b/packages/react-simulator-renderer/src/renderer.ts
@@ -2,7 +2,7 @@ import React, { createElement, ReactInstance } from 'react';
 import { render as reactRender } from 'react-dom';
 import { host } from './host';
 import SimulatorRendererView from './renderer-view';
-import { computed, obx } from '@recore/obx';
+import { computed, obx, untracked } from '@recore/obx';
 import { getClientRects } from './utils/get-client-rects';
 import { reactFindDOMNodes, FIBER_KEY } from './utils/react-find-dom-nodes';
 import {
@@ -37,13 +37,13 @@ export class DocumentInstance {
     return this._schema;
   }
 
-  private dispose?: () => void;
+  private disposeFunctions: Array<() => void> = [];
 
   constructor(readonly container: SimulatorRendererContainer, readonly document: DocumentModel) {
-    this.dispose = host.autorun(() => {
+    this.disposeFunctions.push(host.autorun(() => {
       // sync schema
       this._schema = document.export(1);
-    });
+    }));
   }
 
   @obx.ref private _components: any = {};
@@ -180,11 +180,16 @@ export class DocumentInstance {
   getNode(id: string): Node | null {
     return this.document.getNode(id);
   }
+
+  dispose() {
+    this.disposeFunctions.forEach(fn => fn());
+    this.instancesMap = new Map();
+  }
 }
 
 export class SimulatorRendererContainer implements BuiltinSimulatorRenderer {
   readonly isSimulatorRenderer = true;
-  private dispose?: () => void;
+  private disposeFunctions: Array<() => void> = [];
   readonly history: MemoryHistory;
 
   @obx.ref private _documentInstances: DocumentInstance[] = [];
@@ -193,7 +198,7 @@ export class SimulatorRendererContainer implements BuiltinSimulatorRenderer {
   }
 
   constructor() {
-    this.dispose = host.connect(this, () => {
+    this.disposeFunctions.push(host.connect(this, () => {
       // sync layout config
       this._layout = host.project.get('config').layout;
 
@@ -214,10 +219,10 @@ export class SimulatorRendererContainer implements BuiltinSimulatorRenderer {
 
       // sync device
       this._device = host.device;
-    });
+    }));
     const documentInstanceMap = new Map<string, DocumentInstance>();
     let initialEntry = '/';
-    host.autorun(({ firstRun }) => {
+    this.disposeFunctions.push(host.autorun(({ firstRun }) => {
       this._documentInstances = host.project.documents.map((doc) => {
         let inst = documentInstanceMap.get(doc.id);
         if (!inst) {
@@ -234,7 +239,7 @@ export class SimulatorRendererContainer implements BuiltinSimulatorRenderer {
       } else if (this.history.location.pathname !== path) {
         this.history.replace(path);
       }
-    });
+    }));
     const history = createMemoryHistory({
       initialEntries: [initialEntry],
     });
@@ -468,6 +473,16 @@ export class SimulatorRendererContainer implements BuiltinSimulatorRenderer {
     reactRender(createElement(SimulatorRendererView, { rendererContainer: this }), container);
     host.project.setRendererReady(this);
   }
+
+  dispose() {
+    this.disposeFunctions.forEach(fn => fn());
+    this.documentInstances.forEach(docInst => docInst.dispose());
+    untracked(() => {
+      this._componentsMap = {};
+      this._components = null;
+      this._appContext = null;
+    });
+  }
 }
 
 // Slot/Leaf and Fragment|FunctionComponent polyfill(ref)