fix: total field parsing

vbenjs · Oct 17, 2024 · e603473 · e603473
1 parent 7f89561
commit e603473
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/packages/effects/plugins/src/vxe-table/extends.ts b/packages/effects/plugins/src/vxe-table/extends.ts
@@ -2,7 +2,7 @@ import type { VxeGridProps, VxeUIExport } from 'vxe-table';
 
 import type { VxeGridApi } from './api';
 
-import { isFunction } from '@vben/utils';
+import { getNestedValue, isFunction } from '@vben/utils';
 
 import dayjs from 'dayjs';
 
@@ -40,9 +40,13 @@ function extendProxyOption(
     const data = await configFn(params, formValues, ...args);
 
     // If the page is incorrect, make corrections and request again
-    const total = proxyConfig?.response?.total as string;
+    const response = proxyConfig?.response || {};
+    const total = response?.total;
+
     const { pageSize, currentPage } = params.page;
-    const resultTotal: number = data[total];
+    const resultTotal: number = isFunction(total)
+      ? total({ data, $grid: api.grid })
+      : getNestedValue(data, total || 'total') || 0;
     if (Number(resultTotal)) {
       const currentTotalPage = Math.max(1, Math.ceil(resultTotal / pageSize));
       if (currentPage > currentTotalPage) {