refactor(react-query): split prefetch.ts (#8028)

* fix(react-query): split prefetch.ts * chore: update --------- Co-authored-by: Dominik Dorfmeister <office@dorfmeister.cc>
TanStack · Sep 8, 2024 · 42715e7 · 42715e7
1 parent 5cb0bd4
commit 42715e7
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 19 deletions.
diff --git a/packages/react-query/src/__tests__/prefetch.test-d.tsx b/packages/react-query/src/__tests__/prefetch.test-d.tsx
@@ -1,5 +1,5 @@
 import { describe, expectTypeOf, it } from 'vitest'
-import { usePrefetchInfiniteQuery, usePrefetchQuery } from '../prefetch'
+import { usePrefetchInfiniteQuery, usePrefetchQuery } from '..'
 
 describe('usePrefetchQuery', () => {
   it('should return nothing', () => {

diff --git a/packages/react-query/src/index.ts b/packages/react-query/src/index.ts
@@ -15,7 +15,8 @@ export type {
   SuspenseQueriesResults,
   SuspenseQueriesOptions,
 } from './useSuspenseQueries'
-export { usePrefetchQuery, usePrefetchInfiniteQuery } from './prefetch'
+export { usePrefetchQuery } from './usePrefetchQuery'
+export { usePrefetchInfiniteQuery } from './usePrefetchInfiniteQuery'
 export { queryOptions } from './queryOptions'
 export type {
   DefinedInitialDataOptions,

diff --git a/packages/react-query/src/prefetch.ts → ...ct-query/src/usePrefetchInfiniteQuery.tsx b/packages/react-query/src/prefetch.ts → ...ct-query/src/usePrefetchInfiniteQuery.tsx
@@ -2,27 +2,10 @@ import { useQueryClient } from './QueryClientProvider'
 import type {
   DefaultError,
   FetchInfiniteQueryOptions,
-  FetchQueryOptions,
   QueryClient,
   QueryKey,
 } from '@tanstack/query-core'
 
-export function usePrefetchQuery<
-  TQueryFnData = unknown,
-  TError = DefaultError,
-  TData = TQueryFnData,
-  TQueryKey extends QueryKey = QueryKey,
->(
-  options: FetchQueryOptions<TQueryFnData, TError, TData, TQueryKey>,
-  queryClient?: QueryClient,
-) {
-  const client = useQueryClient(queryClient)
-
-  if (!client.getQueryState(options.queryKey)) {
-    client.prefetchQuery(options)
-  }
-}
-
 export function usePrefetchInfiniteQuery<
   TQueryFnData = unknown,
   TError = DefaultError,

diff --git a/packages/react-query/src/usePrefetchQuery.tsx b/packages/react-query/src/usePrefetchQuery.tsx
@@ -0,0 +1,23 @@
+import { useQueryClient } from './QueryClientProvider'
+import type {
+  DefaultError,
+  FetchQueryOptions,
+  QueryClient,
+  QueryKey,
+} from '@tanstack/query-core'
+
+export function usePrefetchQuery<
+  TQueryFnData = unknown,
+  TError = DefaultError,
+  TData = TQueryFnData,
+  TQueryKey extends QueryKey = QueryKey,
+>(
+  options: FetchQueryOptions<TQueryFnData, TError, TData, TQueryKey>,
+  queryClient?: QueryClient,
+) {
+  const client = useQueryClient(queryClient)
+
+  if (!client.getQueryState(options.queryKey)) {
+    client.prefetchQuery(options)
+  }
+}