From 01d8888601b7985ac6d6d616080961047473c3db Mon Sep 17 00:00:00 2001
From: fullex <0xfullex@gmail.com>
Date: Sat, 3 Jan 2026 18:52:11 +0800
Subject: [PATCH] feat: extend UpdateMessageDto with traceId and stats fields

- Added optional fields `traceId` and `stats` to the `UpdateMessageDto` interface for enhanced message tracking and statistics.
- Updated `MessageService` to handle the new fields during message updates, ensuring they are correctly processed in the database.
---
 packages/shared/data/api/schemas/messages.ts | 4 ++++
 src/main/data/services/MessageService.ts     | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/packages/shared/data/api/schemas/messages.ts b/packages/shared/data/api/schemas/messages.ts
index b952156da5..a2462ca489 100644
--- a/packages/shared/data/api/schemas/messages.ts
+++ b/packages/shared/data/api/schemas/messages.ts
@@ -62,6 +62,10 @@ export interface UpdateMessageDto {
   siblingsGroupId?: number
   /** Update status */
   status?: MessageStatus
+  /** Update trace ID */
+  traceId?: string | null
+  /** Update statistics */
+  stats?: MessageStats | null
 }
 
 /**
diff --git a/src/main/data/services/MessageService.ts b/src/main/data/services/MessageService.ts
index 5f1c05e749..7542ffc02b 100644
--- a/src/main/data/services/MessageService.ts
+++ b/src/main/data/services/MessageService.ts
@@ -558,6 +558,8 @@ export class MessageService {
       if (dto.parentId !== undefined) updates.parentId = dto.parentId
       if (dto.siblingsGroupId !== undefined) updates.siblingsGroupId = dto.siblingsGroupId
       if (dto.status !== undefined) updates.status = dto.status
+      if (dto.traceId !== undefined) updates.traceId = dto.traceId
+      if (dto.stats !== undefined) updates.stats = dto.stats
 
       const [row] = await tx.update(messageTable).set(updates).where(eq(messageTable.id, id)).returning()