Memoize tree nodes

Reviewed By: antonk52

Differential Revision: D41838164

fbshipit-source-id: 3184f4ee607f0dd47604265fc259480403c083f0

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,