forked from TanStack/query
-
Notifications
You must be signed in to change notification settings - Fork 0
/
QueryClientProvider.tsx
48 lines (38 loc) 路 1.19 KB
/
QueryClientProvider.tsx
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import type { QueryClient } from '@tanstack/query-core'
import type { Context, JSX } from 'solid-js'
import { createContext, useContext, onMount, onCleanup } from 'solid-js'
import type { ContextOptions } from './types'
export const defaultContext = createContext<QueryClient | undefined>(undefined)
function getQueryClientContext(
context: Context<QueryClient | undefined> | undefined,
) {
if (context) {
return context
}
return defaultContext
}
export const useQueryClient = ({ context }: ContextOptions = {}) => {
const queryClient = useContext(getQueryClientContext(context))
if (!queryClient) {
throw new Error('No QueryClient set, use QueryClientProvider to set one')
}
return queryClient
}
export type QueryClientProviderProps = {
client: QueryClient
children?: JSX.Element
} & ContextOptions
export const QueryClientProvider = (
props: QueryClientProviderProps,
): JSX.Element => {
onMount(() => {
props.client.mount()
})
onCleanup(() => props.client.unmount())
const QueryClientContext = getQueryClientContext(props.context)
return (
<QueryClientContext.Provider value={props.client}>
{props.children}
</QueryClientContext.Provider>
)
}