Merge pull request #5101 from Tongzai123/feat/admin-select-all-filtered-results

feat(admin): 支持按筛选结果全选账号

frontend/src/utils/__tests__/accountSelection.spec.ts

@@ -0,0 +1,68 @@
+import { describe, expect, it, vi } from 'vitest'
+
+import { fetchAllAccountIds } from '../accountSelection'
+
+describe('fetchAllAccountIds', () => {
+  it('loads every page with the same filter snapshot and returns unique account IDs', async () => {
+    const fetchPage = vi.fn(async (page: number, pageSize: number, _filters: Record<string, unknown>) => {
+      const start = (page - 1) * pageSize + 1
+      const end = Math.min(page * pageSize, 2505)
+      return {
+        items: Array.from({ length: end - start + 1 }, (_, index) => ({ id: start + index })),
+        total: 2505,
+        page,
+        page_size: pageSize,
+        pages: 3
+      }
+    })
+
+    const filters = {
+      platform: 'grok',
+      status: 'active',
+      search: 'example'
+    }
+
+    const ids = await fetchAllAccountIds(fetchPage, filters)
+
+    expect(ids).toHaveLength(2505)
+    expect(ids[0]).toBe(1)
+    expect(ids[2504]).toBe(2505)
+    expect(fetchPage).toHaveBeenCalledTimes(3)
+    expect(fetchPage).toHaveBeenNthCalledWith(1, 1, 1000, {
+      ...filters,
+      lite: '1',
+      include_scheduler_score: '0'
+    })
+    expect(fetchPage).toHaveBeenNthCalledWith(3, 3, 1000, {
+      ...filters,
+      lite: '1',
+      include_scheduler_score: '0'
+    })
+  })
+
+  it('rejects an incomplete or duplicated result instead of returning a partial selection', async () => {
+    const fetchPage = vi.fn().mockResolvedValue({
+      items: [{ id: 1 }, { id: 1 }],
+      total: 2,
+      page: 1,
+      page_size: 1000,
+      pages: 1
+    })
+
+    await expect(fetchAllAccountIds(fetchPage, {})).rejects.toThrow('账号列表结果不完整')
+  })
+
+  it('propagates a later page failure', async () => {
+    const fetchPage = vi.fn()
+      .mockResolvedValueOnce({
+        items: Array.from({ length: 1000 }, (_, index) => ({ id: index + 1 })),
+        total: 1001,
+        page: 1,
+        page_size: 1000,
+        pages: 2
+      })
+      .mockRejectedValueOnce(new Error('page 2 failed'))
+
+    await expect(fetchAllAccountIds(fetchPage, { group: '7' })).rejects.toThrow('page 2 failed')
+  })
+})

frontend/src/utils/accountSelection.ts

@@ -0,0 +1,45 @@
+interface AccountIDRow {
+  id: number
+}
+
+interface AccountListPage {
+  items: AccountIDRow[]
+  total: number
+  pages?: number
+}
+
+type AccountPageFetcher = (
+  page: number,
+  pageSize: number,
+  filters: Record<string, unknown>
+) => Promise<AccountListPage>
+
+const SELECT_ALL_PAGE_SIZE = 1000
+
+export async function fetchAllAccountIds(
+  fetchPage: AccountPageFetcher,
+  filters: Record<string, unknown>
+): Promise<number[]> {
+  const requestFilters = {
+    ...filters,
+    lite: '1',
+    include_scheduler_score: '0'
+  }
+  const firstPage = await fetchPage(1, SELECT_ALL_PAGE_SIZE, requestFilters)
+  const pageCount = Math.max(
+    firstPage.pages ?? 0,
+    Math.ceil(firstPage.total / SELECT_ALL_PAGE_SIZE)
+  )
+  const ids = firstPage.items.map(account => account.id)
+
+  for (let page = 2; page <= pageCount; page++) {
+    const result = await fetchPage(page, SELECT_ALL_PAGE_SIZE, requestFilters)
+    ids.push(...result.items.map(account => account.id))
+  }
+
+  const uniqueIDs = Array.from(new Set(ids))
+  if (uniqueIDs.length !== firstPage.total) {
+    throw new Error('账号列表结果不完整')
+  }
+  return uniqueIDs
+}