From 877ab6b0b2e5f7d830502768cc6cef997e09ec8d Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Fri, 21 Aug 2026 19:04:24 -0700
Subject: [PATCH] perf(gateway): bound health recent-session projection
 (#127744)

Amp-Thread-ID: https://ampcode.com/threads/T-01a02525-2a28-740a-b408-ee5382998a8c

Co-authored-by: Amp <amp@ampcode.com>
---
 src/gateway/health/collector.ts | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/src/gateway/health/collector.ts b/src/gateway/health/collector.ts
index 7b642215316f..c48bb2ee7fc7 100644
--- a/src/gateway/health/collector.ts
+++ b/src/gateway/health/collector.ts
@@ -44,6 +44,7 @@ import type {
 } from "./types.js";
 
 const DEFAULT_HEALTH_TIMEOUT_MS = 10_000;
+const HEALTH_RECENT_SESSION_LIMIT = 5;
 const healthLog = createSubsystemLogger("health");
 
 type HealthSnapshotAudience = "public" | "admin";
@@ -124,18 +125,36 @@ export async function buildHealthSessionSummary(storePath: string, agentId?: str
     // Health is best-effort: an empty snapshot beats failing on a transient lock.
     listed = [];
   }
-  const sessions = listed
-    .filter(({ sessionKey }) => sessionKey !== "global" && sessionKey !== "unknown")
-    .map(({ sessionKey, entry }) => ({ key: sessionKey, updatedAt: entry?.updatedAt ?? 0 }))
-    .toSorted((a, b) => b.updatedAt - a.updatedAt);
-  const recent = sessions.slice(0, 5).map((session) => ({
+  const recentSessions: Array<{ key: string; updatedAt: number }> = [];
+  let sessionCount = 0;
+  for (const { sessionKey, entry } of listed) {
+    if (sessionKey === "global" || sessionKey === "unknown") {
+      continue;
+    }
+    sessionCount += 1;
+    const session = { key: sessionKey, updatedAt: entry?.updatedAt ?? 0 };
+    const insertAt = recentSessions.findIndex(
+      (recentSession) => session.updatedAt > recentSession.updatedAt,
+    );
+    // Health returns only five rows. Keep the projection bounded while scanning
+    // so refreshes never sort the complete session snapshot.
+    if (insertAt >= 0) {
+      recentSessions.splice(insertAt, 0, session);
+      if (recentSessions.length > HEALTH_RECENT_SESSION_LIMIT) {
+        recentSessions.pop();
+      }
+    } else if (recentSessions.length < HEALTH_RECENT_SESSION_LIMIT) {
+      recentSessions.push(session);
+    }
+  }
+  const recent = recentSessions.map((session) => ({
     key: session.key,
     updatedAt: session.updatedAt || null,
     age: session.updatedAt ? Date.now() - session.updatedAt : null,
   }));
   return {
     path: databasePath,
-    count: sessions.length,
+    count: sessionCount,
     recent,
   } satisfies HealthSummary["sessions"];
 }