UNPKG

2.95 kBJavaScriptView Raw
1import { __assign } from "tslib";
2import { useCallback, useEffect, useMemo, useRef } from 'react';
3import { mergeOptions } from "../../utilities/index.js";
4import { useInternalState } from "./useQuery.js";
5import { useApolloClient } from "./useApolloClient.js";
6var EAGER_METHODS = [
7 'refetch',
8 'reobserve',
9 'fetchMore',
10 'updateQuery',
11 'startPolling',
12 'subscribeToMore',
13];
14export function useLazyQuery(query, options) {
15 var _a;
16 var abortControllersRef = useRef(new Set());
17 var execOptionsRef = useRef();
18 var merged = execOptionsRef.current ? mergeOptions(options, execOptionsRef.current) : options;
19 var internalState = useInternalState(useApolloClient(options && options.client), (_a = merged === null || merged === void 0 ? void 0 : merged.query) !== null && _a !== void 0 ? _a : query);
20 var useQueryResult = internalState.useQuery(__assign(__assign({}, merged), { skip: !execOptionsRef.current }));
21 var initialFetchPolicy = useQueryResult.observable.options.initialFetchPolicy ||
22 internalState.getDefaultFetchPolicy();
23 var result = Object.assign(useQueryResult, {
24 called: !!execOptionsRef.current,
25 });
26 var eagerMethods = useMemo(function () {
27 var eagerMethods = {};
28 var _loop_1 = function (key) {
29 var method = result[key];
30 eagerMethods[key] = function () {
31 if (!execOptionsRef.current) {
32 execOptionsRef.current = Object.create(null);
33 internalState.forceUpdate();
34 }
35 return method.apply(this, arguments);
36 };
37 };
38 for (var _i = 0, EAGER_METHODS_1 = EAGER_METHODS; _i < EAGER_METHODS_1.length; _i++) {
39 var key = EAGER_METHODS_1[_i];
40 _loop_1(key);
41 }
42 return eagerMethods;
43 }, []);
44 Object.assign(result, eagerMethods);
45 useEffect(function () {
46 return function () {
47 abortControllersRef.current.forEach(function (controller) {
48 controller.abort();
49 });
50 };
51 }, []);
52 var execute = useCallback(function (executeOptions) {
53 var controller = new AbortController();
54 abortControllersRef.current.add(controller);
55 execOptionsRef.current = executeOptions ? __assign(__assign({}, executeOptions), { fetchPolicy: executeOptions.fetchPolicy || initialFetchPolicy }) : {
56 fetchPolicy: initialFetchPolicy,
57 };
58 var promise = internalState
59 .asyncUpdate(controller.signal)
60 .then(function (queryResult) {
61 abortControllersRef.current.delete(controller);
62 return Object.assign(queryResult, eagerMethods);
63 });
64 promise.catch(function () {
65 abortControllersRef.current.delete(controller);
66 });
67 return promise;
68 }, []);
69 return [execute, result];
70}
71//# sourceMappingURL=useLazyQuery.js.map
\No newline at end of file