80 lines
2.0 KiB
TypeScript
80 lines
2.0 KiB
TypeScript
|
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],
|
||
|
|
})
|
||
|
|
}
|