feat(code-gen): support Updater in RQ setQueryData

compasjs · Jan 9, 2024 · 7ba289b · 7ba289b
1 parent c66bb47
commit 7ba289b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/examples/react-fetch/src/App.tsx b/examples/react-fetch/src/App.tsx
@@ -1,11 +1,10 @@
-import { keepPreviousData } from "@tanstack/react-query";
-import { useEffect, useState } from "react";
+import { useEffect } from "react";
 import "./App.css";
-import { useEmojisEmojisGet } from "./generated/emojis/reactQueries.tsx";
 import {
   fetchWithBaseUrl,
   setFetchFn,
 } from "./generated/common/api-client.tsx";
+import { useEmojisEmojisGet } from "./generated/emojis/reactQueries.tsx";
 
 function App() {
   useEffect(() => {

diff --git a/packages/code-gen/src/api-client/react-query.js b/packages/code-gen/src/api-client/react-query.js
@@ -161,6 +161,7 @@ export function reactQueryGetApiClientFile(generateContext, route) {
 
   // @tanstack/react-query imports
   importCollector.destructure("@tanstack/react-query", "QueryKey");
+  importCollector.destructure("@tanstack/react-query", "Updater");
   importCollector.destructure("@tanstack/react-query", "UseMutationOptions");
   importCollector.destructure("@tanstack/react-query", "UseMutationResult");
   importCollector.destructure("@tanstack/react-query", "UseQueryOptions");
@@ -607,7 +608,9 @@ ${hookName}.setQueryData = (
         })},`
       : ""
   }
-  data: ${contextNames.responseTypeName ?? "unknown"},
+  data: Updater<${contextNames.responseTypeName ?? "unknown"}, ${
+    contextNames.responseTypeName ?? "unknown"
+  }>,
 ) => {
   ${reactQueryCheckIfRequiredVariablesArePresent(
     generateContext,