refactor: inline timing map writes

Author: Tarquinen

lib/compress/timing.ts

@@ -18,29 +18,12 @@ export function createCompressionTimingState(): CompressionTimingState {
     }
 }
 
-export function recordCompressionStart(
-    state: SessionState,
-    callId: string,
-    startedAt: number,
-): boolean {
-    if (state.compressionTiming.startsByCallId.has(callId)) {
-        return false
-    }
-
-    state.compressionTiming.startsByCallId.set(callId, startedAt)
-    return true
-}
-
 export function consumeCompressionStart(state: SessionState, callId: string): number | undefined {
     const start = state.compressionTiming.startsByCallId.get(callId)
     state.compressionTiming.startsByCallId.delete(callId)
     return start
 }
 
-export function clearCompressionStart(state: SessionState, callId: string): void {
-    state.compressionTiming.startsByCallId.delete(callId)
-}
-
 export function resolveCompressionDuration(
     startedAt: number | undefined,
     eventTime: number | undefined,
@@ -68,14 +51,6 @@ export function resolveCompressionDuration(
     return typeof pendingToRunningMs === "number" ? pendingToRunningMs : runtimeMs
 }
 
-export function queueCompressionDuration(
-    state: SessionState,
-    callId: string,
-    durationMs: number,
-): void {
-    state.compressionTiming.pendingByCallId.set(callId, { callId, durationMs })
-}
-
 export function applyPendingCompressionDurations(state: SessionState): number {
     if (state.compressionTiming.pendingByCallId.size === 0) {
         return 0

lib/hooks.ts

@@ -18,10 +18,7 @@ import { renderSystemPrompt, type PromptStore } from "./prompts"
 import { buildProtectedToolsExtension } from "./prompts/extensions/system"
 import {
     applyPendingCompressionDurations,
-    clearCompressionStart,
     consumeCompressionStart,
-    queueCompressionDuration,
-    recordCompressionStart,
     resolveCompressionDuration,
 } from "./compress/timing"
 import {
@@ -299,9 +296,10 @@ export function createEventHandler(state: SessionState, logger: Logger) {
             }
 
             const startedAt = eventTime ?? Date.now()
-            if (!recordCompressionStart(state, part.callID, startedAt)) {
+            if (state.compressionTiming.startsByCallId.has(part.callID)) {
                 return
             }
+            state.compressionTiming.startsByCallId.set(part.callID, startedAt)
             logger.debug("Recorded compression start", {
                 callID: part.callID,
                 startedAt,
@@ -320,7 +318,10 @@ export function createEventHandler(state: SessionState, logger: Logger) {
                 return
             }
 
-            queueCompressionDuration(state, part.callID, durationMs)
+            state.compressionTiming.pendingByCallId.set(part.callID, {
+                callId: part.callID,
+                durationMs,
+            })
 
             const updates = applyPendingCompressionDurations(state)
             if (updates === 0) {
@@ -342,7 +343,7 @@ export function createEventHandler(state: SessionState, logger: Logger) {
         }
 
         if (typeof part.callID === "string") {
-            clearCompressionStart(state, part.callID)
+            state.compressionTiming.startsByCallId.delete(part.callID)
         }
     }
 }