From fc43580134c2c8b261a0c50c7e61109f9af215da Mon Sep 17 00:00:00 2001
From: Luke De Feo <lukedefeo@meta.com>
Date: Mon, 12 Dec 2022 07:28:37 -0800
Subject: [PATCH] Memoize tree nodes

Reviewed By: antonk52

Differential Revision: D41838164

fbshipit-source-id: 3184f4ee607f0dd47604265fc259480403c083f0
---
 .../public/ui-debugger/components/Tree2.tsx        | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/desktop/plugins/public/ui-debugger/components/Tree2.tsx b/desktop/plugins/public/ui-debugger/components/Tree2.tsx
index 80e72a392..50ca804b3 100644
--- a/desktop/plugins/public/ui-debugger/components/Tree2.tsx
+++ b/desktop/plugins/public/ui-debugger/components/Tree2.tsx
@@ -88,7 +88,7 @@ export function Tree2({
           instance.uiState.hoveredNodes.set([]);
         }}>
         {treeNodes.map((treeNode, index) => (
-          <TreeItemContainer
+          <MemoTreeItemContainer
             innerRef={refs[index]}
             isUsingKBToScroll={isUsingKBToScroll}
             key={treeNode.id}
@@ -107,6 +107,18 @@ export type TreeNode = UINode & {
   isExpanded: boolean;
 };
 
+const MemoTreeItemContainer = React.memo(
+  TreeItemContainer,
+  (prevProps, nextProps) => {
+    const id = prevProps.treeNode.id;
+    return (
+      prevProps.treeNode === nextProps.treeNode &&
+      id !== prevProps.selectedNode &&
+      id !== nextProps.selectedNode
+    );
+  },
+);
+
 function TreeItemContainer({
   innerRef,
   isUsingKBToScroll,