[DataGrid] Fix scrollbar position not being updated after `scrollToIn…

…dexes` (#14877)
mui · Oct 8, 2024 · 018dc1d · 018dc1d
1 parent 309199a
commit 018dc1d
Showing 1 changed file with 7 additions and 13 deletions.
diff --git a/packages/x-data-grid/src/components/virtualization/GridVirtualScrollbar.tsx b/packages/x-data-grid/src/components/virtualization/GridVirtualScrollbar.tsx
@@ -70,8 +70,8 @@ const GridVirtualScrollbar = React.forwardRef<HTMLDivElement, GridVirtualScrollb
   function GridVirtualScrollbar(props, ref) {
     const apiRef = useGridPrivateApiContext();
     const rootProps = useGridRootProps();
-    const isLocked = React.useRef(false);
-    const lastPosition = React.useRef(0);
+    const lastPositionScroller = React.useRef(0);
+    const lastPositionScrollbar = React.useRef(0);
     const scrollbarRef = React.useRef<HTMLDivElement>(null);
     const contentRef = React.useRef<HTMLDivElement>(null);
     const classes = useUtilityClasses(rootProps, props.position);
@@ -96,34 +96,28 @@ const GridVirtualScrollbar = React.forwardRef<HTMLDivElement, GridVirtualScrollb
       const scroller = apiRef.current.virtualScrollerRef.current!;
       const scrollbar = scrollbarRef.current!;
 
-      if (scroller[propertyScroll] === lastPosition.current) {
+      if (scroller[propertyScroll] === lastPositionScroller.current) {
         return;
       }
 
-      if (isLocked.current) {
-        isLocked.current = false;
-        return;
-      }
-      isLocked.current = true;
-
       const value = scroller[propertyScroll] / contentSize;
       scrollbar[propertyScroll] = value * scrollbarInnerSize;
 
-      lastPosition.current = scroller[propertyScroll];
+      lastPositionScrollbar.current = scrollbar[propertyScroll];
     });
 
     const onScrollbarScroll = useEventCallback(() => {
       const scroller = apiRef.current.virtualScrollerRef.current!;
       const scrollbar = scrollbarRef.current!;
 
-      if (isLocked.current) {
-        isLocked.current = false;
+      if (scrollbar[propertyScroll] === lastPositionScrollbar.current) {
         return;
       }
-      isLocked.current = true;
 
       const value = scrollbar[propertyScroll] / scrollbarInnerSize;
       scroller[propertyScroll] = value * contentSize;
+
+      lastPositionScroller.current = scroller[propertyScroll];
     });
 
     useOnMount(() => {