feat(react-query): add optional queryClient parameter to usePrefetchQuery (#8014)

Co-authored-by: Dominik Dorfmeister <[email protected]>

Author: manudeli

packages/react-query/src/prefetch.ts

@@ -3,6 +3,7 @@ import type {
   DefaultError,
   FetchInfiniteQueryOptions,
   FetchQueryOptions,
+  QueryClient,
   QueryKey,
 } from '@tanstack/query-core'
 
@@ -11,11 +12,14 @@ export function usePrefetchQuery<
   TError = DefaultError,
   TData = TQueryFnData,
   TQueryKey extends QueryKey = QueryKey,
->(options: FetchQueryOptions<TQueryFnData, TError, TData, TQueryKey>) {
-  const queryClient = useQueryClient()
+>(
+  options: FetchQueryOptions<TQueryFnData, TError, TData, TQueryKey>,
+  queryClient?: QueryClient,
+) {
+  const client = useQueryClient(queryClient)
 
-  if (!queryClient.getQueryState(options.queryKey)) {
-    queryClient.prefetchQuery(options)
+  if (!client.getQueryState(options.queryKey)) {
+    client.prefetchQuery(options)
   }
 }
 
@@ -33,10 +37,11 @@ export function usePrefetchInfiniteQuery<
     TQueryKey,
     TPageParam
   >,
+  queryClient?: QueryClient,
 ) {
-  const queryClient = useQueryClient()
+  const client = useQueryClient(queryClient)
 
-  if (!queryClient.getQueryState(options.queryKey)) {
-    queryClient.prefetchInfiniteQuery(options)
+  if (!client.getQueryState(options.queryKey)) {
+    client.prefetchInfiniteQuery(options)
   }
 }