File size: 743 Bytes
246d201
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import { useQuery } from "@tanstack/react-query";
import React from "react";
import InvariantService from "#/api/invariant-service";

type ResponseData = number;

interface UseGetRiskSeverityConfig {
  onSuccess: (data: ResponseData) => void;
}

export const useGetRiskSeverity = (config?: UseGetRiskSeverityConfig) => {
  const data = useQuery<ResponseData>({
    queryKey: ["risk_severity"],
    queryFn: InvariantService.getRiskSeverity,
  });

  const { isFetching, isSuccess, data: riskSeverity } = data;

  React.useEffect(() => {
    if (!isFetching && isSuccess && riskSeverity) {
      config?.onSuccess(riskSeverity);
    }
  }, [isFetching, isSuccess, riskSeverity, config?.onSuccess]);

  return data;
};