forked from mirrors/thatmattlove-hyperglass
50 lines
1.5 KiB
TypeScript
50 lines
1.5 KiB
TypeScript
import { useQuery } from 'react-query';
|
|
import { useConfig } from '~/context';
|
|
import { fetchWithTimeout } from '~/util';
|
|
|
|
import type { TFormQuery } from '~/types';
|
|
import type { TUseLGQueryFn } from './types';
|
|
|
|
/**
|
|
* Custom hook handle submission of a query to the hyperglass backend.
|
|
*/
|
|
export function useLGQuery(query: TFormQuery) {
|
|
const { request_timeout, cache } = useConfig();
|
|
const controller = new AbortController();
|
|
|
|
async function runQuery(ctx: TUseLGQueryFn): Promise<TQueryResponse> {
|
|
const [url, data] = ctx.queryKey;
|
|
const { queryLocation, queryTarget, queryType, queryVrf } = data;
|
|
const res = await fetchWithTimeout(
|
|
url,
|
|
{
|
|
method: 'POST',
|
|
headers: { 'content-type': 'application/json' },
|
|
body: JSON.stringify({
|
|
query_location: queryLocation,
|
|
query_target: queryTarget,
|
|
query_type: queryType,
|
|
query_vrf: queryVrf,
|
|
}),
|
|
mode: 'cors',
|
|
},
|
|
request_timeout * 1000,
|
|
controller,
|
|
);
|
|
return await res.json();
|
|
}
|
|
return useQuery<TQueryResponse, Response | TQueryResponse | Error>(
|
|
['/api/query/', query],
|
|
runQuery,
|
|
{
|
|
// Invalidate react-query's cache just shy of the configured cache timeout.
|
|
cacheTime: cache.timeout * 1000 * 0.95,
|
|
// Don't refetch when window refocuses.
|
|
refetchOnWindowFocus: false,
|
|
// Don't automatically refetch query data (queries should be on-off).
|
|
refetchInterval: false,
|
|
// Don't refetch on component remount.
|
|
refetchOnMount: false,
|
|
},
|
|
);
|
|
}
|