Skip to content

Commit 7ba289b

Browse files
committed
feat(code-gen): support Updater in RQ setQueryData
1 parent c66bb47 commit 7ba289b

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

examples/react-fetch/src/App.tsx

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,10 @@
1-
import { keepPreviousData } from "@tanstack/react-query";
2-
import { useEffect, useState } from "react";
1+
import { useEffect } from "react";
32
import "./App.css";
4-
import { useEmojisEmojisGet } from "./generated/emojis/reactQueries.tsx";
53
import {
64
fetchWithBaseUrl,
75
setFetchFn,
86
} from "./generated/common/api-client.tsx";
7+
import { useEmojisEmojisGet } from "./generated/emojis/reactQueries.tsx";
98

109
function App() {
1110
useEffect(() => {

packages/code-gen/src/api-client/react-query.js

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,7 @@ export function reactQueryGetApiClientFile(generateContext, route) {
161161

162162
// @tanstack/react-query imports
163163
importCollector.destructure("@tanstack/react-query", "QueryKey");
164+
importCollector.destructure("@tanstack/react-query", "Updater");
164165
importCollector.destructure("@tanstack/react-query", "UseMutationOptions");
165166
importCollector.destructure("@tanstack/react-query", "UseMutationResult");
166167
importCollector.destructure("@tanstack/react-query", "UseQueryOptions");
@@ -607,7 +608,9 @@ ${hookName}.setQueryData = (
607608
})},`
608609
: ""
609610
}
610-
data: ${contextNames.responseTypeName ?? "unknown"},
611+
data: Updater<${contextNames.responseTypeName ?? "unknown"}, ${
612+
contextNames.responseTypeName ?? "unknown"
613+
}>,
611614
) => {
612615
${reactQueryCheckIfRequiredVariablesArePresent(
613616
generateContext,

0 commit comments

Comments
 (0)