urbanLifeline/dify/web/service/use-base.ts

import {
  type QueryKey,
  useQueryClient,
} from '@tanstack/react-query'

export const useInvalid = (key?: QueryKey) => {
  const queryClient = useQueryClient()
  return () => {
    if (!key)
      return
    queryClient.invalidateQueries(
      {
        queryKey: key,
      },
    )
  }
}

export const useReset = (key?: QueryKey) => {
  const queryClient = useQueryClient()
  return () => {
    if (!key)
      return
    queryClient.resetQueries(
      {
        queryKey: key,
      },
    )
  }
}