@connectrpc/connect-query
Version:
TypeScript-first expansion pack for TanStack Query that gives you Protobuf superpowers.
61 lines • 2.31 kB
JavaScript
// Copyright 2021-2023 The Connect Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
import { callUnaryMethod } from "./call-unary-method.js";
import { createConnectInfiniteQueryKey, } from "./connect-query-key.js";
import { assert, disableQuery } from "./utils.js";
function createUnaryInfiniteQueryFn(methodType, input, { callOptions, transport, pageParamKey, }) {
return async (context) => {
assert(input !== disableQuery, "Disabled query cannot be fetched");
assert("pageParam" in context, "pageParam must be part of context");
const inputCombinedWithPageParam = {
...input,
[pageParamKey]: context.pageParam,
};
return callUnaryMethod(methodType, inputCombinedWithPageParam, {
callOptions: {
...callOptions,
signal: callOptions?.signal ?? context.signal,
},
transport,
});
};
}
/**
* Query the method provided. Maps to useInfiniteQuery on tanstack/react-query
*
* @param methodSig
* @returns
*/
export function createUseInfiniteQueryOptions(methodSig, input, { transport, getNextPageParam, pageParamKey, callOptions, }) {
const queryKey = createConnectInfiniteQueryKey(methodSig, input === disableQuery
? undefined
: {
...input,
[pageParamKey]: undefined,
});
return {
getNextPageParam,
initialPageParam: input === disableQuery
? undefined
: input[pageParamKey],
queryKey,
queryFn: createUnaryInfiniteQueryFn(methodSig, input, {
transport,
callOptions,
pageParamKey,
}),
enabled: input !== disableQuery,
};
}
//# sourceMappingURL=create-use-infinite-query-options.js.map