@apollo/client
Version:
A fully-featured caching GraphQL client.
71 lines • 2.95 kB
JavaScript
import { __assign } from "tslib";
import { useCallback, useEffect, useMemo, useRef } from 'react';
import { mergeOptions } from "../../utilities/index.js";
import { useInternalState } from "./useQuery.js";
import { useApolloClient } from "./useApolloClient.js";
var EAGER_METHODS = [
'refetch',
'reobserve',
'fetchMore',
'updateQuery',
'startPolling',
'subscribeToMore',
];
export function useLazyQuery(query, options) {
var _a;
var abortControllersRef = useRef(new Set());
var execOptionsRef = useRef();
var merged = execOptionsRef.current ? mergeOptions(options, execOptionsRef.current) : options;
var internalState = useInternalState(useApolloClient(options && options.client), (_a = merged === null || merged === void 0 ? void 0 : merged.query) !== null && _a !== void 0 ? _a : query);
var useQueryResult = internalState.useQuery(__assign(__assign({}, merged), { skip: !execOptionsRef.current }));
var initialFetchPolicy = useQueryResult.observable.options.initialFetchPolicy ||
internalState.getDefaultFetchPolicy();
var result = Object.assign(useQueryResult, {
called: !!execOptionsRef.current,
});
var eagerMethods = useMemo(function () {
var eagerMethods = {};
var _loop_1 = function (key) {
var method = result[key];
eagerMethods[key] = function () {
if (!execOptionsRef.current) {
execOptionsRef.current = Object.create(null);
internalState.forceUpdate();
}
return method.apply(this, arguments);
};
};
for (var _i = 0, EAGER_METHODS_1 = EAGER_METHODS; _i < EAGER_METHODS_1.length; _i++) {
var key = EAGER_METHODS_1[_i];
_loop_1(key);
}
return eagerMethods;
}, []);
Object.assign(result, eagerMethods);
useEffect(function () {
return function () {
abortControllersRef.current.forEach(function (controller) {
controller.abort();
});
};
}, []);
var execute = useCallback(function (executeOptions) {
var controller = new AbortController();
abortControllersRef.current.add(controller);
execOptionsRef.current = executeOptions ? __assign(__assign({}, executeOptions), { fetchPolicy: executeOptions.fetchPolicy || initialFetchPolicy }) : {
fetchPolicy: initialFetchPolicy,
};
var promise = internalState
.asyncUpdate(controller.signal)
.then(function (queryResult) {
abortControllersRef.current.delete(controller);
return Object.assign(queryResult, eagerMethods);
});
promise.catch(function () {
abortControllersRef.current.delete(controller);
});
return promise;
}, []);
return [execute, result];
}
//# sourceMappingURL=useLazyQuery.js.map