🎨 消息处理逻辑优化

handler/at_message.go

@@ -4,9 +4,10 @@ import (
 	"context"
 	"fmt"
 	"github.com/sashabaranov/go-openai"
+	"go-wechat/client"
 	"go-wechat/config"
 	"go-wechat/entity"
-	"go-wechat/service"
+	"go-wechat/model"
 	"go-wechat/utils"
 	"log"
 	"regexp"
@@ -16,26 +17,15 @@ import (
 // handleAtMessage
 // @description: 处理At机器人的消息
 // @param m
-func handleAtMessage(m entity.Message) {
+func handleAtMessage(m model.Message) {
 	if !config.Conf.Ai.Enable {
 		return
 	}
 
 	// 取出所有启用了AI的好友或群组
-	us, err := service.GetAllEnableAI()
+	var count int64
-	if err != nil {
+	client.MySQL.Model(&entity.Friend{}).Where("enable_ai IS TRUE").Where("wxid = ?", m.FromUser).Count(&count)
-		utils.SendMessage(m.FromUser, m.GroupUser, "#系统异常\n"+err.Error(), 0)
+	if count < 1 {
-		return
-	}
-	// 判断是否启用，如果没有启用，直接返回
-	var canUse bool
-	for _, u := range us {
-		if u.Wxid == m.FromUser {
-			canUse = true
-			break
-		}
-	}
-	if !canUse {
 		return
 	}
 
@@ -63,9 +53,9 @@ func handleAtMessage(m entity.Message) {
 	})
 
 	// 配置模型
-	model := openai.GPT3Dot5Turbo0613
+	chatModel := openai.GPT3Dot5Turbo0613
 	if config.Conf.Ai.Model != "" {
-		model = config.Conf.Ai.Model
+		chatModel = config.Conf.Ai.Model
 	}
 
 	// 默认使用AI回复
@@ -77,7 +67,7 @@ func handleAtMessage(m entity.Message) {
 	resp, err := client.CreateChatCompletion(
 		context.Background(),
 		openai.ChatCompletionRequest{
-			Model:    model,
+			Model:    chatModel,
 			Messages: messages,
 		},
 	)

handler/parse.go

@@ -6,6 +6,7 @@ import (
 	"go-wechat/model"
 	"go-wechat/service"
 	"go-wechat/types"
+	"go-wechat/utils"
 	"log"
 	"net"
 	"strings"
@@ -23,43 +24,47 @@ func Parse(remoteAddr net.Addr, msg []byte) {
 		return
 	}
 	// 提取出群成员信息
-	groupUser := ""
+	//groupUser := ""
-	msgStr := m.Content
+	//msgStr := m.Content
 	if strings.Contains(m.FromUser, "@") {
-		switch m.Type {
+		// 群消息，处理一下消息和发信人
-		case types.MsgTypeRecalled:
+		groupUser := strings.Split(m.Content, "\n")[0]
-			// 消息撤回
-		case types.MsgTypeSys:
-			// 系统消息
-			go handleSysMessage(m)
-		default:
-			// 默认消息处理
-			groupUser = strings.Split(m.Content, "\n")[0]
 		groupUser = strings.ReplaceAll(groupUser, ":", "")
+		// 如果两个id一致，说明是系统发的
+		if m.FromUser != groupUser {
+			m.GroupUser = groupUser
+		}
+		// 用户的操作单独提出来处理一下
+		m.Content = strings.Join(strings.Split(m.Content, "\n")[1:], "\n")
+	}
+	log.Printf("%s\n消息来源: %s\n群成员: %s\n消息类型: %v\n消息内容: %s", remoteAddr, m.FromUser, m.GroupUser, m.Type, m.Content)
 
-			// 文字消息单独提出来处理一下
+	// 异步处理消息
-			msgStr = strings.Join(strings.Split(m.Content, "\n")[1:], "\n")
+	go func() {
+		if m.IsNewUserJoin() {
+			// 欢迎新成员
+			go handleNewUserJoin(m)
+		} else if m.IsAt() {
+			// @机器人的消息
+			go handleAtMessage(m)
+		} else if !strings.Contains(m.FromUser, "@") && m.Type == types.MsgTypeText {
+			// 私聊消息处理
+			utils.SendMessage(m.FromUser, "", "暂未开启私聊AI", 0)
 		}
-	}
+	}()
-	log.Printf("%s\n消息来源: %s\n群成员: %s\n消息类型: %v\n消息内容: %s", remoteAddr, m.FromUser, groupUser, m.Type, msgStr)
 
 	// 转换为结构体之后入库
 	var ent entity.Message
 	ent.MsgId = m.MsgId
 	ent.CreateTime = m.CreateTime
 	ent.CreateAt = time.Unix(int64(m.CreateTime), 0)
-	ent.Content = msgStr
+	ent.Content = m.Content
 	ent.FromUser = m.FromUser
-	ent.GroupUser = groupUser
+	ent.GroupUser = m.GroupUser
 	ent.ToUser = m.ToUser
 	ent.Type = m.Type
 	ent.DisplayFullContent = m.DisplayFullContent
 	ent.Raw = string(msg)
 
-	// 处理At机器人的消息
-	if strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你") {
-		go handleAtMessage(ent)
-	}
-
 	go service.SaveMessage(ent)
 }

model/message.go

@@ -1,18 +1,87 @@
 package model
 
-import "go-wechat/types"
+import (
+	"encoding/xml"
+	"go-wechat/types"
+	"strings"
+)
 
 // Message
 // @description: 消息
 type Message struct {
-	MsgId              int64             `json:"msgId" gorm:"primarykey"`
+	MsgId              int64             `json:"msgId"`
 	CreateTime         int               `json:"createTime"`
 	Content            string            `json:"content"`
-	DisplayFullContent string            `json:"displayFullContent" gorm:"-"`
+	DisplayFullContent string            `json:"displayFullContent"`
 	FromUser           string            `json:"fromUser"`
+	GroupUser          string            `json:"-"`
 	MsgSequence        int               `json:"msgSequence"`
 	Pid                int               `json:"pid"`
 	Signature          string            `json:"signature"`
 	ToUser             string            `json:"toUser"`
 	Type               types.MessageType `json:"type"`
 }
+
+// systemMsgDataXml
+// @description: 微信系统消息的xml结构
+type systemMsgDataXml struct {
+	SysMsg sysMsg `xml:"sysmsg"`
+	Type   string `xml:"type,attr"`
+}
+
+// sysMsg
+// @description: 消息主体
+type sysMsg struct{}
+
+// IsPat
+// @description: 是否是拍一拍消息
+// @receiver m
+// @return bool
+func (m Message) IsPat() bool {
+	// 解析xml
+	var d systemMsgDataXml
+	if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
+		return false
+	}
+
+	return m.Type == types.MsgTypeRecalled && d.Type == "pat"
+}
+
+// IsRevokeMsg
+// @description: 是否是撤回消息
+// @receiver m
+// @return bool
+func (m Message) IsRevokeMsg() bool {
+	// 解析xml
+	var d systemMsgDataXml
+	if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
+		return false
+	}
+
+	return m.Type == types.MsgTypeRecalled && d.Type == "revokemsg"
+}
+
+// IsNewUserJoin
+// @description: 是否是新人入群
+// @receiver m
+// @return bool
+func (m Message) IsNewUserJoin() bool {
+	sysFlag := m.Type == types.MsgTypeSys && strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊")
+	if sysFlag {
+		return true
+	}
+	// 解析另一种情况
+	var d systemMsgDataXml
+	if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
+		return false
+	}
+	return m.Type == types.MsgTypeSys && d.Type == "delchatroommember"
+}
+
+// IsAt
+// @description: 是否是At机器人的消息
+// @receiver m
+// @return bool
+func (m Message) IsAt() bool {
+	return strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你")
+}

types/message.go

@@ -4,6 +4,7 @@ import "fmt"
 
 type MessageType int
 
+// 微信定义的消息类型
 const (
 	MsgTypeText           MessageType = 1     // 文本消息
 	MsgTypeImage          MessageType = 3     // 图片消息