diff --git a/lowcode-editor/src/editor/components/SelectedMask/index.tsx b/lowcode-editor/src/editor/components/SelectedMask/index.tsx index 8abff0c..4abe575 100644 --- a/lowcode-editor/src/editor/components/SelectedMask/index.tsx +++ b/lowcode-editor/src/editor/components/SelectedMask/index.tsx @@ -76,9 +76,15 @@ function SelectedMask({ containerClassName, portalWrapperClassName, componentId }); } - const el = useMemo(() => { - return document.querySelector(`.${portalWrapperClassName}`)! - }, []); + const [el, setEl] = useState( + document.querySelector(`.${portalWrapperClassName}`) + ); + + useEffect(() => { + if (!el) { + setEl(document.querySelector(`.${portalWrapperClassName}`)); + } + }); const curSelectedComponent = useMemo(() => { return getComponentById(componentId, components); @@ -102,44 +108,46 @@ function SelectedMask({ containerClassName, portalWrapperClassName, componentId }, [curComponent]); - return createPortal(( - <> -
-