fix: hoc 组件缓存

alibaba · Oct 17, 2023 · 76557f4 · 76557f4
1 parent d4822af
commit 76557f4
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/packages/renderer-core/src/hoc/index.tsx b/packages/renderer-core/src/hoc/index.tsx
@@ -62,6 +62,8 @@ function patchDidCatch(Comp: any, { baseRenderer }: Options) {
   }
 }
 
+const cache = new Map();
+
 export function compWrapper(Comp: any, options: Options) {
   const { createElement, Component, forwardRef } = adapter.getRuntime();
   if (
@@ -72,6 +74,11 @@ export function compWrapper(Comp: any, options: Options) {
     patchDidCatch(Comp, options);
     return Comp;
   }
+
+  if (cache.has(options.schema.id)) {
+    return cache.get(options.schema.id);
+  }
+
   class Wrapper extends Component {
     render() {
       return createElement(Comp, { ...this.props, ref: this.props.forwardRef });
@@ -82,10 +89,14 @@ export function compWrapper(Comp: any, options: Options) {
   patchDidCatch(Wrapper, options);
   patchDidMount(Wrapper, options);
 
-  return cloneEnumerableProperty(
+  const WrapperComponent = cloneEnumerableProperty(
     forwardRef((props: any, ref: any) => {
       return createElement(Wrapper, { ...props, forwardRef: ref });
     }),
     Comp,
   );
+
+  cache.set(options.schema.id, WrapperComponent);
+
+  return WrapperComponent;
 }