fix: set data methods for queries

Author: js2me

src/mobx-inifinite-query.ts

@@ -11,6 +11,7 @@ import {
   InfiniteQueryObserverResult,
   InfiniteData,
   RefetchOptions,
+  SetDataOptions,
 } from '@tanstack/query-core';
 import { IDisposer } from 'disposer-util';
 import { LinkedAbortController } from 'linked-abort-controller';
@@ -247,8 +248,17 @@ export class MobxInfiniteQuery<
     return hashKey(queryKey);
   }
 
-  setData(data: TData) {
-    this.queryClient.setQueryData<TData>(this.options.queryKey, data);
+  setData(
+    data:
+      | InfiniteData<TData>
+      | ((input: InfiniteData<TData> | undefined) => InfiniteData<TData>),
+    options?: SetDataOptions,
+  ) {
+    this.queryClient.setQueryData<InfiniteData<TData>>(
+      this.options.queryKey,
+      data,
+      options,
+    );
   }
 
   fetchNextPage(options?: FetchNextPageOptions | undefined) {

src/mobx-query.ts

@@ -10,6 +10,7 @@ import {
   QueryObserverOptions,
   QueryObserverResult,
   RefetchOptions,
+  SetDataOptions,
 } from '@tanstack/query-core';
 import { IDisposer } from 'disposer-util';
 import { LinkedAbortController } from 'linked-abort-controller';
@@ -210,8 +211,11 @@ export class MobxQuery<
     return hashKey(queryKey);
   }
 
-  setData(data: TData) {
-    this.queryClient.setQueryData<TData>(this.options.queryKey, data);
+  setData(
+    data: TData | ((input: TData | undefined) => TData),
+    options?: SetDataOptions,
+  ) {
+    this.queryClient.setQueryData<TData>(this.options.queryKey, data, options);
   }
 
   update(options: Partial<QueryObserverOptions<TData, TError, TQueryKey>>) {