test new approach

k-rajat19 · Oct 19, 2024 · a3dec2d · a3dec2d
1 parent 0d0201a
commit a3dec2d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/packages/x-data-grid/src/utils/createSelector.ts b/packages/x-data-grid/src/utils/createSelector.ts
@@ -148,7 +148,15 @@ export const createSelector = ((
 }) as unknown as CreateSelectorFunction;
 
 export const createSelectorMemoized: CreateSelectorFunction = (...args: any) => {
-  const selector = (stateOrApiRef: any, selectorArgs: any, instanceId?: any) => {
+  const selector = (...Args: any[]) => {
+    const [stateOrApiRef, ...other] = Args;
+    let instanceId: any;
+    let selectorArgs: any;
+    if (typeof other === 'object' && 'id' in other && typeof other.id === 'number') {
+      instanceId = other;
+    } else {
+      selectorArgs = other;
+    }
     const isAPIRef = checkIsAPIRef(stateOrApiRef);
     const cacheKey = isAPIRef
       ? stateOrApiRef.current.instanceId