46 lines
919 B
TypeScript
46 lines
919 B
TypeScript
|
'use client'
|
||
|
|
import * as React from 'react'
|
||
|
|
|
||
|
|
import type { QueryClient } from '@tanstack/query-core'
|
||
|
|
|
||
|
|
export const QueryClientContext = React.createContext<QueryClient | undefined>(
|
||
|
|
undefined,
|
||
|
|
)
|
||
|
|
|
||
|
|
export const useQueryClient = (queryClient?: QueryClient) => {
|
||
|
|
const client = React.useContext(QueryClientContext)
|
||
|
|
|
||
|
|
if (queryClient) {
|
||
|
|
return queryClient
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!client) {
|
||
|
|
throw new Error('No QueryClient set, use QueryClientProvider to set one')
|
||
|
|
}
|
||
|
|
|
||
|
|
return client
|
||
|
|
}
|
||
|
|
|
||
|
|
export type QueryClientProviderProps = {
|
||
|
|
client: QueryClient
|
||
|
|
children?: React.ReactNode
|
||
|
|
}
|
||
|
|
|
||
|
|
export const QueryClientProvider = ({
|
||
|
|
client,
|
||
|
|
children,
|
||
|
|
}: QueryClientProviderProps): React.JSX.Element => {
|
||
|
|
React.useEffect(() => {
|
||
|
|
client.mount()
|
||
|
|
return () => {
|
||
|
|
client.unmount()
|
||
|
|
}
|
||
|
|
}, [client])
|
||
|
|
|
||
|
|
return (
|
||
|
|
<QueryClientContext.Provider value={client}>
|
||
|
|
{children}
|
||
|
|
</QueryClientContext.Provider>
|
||
|
|
)
|
||
|
|
}
|