Move first render marked in an effect

mui · Oct 11, 2024 · 532a1bf · 532a1bf
1 parent 3e1a1fd
commit 532a1bf
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/x-data-grid/src/hooks/features/rowSelection/useGridRowSelection.ts b/packages/x-data-grid/src/hooks/features/rowSelection/useGridRowSelection.ts
@@ -446,7 +446,6 @@ export const useGridRowSelection = (
   const removeOutdatedSelection = React.useCallback(
     (sortModelUpdated = false) => {
       if (isFirstRender.current) {
-        isFirstRender.current = false;
         return;
       }
       const currentSelection = gridRowSelectionStateSelector(apiRef.current.state);
@@ -762,4 +761,10 @@ export const useGridRowSelection = (
   React.useEffect(() => {
     runIfRowSelectionIsEnabled(removeOutdatedSelection);
   }, [removeOutdatedSelection, runIfRowSelectionIsEnabled]);
+
+  React.useEffect(() => {
+    if (isFirstRender.current) {
+      isFirstRender.current = false;
+    }
+  }, []);
 };