UNPKG

@connectrpc/connect-query

Version:

TypeScript-first expansion pack for TanStack Query that gives you Protobuf superpowers.

30 lines (29 loc) 1.29 kB
// 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 "@connectrpc/connect-query-core"; import { useMutation as tsUseMutation } from "@tanstack/react-query"; import { useCallback } from "react"; import { useTransport } from "./use-transport.js"; /** * Query the method provided. Maps to useMutation on tanstack/react-query */ export function useMutation(schema, { transport, ...queryOptions } = {}) { const transportFromCtx = useTransport(); const transportToUse = transport ?? transportFromCtx; const mutationFn = useCallback(async (input) => callUnaryMethod(transportToUse, schema, input), [transportToUse, schema]); return tsUseMutation({ ...queryOptions, mutationFn, }); }