From 1841dfa2cae93e2ab4eb05d63f3a832a095983da Mon Sep 17 00:00:00 2001 From: moklick Date: Wed, 16 Nov 2022 18:28:25 +0100 Subject: [PATCH] refactor(node-toolbar): define zustand selector outside of the component --- packages/node-toolbar/src/NodeToolbarPortal.tsx | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/packages/node-toolbar/src/NodeToolbarPortal.tsx b/packages/node-toolbar/src/NodeToolbarPortal.tsx index f899352f..732362e5 100644 --- a/packages/node-toolbar/src/NodeToolbarPortal.tsx +++ b/packages/node-toolbar/src/NodeToolbarPortal.tsx @@ -2,8 +2,10 @@ import { ReactNode } from 'react'; import { createPortal } from 'react-dom'; import { ReactFlowState, useStore } from '@reactflow/core'; +const selector = (state: ReactFlowState) => state.domNode?.querySelector('.react-flow__renderer'); + function NodeToolbarPortal({ children }: { children: ReactNode }) { - const wrapperRef = useStore((state: ReactFlowState) => state.domNode?.querySelector('.react-flow__renderer')); + const wrapperRef = useStore(selector); if (!wrapperRef) { return null;