Avoid re-rendering nodes on click

clauderic · Sep 29, 2024 · cd4eb60 · cd4eb60
1 parent e2a1776
commit cd4eb60
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/packages/core/src/components/DndContext/DndContext.tsx b/packages/core/src/components/DndContext/DndContext.tsx
@@ -319,6 +319,8 @@ export const DndContext = memo(function DndContext({
     activeNodeRect
   );
 
+  const activeSensorRef = useRef<SensorInstance | null>(null);
+  const activatorEventRef = useRef<Event | null>(null);
   const instantiateSensor = useCallback(
     (
       event: React.SyntheticEvent,
@@ -371,6 +373,8 @@ export const DndContext = memo(function DndContext({
               active: id,
             });
             dispatchMonitorEvent({type: 'onDragStart', event});
+            setActiveSensor(activeSensorRef.current);
+            setActivatorEvent(activatorEventRef.current);
           });
         },
         onMove(coordinates) {
@@ -383,10 +387,8 @@ export const DndContext = memo(function DndContext({
         onCancel: createHandler(Action.DragCancel),
       });
 
-      unstable_batchedUpdates(() => {
-        setActiveSensor(sensorInstance);
-        setActivatorEvent(event.nativeEvent);
-      });
+      activeSensorRef.current = sensorInstance;
+      activatorEventRef.current = event.nativeEvent;
 
       function createHandler(type: Action.DragEnd | Action.DragCancel) {
         return async function handler() {
@@ -422,6 +424,8 @@ export const DndContext = memo(function DndContext({
             setOver(null);
             setActiveSensor(null);
             setActivatorEvent(null);
+            activeSensorRef.current = null;
+            activatorEventRef.current = null;
 
             const eventName =
               type === Action.DragEnd ? 'onDragEnd' : 'onDragCancel';