2023-09-21 17:33:59 +08:00
|
|
|
|
package service
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"go-wechat/client"
|
|
|
|
|
"go-wechat/entity"
|
|
|
|
|
"log"
|
2023-12-10 08:14:09 +08:00
|
|
|
|
"os"
|
|
|
|
|
"strconv"
|
2023-09-21 17:33:59 +08:00
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// SaveMessage
|
|
|
|
|
// @description: 消息入库
|
|
|
|
|
// @param msg
|
|
|
|
|
func SaveMessage(msg entity.Message) {
|
2023-12-10 08:14:09 +08:00
|
|
|
|
if flag, _ := strconv.ParseBool(os.Getenv("DONT_SAVE")); flag {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
2023-09-21 17:33:59 +08:00
|
|
|
|
// 检查消息是否存在,存在就跳过
|
|
|
|
|
var count int64
|
|
|
|
|
err := client.MySQL.Model(&entity.Message{}).Where("msg_id = ?", msg.MsgId).Count(&count).Error
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Printf("检查消息是否存在失败, 错误信息: %v", err)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
if count > 0 {
|
2023-12-09 11:52:11 +08:00
|
|
|
|
//log.Printf("消息已存在,消息Id: %d", msg.MsgId)
|
2023-09-21 17:33:59 +08:00
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
err = client.MySQL.Create(&msg).Error
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Printf("消息入库失败, 错误信息: %v", err)
|
|
|
|
|
}
|
|
|
|
|
log.Printf("消息入库成功,消息Id: %d", msg.MsgId)
|
|
|
|
|
}
|