chore: use upsert to handle conflict

Signed-off-by: Wei Zhang <[email protected]>
TabbyML · Dec 14, 2024 · 860ca46 · 860ca46
1 parent cd0152e
commit 860ca46
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/ee/tabby-db/src/notifications.rs b/ee/tabby-db/src/notifications.rs
@@ -34,14 +34,11 @@ impl DbConn {
     pub async fn mark_notification_read(&self, id: i64, user_id: i64) -> Result<()> {
         query!(
             "INSERT INTO read_notifications (notification_id, user_id)
-            SELECT ?, ?
-            WHERE NOT EXISTS (
-                SELECT 1 FROM read_notifications WHERE notification_id = ? AND user_id = ?
-            )",
+             VALUES (?, ?)
+             ON CONFLICT (notification_id, user_id)
+             DO NOTHING",
             id,
             user_id,
-            id,
-            user_id
         )
         .execute(&self.pool)
         .await?;