Reduce log I/O and fix session cleanup

Co-authored-by: bennylii <13807507+bennylii@users.noreply.github.com>

Author: xiaoY233

src/main/index.ts

@@ -4,6 +4,7 @@ import { createWindow, getMainWindow, loadUrl, loadFile, openDevTools } from './
 import { createTrayManager, TrayManager } from './tray/TrayManager'
 import { registerIpcHandlers } from './ipc/handlers'
 import { UpdaterManager } from './updater'
+import { storeManager } from './store/store'
 
 // Prevent uncaught exceptions from crashing the app
 process.on('uncaughtException', (error) => {
@@ -125,6 +126,7 @@ async function loadAppContent(mainWindow: BrowserWindow): Promise<void> {
 
 function cleanup(): void {
   console.log('Application is exiting, performing cleanup...')
+  storeManager.flushPendingWrites()
   const updaterManager = UpdaterManager.getInstance()
   updaterManager.destroy()
 }

src/main/proxy/forwarder.ts

@@ -828,7 +828,7 @@ CRITICAL RULES:
           const originalEnd = transformedStream.end.bind(transformedStream)
           transformedStream.end = function(chunk?: any, encoding?: any, callback?: any) {
             const realChatId = handler.getConversationId()
-            if (realChatId && realChatId.startsWith('kimi-') === false) {
+            if (realChatId) {
               adapter.deleteConversation(realChatId).catch(err => {
                 console.error('[Kimi] Failed to delete conversation:', err)
               })
@@ -854,7 +854,7 @@ CRITICAL RULES:
 
       if (shouldDeleteSession()) {
         const realChatId = handler.getConversationId()
-        if (realChatId && realChatId.startsWith('kimi-') === false) {
+        if (realChatId) {
           await adapter.deleteConversation(realChatId)
         }
       }
@@ -1006,22 +1006,22 @@ CRITICAL RULES:
         }
       }
 
-      const deleteChatCallback = shouldDeleteSession()
-        ? async (cid: string) => {
-            try {
-              await adapter.deleteChat(cid)
-            } catch (err) {
-              console.error('[QwenAI] Failed to delete chat:', err)
-            }
-          }
-        : undefined
-
-      const handler = new QwenAiStreamHandler(actualModel, deleteChatCallback)
+      const handler = new QwenAiStreamHandler(actualModel)
       handler.setChatId(chatId)
 
       if (request.stream) {
         const transformedStream = await handler.handleStream(response.data)
 
+        if (shouldDeleteSession()) {
+          const originalEnd = transformedStream.end.bind(transformedStream)
+          transformedStream.end = function(chunk?: any, encoding?: any, callback?: any) {
+            adapter.deleteChat(chatId).catch(err => {
+              console.error('[QwenAI] Failed to delete chat:', err)
+            })
+            return originalEnd(chunk, encoding, callback)
+          }
+        }
+
         return {
           success: true,
           status: response.status,
@@ -1037,8 +1037,8 @@ CRITICAL RULES:
 
       this.applyToolCallsToResponse(result, request.model, request.tools)
 
-      if (deleteChatCallback) {
-        await deleteChatCallback(chatId)
+      if (shouldDeleteSession()) {
+        await adapter.deleteChat(chatId)
       }
 
       return {

src/main/proxy/routes/chat.ts

@@ -261,7 +261,7 @@ router.post('/completions', async (ctx: Context) => {
       todayUsed: (account.todayUsed || 0) + 1,
     })
 
-    storeManager.addLog('info', `Request succeeded`, {
+    storeManager.addLog('debug', `Request succeeded`, {
       requestId,
       providerId: provider.id,
       accountId: account.id,

src/main/proxy/routes/completions.ts

@@ -170,7 +170,7 @@ router.post('/completions', async (ctx: Context) => {
       todayUsed: (account.todayUsed || 0) + 1,
     })
 
-    storeManager.addLog('info', `Request succeeded`, {
+    storeManager.addLog('debug', `Request succeeded`, {
       requestId,
       providerId: provider.id,
       accountId: account.id,

src/main/proxy/server.ts

@@ -13,6 +13,8 @@ import { proxyStatusManager } from './status'
 import { storeManager } from '../store/store'
 import { sessionManager } from './sessionManager'
 
+const SLOW_REQUEST_THRESHOLD_MS = 1500
+
 /**
  * Proxy Server Class
  */
@@ -129,16 +131,19 @@ export class ProxyServer {
       await next()
 
       const latency = Date.now() - startTime
-      const logLevel = ctx.status >= 400 ? 'warn' : 'info'
+      const shouldRecordAccessLog =
+        !ctx.path.startsWith('/v1/models') &&
+        (ctx.status >= 400 || latency >= SLOW_REQUEST_THRESHOLD_MS)
 
-      if (!ctx.path.startsWith('/v1/models')) {
-        storeManager.addLog(logLevel, `${ctx.method} ${ctx.path} ${ctx.status} ${latency}ms`, {
+      if (shouldRecordAccessLog) {
+        storeManager.addLog('warn', `${ctx.method} ${ctx.path} ${ctx.status} ${latency}ms`, {
           data: {
             method: ctx.method,
             path: ctx.path,
             status: ctx.status,
             latency,
             clientIP: ctx.ip,
+            slowRequest: latency >= SLOW_REQUEST_THRESHOLD_MS,
           },
         })
       }

src/main/requestLogs/manager.ts

@@ -26,6 +26,9 @@ export class RequestLogManager {
   private requestLogs: RequestLogEntry[] = []
   private config: RequestLogConfig
   private initialized = false
+  private persistTimer: NodeJS.Timeout | null = null
+  private dirty = false
+  private readonly persistDelayMs = 2000
 
   constructor(options: RequestLogManagerOptions) {
     this.storageDir = options.storageDir
@@ -44,7 +47,7 @@ export class RequestLogManager {
   setConfig(config: Partial<RequestLogConfig>): void {
     this.config = normalizeRequestLogConfig(config)
     this.requestLogs = trimRequestLogsToMaxEntries(this.requestLogs, this.config)
-    this.persist()
+    this.schedulePersist()
   }
 
   async migrateLegacyLogs(legacyLogs: RequestLogEntry[]): Promise<boolean> {
@@ -63,7 +66,7 @@ export class RequestLogManager {
         })),
       this.config,
     )
-    this.persist()
+    this.schedulePersist()
 
     return true
   }
@@ -82,7 +85,7 @@ export class RequestLogManager {
 
     this.requestLogs.push(logEntry)
     this.requestLogs = trimRequestLogsToMaxEntries(this.requestLogs, this.config)
-    this.persist()
+    this.schedulePersist()
 
     return logEntry
   }
@@ -102,7 +105,7 @@ export class RequestLogManager {
       ...this.requestLogs[index],
       ...sanitizeRequestLogUpdates(updates, this.config),
     }
-    this.persist()
+    this.schedulePersist()
     return true
   }
 
@@ -135,7 +138,7 @@ export class RequestLogManager {
   clearRequestLogs(): void {
     this.ensureInitialized()
     this.requestLogs = []
-    this.persist()
+    this.schedulePersist()
   }
 
   getRequestLogStats(): RequestLogStats {
@@ -188,6 +191,19 @@ export class RequestLogManager {
     return [...this.requestLogs]
   }
 
+  flushSync(): void {
+    if (!this.initialized) {
+      return
+    }
+
+    if (this.persistTimer) {
+      clearTimeout(this.persistTimer)
+      this.persistTimer = null
+    }
+
+    this.persistNow()
+  }
+
   private ensureInitialized(): void {
     if (!this.initialized) {
       throw new Error('RequestLogManager is not initialized')
@@ -217,9 +233,27 @@ export class RequestLogManager {
       .filter((entry): entry is RequestLogEntry => entry !== null)
   }
 
-  private persist(): void {
+  private schedulePersist(): void {
+    this.dirty = true
+
+    if (this.persistTimer) {
+      clearTimeout(this.persistTimer)
+    }
+
+    this.persistTimer = setTimeout(() => {
+      this.persistTimer = null
+      this.persistNow()
+    }, this.persistDelayMs)
+  }
+
+  private persistNow(): void {
+    if (!this.dirty) {
+      return
+    }
+
     const content = this.requestLogs.map((entry) => JSON.stringify(entry)).join('\n')
     writeFileSync(this.logFile, content, 'utf-8')
+    this.dirty = false
   }
 }