feat: 增加自动回复开关

pkg/router/router.go

@@ -135,6 +135,8 @@ func NewRouter() *gin.Engine {
 		v1.POST("aschat/autoReplyRuler/update", asChat.Handler.UpdateChatAutoReplyRuler)
 		v1.POST("aschat/autoReplyRuler/detail", asChat.Handler.GetChatAutoReplyRulerDetail)
 		v1.POST("aschat/autoReplyRuler/query", asChat.Handler.GetChatAutoReplyRulerList)
+		v1.POST("aschat/autoReplyRuler/userSwitch/get", asChat.ChatHandlerIns.UserSwitchAutoReplyStatus) //获取自动回复开关状态
+		v1.POST("aschat/autoReplyRuler/userSwitch/set", asChat.ChatHandlerIns.SetSwitchAutoReplyStatus)  //设置自动回复开关状态
 
 		v1.POST("/test/user/log/erp", asChat.Handler.ErpLoginDemo)
 		v1.POST("/test/user/log/fiee", asChat.Handler.FieeLoginDemo)

pkg/service/asChat/chatCache/cache.go

@@ -17,6 +17,7 @@ import (
 	"github.com/goccy/go-json"
 	"go.uber.org/zap"
 	"log"
+	"strconv"
 	"strings"
 	"sync"
 	"time"
@@ -25,6 +26,7 @@ import (
 const CacheChatRecordKey = "fiee:chatRecord"
 const CacheSessionKey = "fiee:chatSession"
 const CacheNewMsgStatKey = "fiee:newMsgStat"
+const CacheAutoReplySwitchKey = "fiee:AutoReplySwitch"
 
 var chatCacheLocker sync.RWMutex
 
@@ -248,3 +250,37 @@ func (cr ChatCache) coverOwnerNewMessageStat(ctx context.Context, ownerId int64,
 	err = cache.RedisClient.Set(cr.GetNewMsgStatCacheKey(ownerId), value, 0).Err()
 	return
 }
+
+// -----------------------------------用户自动回复开关
+// erp获取最新的消息统计
+func (cr ChatCache) GetCacheAutoReplySwitchKey(userId int64) string {
+	return fmt.Sprintf("%s:%d", CacheAutoReplySwitchKey, userId)
+}
+
+func (cr ChatCache) SetAutoReplySwitch(ctx context.Context, ownerId int64, enableAutoReply bool) {
+	//chatCacheLocker.RLock()
+	//defer chatCacheLocker.RUnlock()
+	err := cache.RedisClient.Set(cr.GetCacheAutoReplySwitchKey(ownerId), enableAutoReply, 0).Err()
+	if err != nil {
+		log.Print("保存用户会话失败", zap.Error(err))
+	}
+	return
+}
+
+func (cr ChatCache) GetAutoReplySwitch(ctx context.Context, ownerId int64) (enableAutoReply bool) {
+	//chatCacheLocker.RLock()
+	//defer chatCacheLocker.RUnlock()
+	val, err := cache.RedisClient.Get(cr.GetCacheAutoReplySwitchKey(ownerId)).Bytes()
+	if err != nil {
+		log.Print("获取自动回复开关查询失败", zap.Error(err), zap.Int64("ownerId", ownerId))
+		return true
+	}
+	// 解析缓存值为布尔值
+	boolVal, err := strconv.ParseBool(string(val))
+	if err != nil {
+		log.Print("解析自动回复开关值失败", zap.Error(err), zap.String("value", string(val)), zap.Int64("ownerId", ownerId))
+		return true // 解析失败时也返回默认值true
+	}
+
+	return boolVal
+}

pkg/service/asChat/dto/dto.go

@@ -35,6 +35,7 @@ type NewMessageRequest struct {
 	SessionId string `json:"sessionId"`
 	Message
 	AtUserId int64 `json:"atUserId"` //指定发送给sessionId中的某一个用户
+	//EnableAutoReply bool  `json:"-"`
 }
 
 // 服务端接收到消息后，使用websocket发送给userId关联的客户端，通知客户端有新消息，然后调用接口获取消息
@@ -218,3 +219,7 @@ type UserDetailResp struct {
 	Phone       string `json:"phone"`
 	GroupPhoto  string `json:"groupPhoto"`
 }
+
+type UserSwitchAutoReplyReq struct {
+	EnableAutoReply bool `json:"enableAutoReply"`
+}

pkg/service/asChat/handler.go

@@ -612,3 +612,36 @@ func (cr ChatHandler) UserDetail(c *gin.Context) {
 
 	service.Success(c, detail)
 }
+
+func (a *ChatHandler) UserSwitchAutoReplyStatus(c *gin.Context) {
+	chatUser, code := jwt.ParseToChatUser(c)
+	if code != 0 {
+		service.ErrWithCode(c, code)
+		return
+	}
+	var req dto.UserSwitchAutoReplyReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		service.Error(c, err)
+		return
+	}
+	enable := a.cache.GetAutoReplySwitch(c, chatUser.ID)
+	var resp = map[string]any{
+		"enableAutoReply": enable,
+	}
+	service.Success(c, resp)
+}
+
+func (a *ChatHandler) SetSwitchAutoReplyStatus(c *gin.Context) {
+	chatUser, code := jwt.ParseToChatUser(c)
+	if code != 0 {
+		service.ErrWithCode(c, code)
+		return
+	}
+	var req dto.UserSwitchAutoReplyReq
+	if err := c.ShouldBindJSON(&req); err != nil {
+		service.Error(c, err)
+		return
+	}
+	a.cache.SetAutoReplySwitch(c, chatUser.ID, req.EnableAutoReply)
+	service.Success(c)
+}

pkg/service/asChat/robot/robot.go

@@ -165,6 +165,11 @@ func (r *Robot) Run() {
 			fmt.Printf("robot listen event:%#v\n", event)
 			r.mu.Lock()
 			//加入聊天室规则
+			enableAutoReply := false
+			if event.Client != nil {
+				enableAutoReply = new(chatCache.ChatCache).GetAutoReplySwitch(context.Background(), event.Client.UserId)
+			}
+			if enableAutoReply {
 				hasHit := false
 				for _, rule := range r.joinSessionRules {
 					hit := rule.Hit(event, r.Info)
@@ -217,6 +222,7 @@ func (r *Robot) Run() {
 						}
 					}
 				}
+			}
 			r.mu.Unlock()
 		}
 	}