Render sending gRPC events

This commit is contained in:
Gregory Schier
2024-10-17 12:03:35 -07:00
parent c652df82a3
commit 0cd08499aa
2 changed files with 280 additions and 497 deletions

File diff suppressed because it is too large Load Diff

View File

@@ -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,
}; };
} }