Update src/PickerTrigger/index.tsx

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
react-component · Oct 26, 2024 · 1f63ee9 · 1f63ee9
1 parent e5f2671
commit 1f63ee9
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/PickerTrigger/index.tsx b/src/PickerTrigger/index.tsx
@@ -101,14 +101,17 @@ function PickerTrigger({
       stretch="minWidth"
       getPopupContainer={getPopupContainer}
       onPopupAlign={(_, align) => {
-        Object.keys(BUILT_IN_PLACEMENTS).forEach((key) => {
-          if (
+        if (!setAlignedPlacement) return;
+
+        const matchedKey = Object.keys(BUILT_IN_PLACEMENTS).find(
+          (key) =>
             BUILT_IN_PLACEMENTS[key].points[0] === align.points[0] &&
             BUILT_IN_PLACEMENTS[key].points[1] === align.points[1]
-          ) {
-            setAlignedPlacement?.(key);
-          }
-        });
+        );
+
+        if (matchedKey) {
+          setAlignedPlacement(matchedKey);
+        }
       }}
       onPopupVisibleChange={(nextVisible) => {
         if (!nextVisible) {