改造实时事件缓存同步

2026-08-17 17:32:02 +08:00
parent be7c124d65
commit 2aa8769661
16 changed files with 837 additions and 351 deletions
@@ -32,18 +32,15 @@ export default function WorkerLayout() {
  const location = useLocation()
  const isMobile = useIsMobile()
  const { message, modal } = App.useApp()
+  const queryClient = useQueryClient()
  const username = getWorkerUsername() || '打手'
  const status = getWorkerStatus()
-  const queryClient = useQueryClient()
  const currentKey = resolveSelectedKey(location.pathname)

  useRealtimeEvents({
    url: '/api/v1/worker/realtime',
    token: getWorkerToken(),
-    onEvent: (event) => handleWorkerRealtimeEvent(event, queryClient),
-    onConnected: () => {
-      void invalidateWorkerRealtimeQueries(queryClient)
-    },
+    onEvent: (event) => applyWorkerRealtimeEvent(event, queryClient),
  })

  async function submitLogout() {
@@ -156,40 +153,179 @@ export default function WorkerLayout() {
  )
 }

-function handleWorkerRealtimeEvent(
+function applyWorkerRealtimeEvent(
  event: RealtimeEvent,
  queryClient: ReturnType<typeof useQueryClient>,
 ) {
-  if (event.type === 'work_order.changed') {
-    if (event.scopes.includes('worker_hall')) {
-      void queryClient.invalidateQueries({ queryKey: ['worker-hall-orders'] })
-      void queryClient.invalidateQueries({ queryKey: ['worker-hall-summary'] })
-    }
+  if (event.type === 'work_order.changed' && isRecord(event.data)) {
+    const data = event.data
    if (event.scopes.includes('worker_my_orders')) {
-      void queryClient.invalidateQueries({ queryKey: ['worker-my-orders'] })
-      void queryClient.invalidateQueries({ queryKey: ['worker-my-orders-overview'] })
+      patchWorkerMyOrderCaches(queryClient, { ...event, data })
+    }
+    if (event.scopes.includes('worker_hall')) {
+      patchWorkerHallCaches(queryClient, { ...event, data })
    }
  }
-  if (event.type === 'worker_finance_request.changed') {
-    void queryClient.invalidateQueries({ queryKey: ['worker-finance-requests'] })
-    void queryClient.invalidateQueries({ queryKey: ['worker-profile'] })
+
+  if (event.type === 'worker_wallet.changed' && isRecord(event.data)) {
+    const walletData = event.data
+    queryClient.setQueryData(['worker-profile'], (current: unknown) => {
+      if (!isRecord(current) || !isRecord(current.data) || !isRecord(current.data.worker))
+        return current
+      return {
+        ...current,
+        data: {
+          ...current.data,
+          worker: { ...current.data.worker, wallet: walletData.wallet || walletData },
+        },
+      }
+    })
  }
-  if (event.type === 'worker_wallet.changed') {
-    void queryClient.invalidateQueries({ queryKey: ['worker-wallet-ledgers'] })
-    void queryClient.invalidateQueries({ queryKey: ['worker-profile'] })
+
+  if (event.type === 'worker_finance_request.changed' && isRecord(event.data)) {
+    const data = event.data
+    queryClient.setQueriesData({ queryKey: ['worker-finance-requests'] }, (current: unknown) =>
+      patchFinanceRequestListEnvelope(current, { ...event, data }),
+    )
  }
 }

-function invalidateWorkerRealtimeQueries(queryClient: ReturnType<typeof useQueryClient>) {
-  return Promise.all([
-    queryClient.invalidateQueries({ queryKey: ['worker-hall-orders'] }),
-    queryClient.invalidateQueries({ queryKey: ['worker-hall-summary'] }),
-    queryClient.invalidateQueries({ queryKey: ['worker-my-orders'] }),
-    queryClient.invalidateQueries({ queryKey: ['worker-my-orders-overview'] }),
-    queryClient.invalidateQueries({ queryKey: ['worker-profile'] }),
-    queryClient.invalidateQueries({ queryKey: ['worker-wallet-ledgers'] }),
-    queryClient.invalidateQueries({ queryKey: ['worker-finance-requests'] }),
-  ])
+function patchWorkerMyOrderCaches(
+  queryClient: ReturnType<typeof useQueryClient>,
+  event: RealtimeEvent,
+) {
+  for (const [queryKey, current] of queryClient.getQueriesData({
+    queryKey: ['worker-my-orders'],
+  })) {
+    const status = String(queryKey[1] || '')
+    const keyword = String(queryKey[2] || '')
+    const page = Number(queryKey[3] || 1)
+    const item = event.data as {
+      status?: unknown
+      productName?: unknown
+      platformOrderId?: unknown
+    }
+    const matchesStatus = !status || status === String(item.status || '')
+    const matchesKeyword = matchesWorkOrderKeyword(item, keyword)
+    const matches = matchesStatus && matchesKeyword
+    const shouldInsert = event.operation === 'upsert' && page === 1 && matches
+    const operation = event.operation === 'remove' || !matches ? 'remove' : 'upsert'
+    queryClient.setQueryData(
+      queryKey,
+      patchWorkOrderListEnvelope(current, { ...event, operation }, shouldInsert),
+    )
+  }
+}
+
+function patchWorkerHallCaches(
+  queryClient: ReturnType<typeof useQueryClient>,
+  event: RealtimeEvent,
+) {
+  const workerSnapshot = applyWorkerDepositRule(queryClient, event.data)
+  const item = workerSnapshot as {
+    status?: unknown
+    categoryId?: unknown
+    productName?: unknown
+    platformOrderId?: unknown
+  }
+  for (const [queryKey, current] of queryClient.getQueriesData({
+    queryKey: ['worker-hall-orders'],
+  })) {
+    const keyword = String(queryKey[1] || '')
+    const categoryId = Number(queryKey[2] || 0)
+    const matchesCategory = !categoryId || categoryId === Number(item.categoryId || 0)
+    const matches =
+      String(item.status || '') === 'open' &&
+      matchesCategory &&
+      matchesWorkOrderKeyword(item, keyword)
+    const shouldInsert = event.operation === 'upsert' && matches
+    const operation = event.operation === 'remove' || !matches ? 'remove' : 'upsert'
+    queryClient.setQueryData(
+      queryKey,
+      patchInfiniteWorkOrderEnvelope(
+        current,
+        { ...event, data: workerSnapshot, operation },
+        shouldInsert,
+      ),
+    )
+  }
+}
+
+function applyWorkerDepositRule(queryClient: ReturnType<typeof useQueryClient>, data: unknown) {
+  if (!isRecord(data)) return data
+  const profile = queryClient.getQueryData(['worker-profile'])
+  const depositFreeAmount =
+    isRecord(profile) && isRecord(profile.data) && isRecord(profile.data.worker)
+      ? Number(
+          (profile.data.worker.level as { permissions?: { depositFreeAmount?: unknown } } | null)
+            ?.permissions?.depositFreeAmount || 0,
+        )
+      : 0
+  const requiredDepositAmount = Number(data.requiredDepositAmount || 0)
+  return {
+    ...data,
+    freezeDepositAmount: Math.max(0, requiredDepositAmount - depositFreeAmount),
+  }
+}
+
+function matchesWorkOrderKeyword(
+  item: { productName?: unknown; platformOrderId?: unknown },
+  keyword: string,
+) {
+  const normalizedKeyword = keyword.trim().toLowerCase()
+  if (!normalizedKeyword) return true
+  return [item.productName, item.platformOrderId]
+    .map((value) => String(value || '').toLowerCase())
+    .some((value) => value.includes(normalizedKeyword))
+}
+
+function patchWorkOrderListEnvelope(current: unknown, event: RealtimeEvent, insert = false) {
+  if (!isRecord(current) || !isRecord(current.data) || !Array.isArray(current.data.items)) {
+    return current
+  }
+  const id = Number((event.data as { workOrderId?: unknown }).workOrderId || event.entityId)
+  const items = current.data.items as unknown[]
+  const nextItems =
+    event.operation === 'remove'
+      ? items.filter((entry) => Number((entry as { workOrderId?: unknown })?.workOrderId) !== id)
+      : upsertById(items, event.data, 'workOrderId', insert)
+  return { ...current, data: { ...current.data, items: nextItems } }
+}
+
+function patchInfiniteWorkOrderEnvelope(current: unknown, event: RealtimeEvent, insert = false) {
+  if (!isRecord(current) || !Array.isArray(current.pages)) return current
+  return {
+    ...current,
+    pages: current.pages.map((page: unknown, index) =>
+      patchWorkOrderListEnvelope(page, event, insert && index === 0),
+    ),
+  }
+}
+
+function patchFinanceRequestListEnvelope(current: unknown, event: RealtimeEvent) {
+  if (!isRecord(current) || !isRecord(current.data) || !Array.isArray(current.data.items)) {
+    return current
+  }
+  const id = Number((event.data as { requestId?: unknown }).requestId || event.entityId)
+  const items = current.data.items as unknown[]
+  const nextItems =
+    event.operation === 'remove'
+      ? items.filter((entry) => Number((entry as { requestId?: unknown })?.requestId) !== id)
+      : upsertById(items, event.data, 'requestId')
+  return { ...current, data: { ...current.data, items: nextItems } }
+}
+
+function upsertById(items: unknown[], nextItem: unknown, idKey: string, insert = false) {
+  const id = Number((nextItem as Record<string, unknown>)[idKey] || 0)
+  const index = items.findIndex(
+    (item) => Number((item as Record<string, unknown>)?.[idKey] || 0) === id,
+  )
+  if (index < 0) return insert ? [nextItem, ...items] : items
+  return items.map((item, itemIndex) => (itemIndex === index ? nextItem : item))
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+  return Boolean(value && typeof value === 'object' && !Array.isArray(value))
 }

 function resolveSelectedKey(pathname: string) {