github/openclaw
1
1
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-04-25 03:04:29 -04:00
Code Issues Packages Projects Releases Wiki Activity

refactor(gateway): share session store lookup map builder

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-18 22:40:00 +00:00
parent b4cba304e2
commit 5e76cefc70
1 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/gateway/server-methods/usage.ts
View File

@@ -154,6 +154,18 @@ const parseDateRange = (params: {
type DiscoveredSessionWithAgent = DiscoveredSession & { agentId: string }; type DiscoveredSessionWithAgent = DiscoveredSession & { agentId: string };
function buildStoreBySessionId(
store: Record<string, SessionEntry>,
): Map<string, { key: string; entry: SessionEntry }> {
const storeBySessionId = new Map<string, { key: string; entry: SessionEntry }>();
for (const [key, entry] of Object.entries(store)) {
if (entry?.sessionId) {
storeBySessionId.set(entry.sessionId, { key, entry });
}
}
return storeBySessionId;
}
async function discoverAllSessionsForUsage(params: { async function discoverAllSessionsForUsage(params: {
config: ReturnType<typeof loadConfig>; config: ReturnType<typeof loadConfig>;
startMs: number; startMs: number;
@@ -353,12 +365,7 @@ export const usageHandlers: GatewayRequestHandlers = {
// Prefer the store entry when available, even if the caller provides a discovered key // Prefer the store entry when available, even if the caller provides a discovered key
// (`agent:<id>:<sessionId>`) for a session that now has a canonical store key. // (`agent:<id>:<sessionId>`) for a session that now has a canonical store key.
const storeBySessionId = new Map<string, { key: string; entry: SessionEntry }>(); const storeBySessionId = buildStoreBySessionId(store);
for (const [key, entry] of Object.entries(store)) {
if (entry?.sessionId) {
storeBySessionId.set(entry.sessionId, { key, entry });
}
}
const storeMatch = store[specificKey] const storeMatch = store[specificKey]
? { key: specificKey, entry: store[specificKey] } ? { key: specificKey, entry: store[specificKey] }
@@ -409,12 +416,7 @@ export const usageHandlers: GatewayRequestHandlers = {
}); });
// Build a map of sessionId -> store entry for quick lookup // Build a map of sessionId -> store entry for quick lookup
const storeBySessionId = new Map<string, { key: string; entry: SessionEntry }>(); const storeBySessionId = buildStoreBySessionId(store);
for (const [key, entry] of Object.entries(store)) {
if (entry?.sessionId) {
storeBySessionId.set(entry.sessionId, { key, entry });
}
}
for (const discovered of discoveredSessions) { for (const discovered of discoveredSessions) {
const storeMatch = storeBySessionId.get(discovered.sessionId); const storeMatch = storeBySessionId.get(discovered.sessionId);