YeonGyu-Kim
3e06d280a9
Guarding polling re-entry avoids stacked async polls under slow responses, and unref on pending-call cleanup timer reduces idle wakeups.
37 lines
1.0 KiB
TypeScript
37 lines
1.0 KiB
TypeScript
import type { PendingCall } from "./types"
|
|
|
|
const pendingCalls = new Map<string, PendingCall>()
|
|
const PENDING_CALL_TTL = 60_000
|
|
|
|
let cleanupIntervalStarted = false
|
|
let cleanupInterval: ReturnType<typeof setInterval> | undefined
|
|
|
|
function cleanupOldPendingCalls(): void {
|
|
const now = Date.now()
|
|
for (const [callID, call] of pendingCalls) {
|
|
if (now - call.timestamp > PENDING_CALL_TTL) {
|
|
pendingCalls.delete(callID)
|
|
}
|
|
}
|
|
}
|
|
|
|
export function startPendingCallCleanup(): void {
|
|
if (cleanupIntervalStarted) return
|
|
cleanupIntervalStarted = true
|
|
cleanupInterval = setInterval(cleanupOldPendingCalls, 10_000)
|
|
if (typeof cleanupInterval === "object" && "unref" in cleanupInterval) {
|
|
cleanupInterval.unref()
|
|
}
|
|
}
|
|
|
|
export function registerPendingCall(callID: string, pendingCall: PendingCall): void {
|
|
pendingCalls.set(callID, pendingCall)
|
|
}
|
|
|
|
export function takePendingCall(callID: string): PendingCall | undefined {
|
|
const pendingCall = pendingCalls.get(callID)
|
|
if (!pendingCall) return undefined
|
|
pendingCalls.delete(callID)
|
|
return pendingCall
|
|
}
|