@@ -5,6 +5,7 @@ import type {
5
5
InfiniteData ,
6
6
InfiniteQueryObserverOptions ,
7
7
InfiniteQueryObserverResult ,
8
+ QueryKey ,
8
9
} from './types'
9
10
import type { QueryClient } from './queryClient'
10
11
import {
@@ -27,12 +28,14 @@ export class InfiniteQueryObserver<
27
28
TQueryFnData = unknown ,
28
29
TError = unknown ,
29
30
TData = TQueryFnData ,
30
- TQueryData = TQueryFnData
31
+ TQueryData = TQueryFnData ,
32
+ TQueryKey extends QueryKey = QueryKey
31
33
> extends QueryObserver <
32
34
TQueryFnData ,
33
35
TError ,
34
36
InfiniteData < TData > ,
35
- InfiniteData < TQueryData >
37
+ InfiniteData < TQueryData > ,
38
+ TQueryKey
36
39
> {
37
40
// Type override
38
41
subscribe ! : (
@@ -54,7 +57,8 @@ export class InfiniteQueryObserver<
54
57
TQueryFnData ,
55
58
TError ,
56
59
TData ,
57
- TQueryData
60
+ TQueryData ,
61
+ TQueryKey
58
62
>
59
63
) {
60
64
super ( client , options )
@@ -71,7 +75,8 @@ export class InfiniteQueryObserver<
71
75
TQueryFnData ,
72
76
TError ,
73
77
TData ,
74
- TQueryData
78
+ TQueryData ,
79
+ TQueryKey
75
80
> ,
76
81
notifyOptions ?: NotifyOptions
77
82
) : void {
@@ -89,7 +94,8 @@ export class InfiniteQueryObserver<
89
94
TQueryFnData ,
90
95
TError ,
91
96
TData ,
92
- TQueryData
97
+ TQueryData ,
98
+ TQueryKey
93
99
>
94
100
) : InfiniteQueryObserverResult < TData , TError > {
95
101
options . behavior = infiniteQueryBehavior ( )
@@ -125,12 +131,13 @@ export class InfiniteQueryObserver<
125
131
}
126
132
127
133
protected createResult (
128
- query : Query < TQueryFnData , TError , InfiniteData < TQueryData > > ,
134
+ query : Query < TQueryFnData , TError , InfiniteData < TQueryData > , TQueryKey > ,
129
135
options : InfiniteQueryObserverOptions <
130
136
TQueryFnData ,
131
137
TError ,
132
138
TData ,
133
- TQueryData
139
+ TQueryData ,
140
+ TQueryKey
134
141
>
135
142
) : InfiniteQueryObserverResult < TData , TError > {
136
143
const { state } = query
0 commit comments