在 Ant Design 中使用过滤器和 ProTable

我在 Ant Design Pro 中使用ProTable。表的数据源由 API 从服务器接收。如果用户不是管理员，我会通过他的 .dataSource 过滤数据源userid。问题是我无法获得两种情况（管理员和用户）的实际总页数。

这是我的 API 请求函数的代码：

function getRule(req, res, u) {
  let realUrl = u;

  if (!realUrl || Object.prototype.toString.call(realUrl) !== '[object String]') {
    realUrl = req.url;
  }

  const { current = 1, pageSize = 5 } = req.query;
  const params = parse(realUrl, true).query;
  let dataSource = [...tableListDataSource].slice((current - 1) * pageSize, current * pageSize);
  const filter = JSON.parse(params.filter);

  if (filter) {
    if (Object.keys(filter).length > 0) {
      dataSource = tableListDataSource.filter((item) => {
        return Object.keys(filter).some((key) => {
          if (!filter[key]) {
            return true;
          }

          if (filter[key].includes(`${item[key]}`)) {
            return true;
          }

          return false;
        });
      });
    }
  }

  // TODO: Fix total count of pages for admin
  const total = () => {
    if (Object.keys(filter).length > 0) {
      return dataSource.length
    } else {
      return tableListDataSource.length
    }
  }

  const result = {
    data: dataSource,
    total: total,
    success: true,
    pageSize,
    current: parseInt(`${params.currentPage}`, 10) || 1,
  };
  return res.json(result);
}

但是dataSource.length对于用户和tableListDataSource.length管理员，我得到dataSource.length了两种情况。我该如何解决？