UNPKG

@tanstack/react-query

Version:

Hooks for managing, caching and syncing asynchronous and remote data in React

72 lines (58 loc) 3.06 kB
import * as React from 'react'; import { useSyncExternalStore } from './useSyncExternalStore.mjs'; import { notifyManager } from '@tanstack/query-core'; import { useQueryErrorResetBoundary } from './QueryErrorResetBoundary.mjs'; import { useQueryClient } from './QueryClientProvider.mjs'; import { useIsRestoring } from './isRestoring.mjs'; import { ensurePreventErrorBoundaryRetry, useClearResetErrorBoundary, getHasError } from './errorBoundaryUtils.mjs'; import { ensureStaleTime, shouldSuspend, fetchOptimistic } from './suspense.mjs'; function useBaseQuery(options, Observer) { const queryClient = useQueryClient({ context: options.context }); const isRestoring = useIsRestoring(); const errorResetBoundary = useQueryErrorResetBoundary(); const defaultedOptions = queryClient.defaultQueryOptions(options); // Make sure results are optimistically set in fetching state before subscribing or updating options defaultedOptions._optimisticResults = isRestoring ? 'isRestoring' : 'optimistic'; // Include callbacks in batch renders if (defaultedOptions.onError) { defaultedOptions.onError = notifyManager.batchCalls(defaultedOptions.onError); } if (defaultedOptions.onSuccess) { defaultedOptions.onSuccess = notifyManager.batchCalls(defaultedOptions.onSuccess); } if (defaultedOptions.onSettled) { defaultedOptions.onSettled = notifyManager.batchCalls(defaultedOptions.onSettled); } ensureStaleTime(defaultedOptions); ensurePreventErrorBoundaryRetry(defaultedOptions, errorResetBoundary); useClearResetErrorBoundary(errorResetBoundary); const [observer] = React.useState(() => new Observer(queryClient, defaultedOptions)); const result = observer.getOptimisticResult(defaultedOptions); useSyncExternalStore(React.useCallback(onStoreChange => { const unsubscribe = isRestoring ? () => undefined : observer.subscribe(notifyManager.batchCalls(onStoreChange)); // Update result to make sure we did not miss any query updates // between creating the observer and subscribing to it. observer.updateResult(); return unsubscribe; }, [observer, isRestoring]), () => observer.getCurrentResult(), () => observer.getCurrentResult()); React.useEffect(() => { // Do not notify on updates because of changes in the options because // these changes should already be reflected in the optimistic result. observer.setOptions(defaultedOptions, { listeners: false }); }, [defaultedOptions, observer]); // Handle suspense if (shouldSuspend(defaultedOptions, result, isRestoring)) { throw fetchOptimistic(defaultedOptions, observer, errorResetBoundary); } // Handle error boundary if (getHasError({ result, errorResetBoundary, useErrorBoundary: defaultedOptions.useErrorBoundary, query: observer.getCurrentQuery() })) { throw result.error; } // Handle result property usage tracking return !defaultedOptions.notifyOnChangeProps ? observer.trackResult(result) : result; } export { useBaseQuery }; //# sourceMappingURL=useBaseQuery.mjs.map