feat: add options to setQueryData

TanStack · Oct 7, 2020 · 1f5b13e · 1f5b13e
1 parent de0492f
commit 1f5b13e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/core/query.ts b/src/core/query.ts
@@ -64,7 +64,7 @@ interface FetchMoreOptions {
   direction: 'forward' | 'backward'
 }
 
-interface SetDataOptions {
+export interface SetDataOptions {
   pageParams?: unknown[]
   updatedAt?: number
 }

diff --git a/src/core/queryClient.ts b/src/core/queryClient.ts
@@ -21,7 +21,7 @@ import type {
   QueryOptions,
   RefetchOptions,
 } from './types'
-import type { QueryState } from './query'
+import type { QueryState, SetDataOptions } from './query'
 import type { QueryCache } from './queryCache'
 import { QueriesObserver } from './queriesObserver'
 import { QueryObserver } from './queryObserver'
@@ -109,11 +109,12 @@ export class QueryClient {
 
   setQueryData<TData>(
     queryKey: QueryKey,
-    updater: Updater<TData | undefined, TData>
+    updater: Updater<TData | undefined, TData>,
+    options?: SetDataOptions
   ): TData {
     const parsedOptions = parseQueryArgs(queryKey)
     const defaultedOptions = this.defaultQueryOptions(parsedOptions)
-    return this.cache.build(defaultedOptions).setData(updater)
+    return this.cache.build(defaultedOptions).setData(updater, options)
   }
 
   getQueryState<TData = unknown, TError = undefined>(