fix(db): should use update message

TabbyML · Oct 31, 2024 · 640aeaa · 640aeaa
1 parent 2c7de9f
commit 640aeaa
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 6 deletions.
diff --git a/ee/tabby-db/src/threads.rs b/ee/tabby-db/src/threads.rs
@@ -220,6 +220,24 @@ impl DbConn {
         Ok(())
     }
 
+    pub async fn update_thread_message_content(
+        &self,
+        thread_id: i64,
+        message_id: i64,
+        content: &str,
+    ) -> Result<()> {
+        query!(
+            "UPDATE thread_messages SET content = ?, updated_at = DATETIME('now') WHERE thread_id = ? AND id = ?",
+            content,
+            thread_id,
+            message_id
+        )
+        .execute(&self.pool)
+        .await?;
+
+        Ok(())
+    }
+
     pub async fn append_thread_message_content(
         &self,
         message_id: i64,

diff --git a/ee/tabby-schema/src/schema/mod.rs b/ee/tabby-schema/src/schema/mod.rs
@@ -1179,11 +1179,6 @@ impl Mutation {
 
         user.policy.check_update_thread_message(&thread.user_id)?;
 
-        let message = svc.get_thread_message(&input.id).await?;
-        if input.content == message.content {
-            return Ok(true);
-        }
-
         svc.update_thread_message(&input).await?;
         Ok(true)
     }

diff --git a/ee/tabby-webserver/src/service/thread.rs b/ee/tabby-webserver/src/service/thread.rs
@@ -70,7 +70,11 @@ impl ThreadService for ThreadServiceImpl {
 
     async fn update_thread_message(&self, input: &UpdateMessageInput) -> Result<()> {
         self.db
-            .append_thread_message_content(input.id.as_rowid()?, &input.content)
+            .update_thread_message_content(
+                input.thread_id.as_rowid()?,
+                input.id.as_rowid()?,
+                &input.content,
+            )
             .await?;
         Ok(())
     }