135 lines · 3.9 KB · TypeScript
RawDownload
import { useAtomValue } from "@effect/atom-react";import { WS_METHODS } from "@t3tools/contracts";import { Atom } from "effect/unstable/reactivity"; import { appAtomRegistry } from "./atomRegistry"; export const SLOW_RPC_ACK_THRESHOLD_MS = 15_000;export const MAX_TRACKED_RPC_ACK_REQUESTS = 256;let slowRpcAckThresholdMs = SLOW_RPC_ACK_THRESHOLD_MS; export interface SlowRpcAckRequest {  readonly requestId: string;  readonly startedAt: string;  readonly startedAtMs: number;  readonly tag: string;  readonly thresholdMs: number;} interface PendingRpcAckRequest {  readonly request: SlowRpcAckRequest;  readonly timeoutId: ReturnType<typeof setTimeout>;} const pendingRpcAckRequests = new Map<string, PendingRpcAckRequest>();const untrackedRpcAckTags = new Set<string>([WS_METHODS.previewAutomationConnect]); const slowRpcAckRequestsAtom = Atom.make<ReadonlyArray<SlowRpcAckRequest>>([]).pipe(  Atom.keepAlive,  Atom.withLabel("slow-rpc-ack-requests"),); function setSlowRpcAckRequests(requests: ReadonlyArray<SlowRpcAckRequest>) {  appAtomRegistry.set(slowRpcAckRequestsAtom, [...requests]);} function getSlowRpcAckRequestsValue(): ReadonlyArray<SlowRpcAckRequest> {  return appAtomRegistry.get(slowRpcAckRequestsAtom);} function shouldTrackRpcAck(tag: string): boolean {  return !tag.includes("subscribe") && !untrackedRpcAckTags.has(tag);} export function getSlowRpcAckRequests(): ReadonlyArray<SlowRpcAckRequest> {  return getSlowRpcAckRequestsValue();} export function trackRpcRequestSent(requestId: string, tag: string): void {  if (!shouldTrackRpcAck(tag)) {    return;  }   clearTrackedRpcRequest(requestId);  evictOldestPendingRpcRequestIfNeeded();   const startedAtMs = Date.now();  const request: SlowRpcAckRequest = {    requestId,    startedAt: new Date(startedAtMs).toISOString(),    startedAtMs,    tag,    thresholdMs: slowRpcAckThresholdMs,  };  const timeoutId = setTimeout(() => {    pendingRpcAckRequests.delete(requestId);    appendSlowRpcAckRequest(request);  }, slowRpcAckThresholdMs);   pendingRpcAckRequests.set(requestId, {    request,    timeoutId,  });} export function acknowledgeRpcRequest(requestId: string): void {  clearTrackedRpcRequest(requestId);  const slowRequests = getSlowRpcAckRequestsValue();  if (!slowRequests.some((request) => request.requestId === requestId)) {    return;  }   setSlowRpcAckRequests(slowRequests.filter((request) => request.requestId !== requestId));} export function clearAllTrackedRpcRequests(): void {  for (const pending of pendingRpcAckRequests.values()) {    clearTimeout(pending.timeoutId);  }  pendingRpcAckRequests.clear();  setSlowRpcAckRequests([]);} function clearTrackedRpcRequest(requestId: string): void {  const pending = pendingRpcAckRequests.get(requestId);  if (!pending) {    return;  }   clearTimeout(pending.timeoutId);  pendingRpcAckRequests.delete(requestId);} function appendSlowRpcAckRequest(request: SlowRpcAckRequest): void {  const requests = [...getSlowRpcAckRequestsValue(), request];  if (requests.length <= MAX_TRACKED_RPC_ACK_REQUESTS) {    setSlowRpcAckRequests(requests);    return;  }   setSlowRpcAckRequests(requests.slice(-MAX_TRACKED_RPC_ACK_REQUESTS));} function evictOldestPendingRpcRequestIfNeeded(): void {  while (pendingRpcAckRequests.size >= MAX_TRACKED_RPC_ACK_REQUESTS) {    const oldestRequestId = pendingRpcAckRequests.keys().next().value;    if (oldestRequestId === undefined) {      return;    }     clearTrackedRpcRequest(oldestRequestId);  }} export function resetRequestLatencyStateForTests(): void {  slowRpcAckThresholdMs = SLOW_RPC_ACK_THRESHOLD_MS;  clearAllTrackedRpcRequests();} export function setSlowRpcAckThresholdMsForTests(thresholdMs: number): void {  slowRpcAckThresholdMs = thresholdMs;} export function useSlowRpcAckRequests(): ReadonlyArray<SlowRpcAckRequest> {  return useAtomValue(slowRpcAckRequestsAtom);}