mirror of
https://github.com/mountain-loop/yaak.git
synced 2026-04-22 08:38:29 +02:00
Render sending gRPC events
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -3,6 +3,7 @@ import { emit } from '@tauri-apps/api/event';
|
|||||||
import type { GrpcConnection, GrpcRequest } from '@yaakapp-internal/models';
|
import type { GrpcConnection, GrpcRequest } from '@yaakapp-internal/models';
|
||||||
import { trackEvent } from '../lib/analytics';
|
import { trackEvent } from '../lib/analytics';
|
||||||
import { minPromiseMillis } from '../lib/minPromiseMillis';
|
import { minPromiseMillis } from '../lib/minPromiseMillis';
|
||||||
|
import { isResponseLoading } from '../lib/model_util';
|
||||||
import { invokeCmd } from '../lib/tauri';
|
import { invokeCmd } from '../lib/tauri';
|
||||||
import { useActiveEnvironment } from './useActiveEnvironment';
|
import { useActiveEnvironment } from './useActiveEnvironment';
|
||||||
import { useDebouncedValue } from './useDebouncedValue';
|
import { useDebouncedValue } from './useDebouncedValue';
|
||||||
@@ -22,27 +23,27 @@ export function useGrpc(
|
|||||||
|
|
||||||
const go = useMutation<void, string>({
|
const go = useMutation<void, string>({
|
||||||
mutationKey: ['grpc_go', conn?.id],
|
mutationKey: ['grpc_go', conn?.id],
|
||||||
mutationFn: async () =>
|
mutationFn: () =>
|
||||||
await invokeCmd('cmd_grpc_go', { requestId, environmentId: environment?.id, protoFiles }),
|
invokeCmd<void>('cmd_grpc_go', { requestId, environmentId: environment?.id, protoFiles }),
|
||||||
onSettled: () => trackEvent('grpc_request', 'send'),
|
onSettled: () => trackEvent('grpc_request', 'send'),
|
||||||
});
|
});
|
||||||
|
|
||||||
const send = useMutation({
|
const send = useMutation({
|
||||||
mutationKey: ['grpc_send', conn?.id],
|
mutationKey: ['grpc_send', conn?.id],
|
||||||
mutationFn: async ({ message }: { message: string }) =>
|
mutationFn: ({ message }: { message: string }) =>
|
||||||
await emit(`grpc_client_msg_${conn?.id ?? 'none'}`, { Message: message }),
|
emit(`grpc_client_msg_${conn?.id ?? 'none'}`, { Message: message }),
|
||||||
onSettled: () => trackEvent('grpc_connection', 'send'),
|
onSettled: () => trackEvent('grpc_connection', 'send'),
|
||||||
});
|
});
|
||||||
|
|
||||||
const cancel = useMutation({
|
const cancel = useMutation({
|
||||||
mutationKey: ['grpc_cancel', conn?.id ?? 'n/a'],
|
mutationKey: ['grpc_cancel', conn?.id ?? 'n/a'],
|
||||||
mutationFn: async () => await emit(`grpc_client_msg_${conn?.id ?? 'none'}`, 'Cancel'),
|
mutationFn: () => emit(`grpc_client_msg_${conn?.id ?? 'none'}`, 'Cancel'),
|
||||||
onSettled: () => trackEvent('grpc_connection', 'cancel'),
|
onSettled: () => trackEvent('grpc_connection', 'cancel'),
|
||||||
});
|
});
|
||||||
|
|
||||||
const commit = useMutation({
|
const commit = useMutation({
|
||||||
mutationKey: ['grpc_commit', conn?.id ?? 'n/a'],
|
mutationKey: ['grpc_commit', conn?.id ?? 'n/a'],
|
||||||
mutationFn: async () => await emit(`grpc_client_msg_${conn?.id ?? 'none'}`, 'Commit'),
|
mutationFn: () => emit(`grpc_client_msg_${conn?.id ?? 'none'}`, 'Commit'),
|
||||||
onSettled: () => trackEvent('grpc_connection', 'commit'),
|
onSettled: () => trackEvent('grpc_connection', 'commit'),
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -51,11 +52,11 @@ export function useGrpc(
|
|||||||
const reflect = useQuery<ReflectResponseService[], string>({
|
const reflect = useQuery<ReflectResponseService[], string>({
|
||||||
enabled: req != null,
|
enabled: req != null,
|
||||||
queryKey: ['grpc_reflect', req?.id ?? 'n/a', debouncedUrl, protoFiles],
|
queryKey: ['grpc_reflect', req?.id ?? 'n/a', debouncedUrl, protoFiles],
|
||||||
queryFn: async () =>
|
queryFn: () =>
|
||||||
(await minPromiseMillis(
|
minPromiseMillis<ReflectResponseService[]>(
|
||||||
invokeCmd('cmd_grpc_reflect', { requestId, protoFiles }),
|
invokeCmd('cmd_grpc_reflect', { requestId, protoFiles }),
|
||||||
300,
|
300,
|
||||||
)) as ReflectResponseService[],
|
),
|
||||||
});
|
});
|
||||||
|
|
||||||
return {
|
return {
|
||||||
@@ -63,7 +64,7 @@ export function useGrpc(
|
|||||||
reflect,
|
reflect,
|
||||||
cancel,
|
cancel,
|
||||||
commit,
|
commit,
|
||||||
isStreaming: conn != null && conn.elapsed === 0,
|
isStreaming: isResponseLoading(conn),
|
||||||
send,
|
send,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user