ceos/opencode
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixes

Browse source
This commit is contained in:
Aiden Cline 2026-02-20 01:17:08 -06:00
commit cfe2d30a26
1 changed files with 14 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

32
packages/opencode/src/cli/cmd/stats.ts
View file

@ -197,12 +197,14 @@ export async function aggregateSessionStats(days?: number, projectFilter?: strin
const sessionIds = batch.map((s) => s.id) const sessionIds = batch.map((s) => s.id)
// Bulk fetch messages for this batch of sessions // Bulk fetch messages for this batch of sessions
const messageRows = Database.use((db) => db.select().from(MessageTable).where(inArray(MessageTable.session_id, sessionIds)).all()) const messageRows = Database.use((db) =>
db.select().from(MessageTable).where(inArray(MessageTable.session_id, sessionIds)).all(),
)
// Group messages by session_id // Group messages by session_id
const messagesBySession = new Map<string, typeof messageRows>() const messagesBySession = new Map<string, typeof messageRows>()
const messageIds = messageRows.map((r) => r.id) const messageIds = messageRows.map((r) => r.id)
for (const row of messageRows) { for (const row of messageRows) {
const msgs = messagesBySession.get(row.session_id) || [] const msgs = messagesBySession.get(row.session_id) || []
msgs.push(row) msgs.push(row)
@ -210,13 +212,15 @@ export async function aggregateSessionStats(days?: number, projectFilter?: strin
} }
// Bulk fetch parts for all these messages // Bulk fetch parts for all these messages
let partRows: typeof PartTable.$inferSelect[] = [] let partRows: (typeof PartTable.$inferSelect)[] = []
if (messageIds.length > 0) { if (messageIds.length > 0) {
// Chunk message IDs if there are too many for a single IN clause (SQLite has limits) // Chunk message IDs if there are too many for a single IN clause (SQLite has limits)
const PART_BATCH_SIZE = 500 const PART_BATCH_SIZE = 500
for (let j = 0; j < messageIds.length; j += PART_BATCH_SIZE) { for (let j = 0; j < messageIds.length; j += PART_BATCH_SIZE) {
const idBatch = messageIds.slice(j, j + PART_BATCH_SIZE) const idBatch = messageIds.slice(j, j + PART_BATCH_SIZE)
const parts = Database.use((db) => db.select().from(PartTable).where(inArray(PartTable.message_id, idBatch)).all()) const parts = Database.use((db) =>
db.select().from(PartTable).where(inArray(PartTable.message_id, idBatch)).all(),
)
partRows.push(...parts) partRows.push(...parts)
} }
} }
@ -229,18 +233,12 @@ export async function aggregateSessionStats(days?: number, projectFilter?: strin
partsByMessage.set(row.message_id, parts) partsByMessage.set(row.message_id, parts)
} }
for (const session of batch) { const batchResults = batch.map((session) => {
const rawMessages = messagesBySession.get(session.id) || [] const rawMessages = messagesBySession.get(session.id) || []
// Construct the MessageV2 objects locally instead of doing another N queries const messages = rawMessages.map((row) => ({
const messages = rawMessages.map( info: { ...row.data, id: row.id, sessionID: row.session_id } as MessageV2.Info,
(row) => parts: partsByMessage.get(row.id) || [],
({ }))
id: row.id,
sessionID: row.session_id,
info: row.data,
parts: partsByMessage.get(row.id) || [],
}) as MessageV2.WithParts,
)
let sessionCost = 0 let sessionCost = 0
let sessionTokens = { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } } let sessionTokens = { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } }
@ -328,8 +326,6 @@ export async function aggregateSessionStats(days?: number, projectFilter?: strin
} }
}) })
const batchResults = await Promise.all(batchPromises)
for (const result of batchResults) { for (const result of batchResults) {
earliestTime = Math.min(earliestTime, result.earliestTime) earliestTime = Math.min(earliestTime, result.earliestTime)
latestTime = Math.max(latestTime, result.latestTime) latestTime = Math.max(latestTime, result.latestTime)