From 59d065dbc7dfded9b874c4eb1e6aa70c68998167 Mon Sep 17 00:00:00 2001
From: Rogee <rogeecn@qq.com>
Date: Tue, 3 Sep 2024 09:59:39 +0800
Subject: [PATCH] fix: issues

---
 internal/db_channel.go | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/internal/db_channel.go b/internal/db_channel.go
index 4067cff..0c1e31a 100644
--- a/internal/db_channel.go
+++ b/internal/db_channel.go
@@ -126,14 +126,14 @@ func (c *DBChannel) SaveMessage(ctx context.Context, msg *ChannelMessage) error
 
 	tbl := table.ChannelMessages
 
+	cond := tbl.GroupID.EQ(Int(message.GroupID)).AND(
+		tbl.ChannelID.EQ(Int(message.ChannelID)),
+	)
+
 	var m model.ChannelMessages
 	err := tbl.
 		SELECT(tbl.ID).
-		WHERE(
-			tbl.GroupID.EQ(Int(message.GroupID)).AND(
-				tbl.UUID.EQ(Int64(message.UUID)),
-			),
-		).
+		WHERE(cond).
 		LIMIT(1).
 		QueryContext(ctx, db, &m)
 	if err != nil {
@@ -161,11 +161,7 @@ func (c *DBChannel) SaveMessage(ctx context.Context, msg *ChannelMessage) error
 		tbl.Media.SET(RawString(`media || #var::jsonb`, RawArgs{
 			"#var": msg.GetMedias(),
 		})),
-	).WHERE(
-		tbl.GroupID.EQ(Int(message.GroupID)).AND(
-			tbl.UUID.EQ(Int(message.UUID)),
-		),
-	)
+	).WHERE(cond)
 
 	_, err = stmt.ExecContext(ctx, db)
 	// _, err = db.ExecContext(ctx, stmt.DebugSql())