退款消息推送

models/user/user_message.go

@@ -17,6 +17,14 @@ const (
 	MyMessageColumns                    = "id,source_id,type,message"
 )
 
+var (
+	messageTypeMap = []message.SourceType{
+		message.ReportSourceType,
+		message.VideoSourceType,
+		message.AudioSourceType,
+	}
+)
+
 // UserMessage 表示 user_message 表的模型
 type UserMessage struct {
 	Id          int                `gorm:"primaryKey;autoIncrement;column:id"`
@@ -73,7 +81,7 @@ func GetUnReadMessageList(userId int) (messages []UserMessage, err error) {
 
 func ReadMessage(userId int, messageId int) bool {
 	db := models.Main()
-	err := db.Model(&UserMessage{}).Where("id=? and user_id=?", messageId, userId).Update("status", ReadStatus).Error
+	err := db.Model(&UserMessage{}).Where("id=? and user_id=? and type  in ?", messageId, userId, messageTypeMap).Update("status", ReadStatus).Error
 	if err != nil {
 		return false
 	}
@@ -82,7 +90,7 @@ func ReadMessage(userId int, messageId int) bool {
 
 func ReadMessages(userId int, analystId int) bool {
 	db := models.Main()
-	err := db.Model(&UserMessage{}).Where("user_id=? and analyst_id =?", userId, analystId).Update("status", ReadStatus).Error
+	err := db.Model(&UserMessage{}).Where("user_id=? and analyst_id =? and type in ?", userId, analystId, messageTypeMap).Update("status", ReadStatus).Error
 	if err != nil {
 		return false
 	}