feat(selection): 补充策略执行结果查询链路

zhixing-web/src/features/selection/api/selection.query.ts

@@ -0,0 +1,79 @@
+import {
+  useMutation,
+  useQuery,
+  useQueryClient,
+  type QueryClient,
+} from "@tanstack/react-query"
+import { useEffect } from "react"
+
+import {
+  getSelectionResults,
+  getSelectionRun,
+  triggerSelectionRun,
+} from "./selection.api"
+import type { SelectionRunRequest, SelectionStrategy } from "./selection.types"
+
+export const selectionResultsQueryKey = (
+  strategy: SelectionStrategy,
+  targetTradeDate?: string,
+) => ["selection", "results", strategy, targetTradeDate] as const
+
+export const selectionRunQueryKey = (runId: string) =>
+  ["selection", "runs", runId] as const
+
+export function useSelectionResults(
+  strategy: SelectionStrategy,
+  targetTradeDate?: string,
+) {
+  return useQuery({
+    queryFn: ({ signal }) =>
+      getSelectionResults(strategy, targetTradeDate, signal),
+    queryKey: selectionResultsQueryKey(strategy, targetTradeDate),
+  })
+}
+
+export function useSelectionRun(runId: string | null) {
+  const queryClient = useQueryClient()
+  const query = useQuery({
+    enabled: Boolean(runId),
+    queryFn: ({ signal }) => getSelectionRun(runId ?? "", signal),
+    queryKey: selectionRunQueryKey(runId ?? "none"),
+    refetchInterval: (query) =>
+      query.state.data?.status === "running" ? 1500 : false,
+  })
+
+  useEffect(() => {
+    const result = query.data
+    if (!runId || !result || result.status === "running") return
+
+    void invalidateSelectionResults(queryClient, result.strategy)
+  }, [query.data, queryClient, runId])
+
+  return query
+}
+
+export function useTriggerSelectionRun() {
+  const queryClient = useQueryClient()
+
+  return useMutation({
+    mutationFn: ({
+      request,
+      signal,
+    }: {
+      request: SelectionRunRequest
+      signal?: AbortSignal
+    }) => triggerSelectionRun(request, signal),
+    onSuccess: (_accepted, { request }) => {
+      void invalidateSelectionResults(queryClient, request.strategy)
+    },
+  })
+}
+
+export function invalidateSelectionResults(
+  queryClient: QueryClient,
+  strategy: SelectionStrategy,
+) {
+  return queryClient.invalidateQueries({
+    queryKey: ["selection", "results", strategy],
+  })
+}