mirror of
https://github.com/hanzoai/agency.git
synced 2026-11-02 19:13:28 +00:00
305 lines
9.3 KiB
Plaintext
305 lines
9.3 KiB
Plaintext
// src/queryClient.ts
|
|
import {
|
|
functionalUpdate,
|
|
hashKey,
|
|
hashQueryKeyByOptions,
|
|
noop,
|
|
partialMatchKey,
|
|
resolveStaleTime,
|
|
skipToken
|
|
} from "./utils.js";
|
|
import { QueryCache } from "./queryCache.js";
|
|
import { MutationCache } from "./mutationCache.js";
|
|
import { focusManager } from "./focusManager.js";
|
|
import { onlineManager } from "./onlineManager.js";
|
|
import { notifyManager } from "./notifyManager.js";
|
|
import { infiniteQueryBehavior } from "./infiniteQueryBehavior.js";
|
|
var QueryClient = class {
|
|
#queryCache;
|
|
#mutationCache;
|
|
#defaultOptions;
|
|
#queryDefaults;
|
|
#mutationDefaults;
|
|
#mountCount;
|
|
#unsubscribeFocus;
|
|
#unsubscribeOnline;
|
|
constructor(config = {}) {
|
|
this.#queryCache = config.queryCache || new QueryCache();
|
|
this.#mutationCache = config.mutationCache || new MutationCache();
|
|
this.#defaultOptions = config.defaultOptions || {};
|
|
this.#queryDefaults = /* @__PURE__ */ new Map();
|
|
this.#mutationDefaults = /* @__PURE__ */ new Map();
|
|
this.#mountCount = 0;
|
|
}
|
|
mount() {
|
|
this.#mountCount++;
|
|
if (this.#mountCount !== 1) return;
|
|
this.#unsubscribeFocus = focusManager.subscribe(async (focused) => {
|
|
if (focused) {
|
|
await this.resumePausedMutations();
|
|
this.#queryCache.onFocus();
|
|
}
|
|
});
|
|
this.#unsubscribeOnline = onlineManager.subscribe(async (online) => {
|
|
if (online) {
|
|
await this.resumePausedMutations();
|
|
this.#queryCache.onOnline();
|
|
}
|
|
});
|
|
}
|
|
unmount() {
|
|
this.#mountCount--;
|
|
if (this.#mountCount !== 0) return;
|
|
this.#unsubscribeFocus?.();
|
|
this.#unsubscribeFocus = void 0;
|
|
this.#unsubscribeOnline?.();
|
|
this.#unsubscribeOnline = void 0;
|
|
}
|
|
isFetching(filters) {
|
|
return this.#queryCache.findAll({ ...filters, fetchStatus: "fetching" }).length;
|
|
}
|
|
isMutating(filters) {
|
|
return this.#mutationCache.findAll({ ...filters, status: "pending" }).length;
|
|
}
|
|
/**
|
|
* Imperative (non-reactive) way to retrieve data for a QueryKey.
|
|
* Should only be used in callbacks or functions where reading the latest data is necessary, e.g. for optimistic updates.
|
|
*
|
|
* Hint: Do not use this function inside a component, because it won't receive updates.
|
|
* Use `useQuery` to create a `QueryObserver` that subscribes to changes.
|
|
*/
|
|
getQueryData(queryKey) {
|
|
const options = this.defaultQueryOptions({ queryKey });
|
|
return this.#queryCache.get(options.queryHash)?.state.data;
|
|
}
|
|
ensureQueryData(options) {
|
|
const defaultedOptions = this.defaultQueryOptions(options);
|
|
const query = this.#queryCache.build(this, defaultedOptions);
|
|
const cachedData = query.state.data;
|
|
if (cachedData === void 0) {
|
|
return this.fetchQuery(options);
|
|
}
|
|
if (options.revalidateIfStale && query.isStaleByTime(resolveStaleTime(defaultedOptions.staleTime, query))) {
|
|
void this.prefetchQuery(defaultedOptions);
|
|
}
|
|
return Promise.resolve(cachedData);
|
|
}
|
|
getQueriesData(filters) {
|
|
return this.#queryCache.findAll(filters).map(({ queryKey, state }) => {
|
|
const data = state.data;
|
|
return [queryKey, data];
|
|
});
|
|
}
|
|
setQueryData(queryKey, updater, options) {
|
|
const defaultedOptions = this.defaultQueryOptions({ queryKey });
|
|
const query = this.#queryCache.get(
|
|
defaultedOptions.queryHash
|
|
);
|
|
const prevData = query?.state.data;
|
|
const data = functionalUpdate(updater, prevData);
|
|
if (data === void 0) {
|
|
return void 0;
|
|
}
|
|
return this.#queryCache.build(this, defaultedOptions).setData(data, { ...options, manual: true });
|
|
}
|
|
setQueriesData(filters, updater, options) {
|
|
return notifyManager.batch(
|
|
() => this.#queryCache.findAll(filters).map(({ queryKey }) => [
|
|
queryKey,
|
|
this.setQueryData(queryKey, updater, options)
|
|
])
|
|
);
|
|
}
|
|
getQueryState(queryKey) {
|
|
const options = this.defaultQueryOptions({ queryKey });
|
|
return this.#queryCache.get(
|
|
options.queryHash
|
|
)?.state;
|
|
}
|
|
removeQueries(filters) {
|
|
const queryCache = this.#queryCache;
|
|
notifyManager.batch(() => {
|
|
queryCache.findAll(filters).forEach((query) => {
|
|
queryCache.remove(query);
|
|
});
|
|
});
|
|
}
|
|
resetQueries(filters, options) {
|
|
const queryCache = this.#queryCache;
|
|
return notifyManager.batch(() => {
|
|
queryCache.findAll(filters).forEach((query) => {
|
|
query.reset();
|
|
});
|
|
return this.refetchQueries(
|
|
{
|
|
type: "active",
|
|
...filters
|
|
},
|
|
options
|
|
);
|
|
});
|
|
}
|
|
cancelQueries(filters, cancelOptions = {}) {
|
|
const defaultedCancelOptions = { revert: true, ...cancelOptions };
|
|
const promises = notifyManager.batch(
|
|
() => this.#queryCache.findAll(filters).map((query) => query.cancel(defaultedCancelOptions))
|
|
);
|
|
return Promise.all(promises).then(noop).catch(noop);
|
|
}
|
|
invalidateQueries(filters, options = {}) {
|
|
return notifyManager.batch(() => {
|
|
this.#queryCache.findAll(filters).forEach((query) => {
|
|
query.invalidate();
|
|
});
|
|
if (filters?.refetchType === "none") {
|
|
return Promise.resolve();
|
|
}
|
|
return this.refetchQueries(
|
|
{
|
|
...filters,
|
|
type: filters?.refetchType ?? filters?.type ?? "active"
|
|
},
|
|
options
|
|
);
|
|
});
|
|
}
|
|
refetchQueries(filters, options = {}) {
|
|
const fetchOptions = {
|
|
...options,
|
|
cancelRefetch: options.cancelRefetch ?? true
|
|
};
|
|
const promises = notifyManager.batch(
|
|
() => this.#queryCache.findAll(filters).filter((query) => !query.isDisabled() && !query.isStatic()).map((query) => {
|
|
let promise = query.fetch(void 0, fetchOptions);
|
|
if (!fetchOptions.throwOnError) {
|
|
promise = promise.catch(noop);
|
|
}
|
|
return query.state.fetchStatus === "paused" ? Promise.resolve() : promise;
|
|
})
|
|
);
|
|
return Promise.all(promises).then(noop);
|
|
}
|
|
fetchQuery(options) {
|
|
const defaultedOptions = this.defaultQueryOptions(options);
|
|
if (defaultedOptions.retry === void 0) {
|
|
defaultedOptions.retry = false;
|
|
}
|
|
const query = this.#queryCache.build(this, defaultedOptions);
|
|
return query.isStaleByTime(
|
|
resolveStaleTime(defaultedOptions.staleTime, query)
|
|
) ? query.fetch(defaultedOptions) : Promise.resolve(query.state.data);
|
|
}
|
|
prefetchQuery(options) {
|
|
return this.fetchQuery(options).then(noop).catch(noop);
|
|
}
|
|
fetchInfiniteQuery(options) {
|
|
options.behavior = infiniteQueryBehavior(options.pages);
|
|
return this.fetchQuery(options);
|
|
}
|
|
prefetchInfiniteQuery(options) {
|
|
return this.fetchInfiniteQuery(options).then(noop).catch(noop);
|
|
}
|
|
ensureInfiniteQueryData(options) {
|
|
options.behavior = infiniteQueryBehavior(options.pages);
|
|
return this.ensureQueryData(options);
|
|
}
|
|
resumePausedMutations() {
|
|
if (onlineManager.isOnline()) {
|
|
return this.#mutationCache.resumePausedMutations();
|
|
}
|
|
return Promise.resolve();
|
|
}
|
|
getQueryCache() {
|
|
return this.#queryCache;
|
|
}
|
|
getMutationCache() {
|
|
return this.#mutationCache;
|
|
}
|
|
getDefaultOptions() {
|
|
return this.#defaultOptions;
|
|
}
|
|
setDefaultOptions(options) {
|
|
this.#defaultOptions = options;
|
|
}
|
|
setQueryDefaults(queryKey, options) {
|
|
this.#queryDefaults.set(hashKey(queryKey), {
|
|
queryKey,
|
|
defaultOptions: options
|
|
});
|
|
}
|
|
getQueryDefaults(queryKey) {
|
|
const defaults = [...this.#queryDefaults.values()];
|
|
const result = {};
|
|
defaults.forEach((queryDefault) => {
|
|
if (partialMatchKey(queryKey, queryDefault.queryKey)) {
|
|
Object.assign(result, queryDefault.defaultOptions);
|
|
}
|
|
});
|
|
return result;
|
|
}
|
|
setMutationDefaults(mutationKey, options) {
|
|
this.#mutationDefaults.set(hashKey(mutationKey), {
|
|
mutationKey,
|
|
defaultOptions: options
|
|
});
|
|
}
|
|
getMutationDefaults(mutationKey) {
|
|
const defaults = [...this.#mutationDefaults.values()];
|
|
const result = {};
|
|
defaults.forEach((queryDefault) => {
|
|
if (partialMatchKey(mutationKey, queryDefault.mutationKey)) {
|
|
Object.assign(result, queryDefault.defaultOptions);
|
|
}
|
|
});
|
|
return result;
|
|
}
|
|
defaultQueryOptions(options) {
|
|
if (options._defaulted) {
|
|
return options;
|
|
}
|
|
const defaultedOptions = {
|
|
...this.#defaultOptions.queries,
|
|
...this.getQueryDefaults(options.queryKey),
|
|
...options,
|
|
_defaulted: true
|
|
};
|
|
if (!defaultedOptions.queryHash) {
|
|
defaultedOptions.queryHash = hashQueryKeyByOptions(
|
|
defaultedOptions.queryKey,
|
|
defaultedOptions
|
|
);
|
|
}
|
|
if (defaultedOptions.refetchOnReconnect === void 0) {
|
|
defaultedOptions.refetchOnReconnect = defaultedOptions.networkMode !== "always";
|
|
}
|
|
if (defaultedOptions.throwOnError === void 0) {
|
|
defaultedOptions.throwOnError = !!defaultedOptions.suspense;
|
|
}
|
|
if (!defaultedOptions.networkMode && defaultedOptions.persister) {
|
|
defaultedOptions.networkMode = "offlineFirst";
|
|
}
|
|
if (defaultedOptions.queryFn === skipToken) {
|
|
defaultedOptions.enabled = false;
|
|
}
|
|
return defaultedOptions;
|
|
}
|
|
defaultMutationOptions(options) {
|
|
if (options?._defaulted) {
|
|
return options;
|
|
}
|
|
return {
|
|
...this.#defaultOptions.mutations,
|
|
...options?.mutationKey && this.getMutationDefaults(options.mutationKey),
|
|
...options,
|
|
_defaulted: true
|
|
};
|
|
}
|
|
clear() {
|
|
this.#queryCache.clear();
|
|
this.#mutationCache.clear();
|
|
}
|
|
};
|
|
export {
|
|
QueryClient
|
|
};
|
|
//# sourceMappingURL=queryClient.js.map |