102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
|
// Package service - Webhook服务
|
|||
|
|
// 功能：接收并处理平台（小红书、抖音）的webhook消息
|
|||
|
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"customer-server/dao"
|
|||
|
|
"customer-server/model/dto"
|
|||
|
|
|
|||
|
|
"gitea.com/red-future/common/jaeger"
|
|||
|
|
"gitea.com/red-future/common/rabbitmq"
|
|||
|
|
"gitea.com/red-future/common/redis"
|
|||
|
|
"gitea.com/red-future/common/utils"
|
|||
|
|
"github.com/gogf/gf/v2/os/glog"
|
|||
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|||
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var Webhook = new(webhookService)
|
|||
|
|
|
|||
|
|
type webhookService struct{}
|
|||
|
|
|
|||
|
|
// Receive 接收平台消息并写入队列
|
|||
|
|
func (s *webhookService) Receive(ctx context.Context, req *dto.WebhookReceiveReq) (res *dto.WebhookReceiveRes, err error) {
|
|||
|
|
glog.Infof(ctx, "收到 Webhook 消息 - 平台: %s, 用户: %s, 内容: %s", req.Platform, req.UserId, req.Content)
|
|||
|
|
|
|||
|
|
// 生成消息ID
|
|||
|
|
now := gtime.Now()
|
|||
|
|
messageId := req.Platform + "_" + req.UserId + "_" + gconv.String(now.TimestampNano())
|
|||
|
|
if req.MsgId != "" {
|
|||
|
|
messageId = req.MsgId // 使用平台消息ID（便于去重）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 从 token 获取租户ID
|
|||
|
|
var tenantId string
|
|||
|
|
if user, userErr := utils.GetUserInfo(ctx); userErr == nil {
|
|||
|
|
tenantId = gconv.String(user.TenantId)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构造消息
|
|||
|
|
userId := req.Platform + "_" + req.UserId // 添加平台前缀
|
|||
|
|
// 获取当前实例的动态响应队列名（自动生成，支持多实例部署）
|
|||
|
|
baseQueue := GetConfigString(ctx, "rabbitmq.responseQueue")
|
|||
|
|
replyQueue := rabbitmq.GetInstanceQueueName(baseQueue)
|
|||
|
|
msg := &redis.SendStreamMessage{
|
|||
|
|
UserId: userId,
|
|||
|
|
TenantId: tenantId,
|
|||
|
|
Content: req.Content,
|
|||
|
|
Timestamp: now.Timestamp(),
|
|||
|
|
MessageId: messageId,
|
|||
|
|
Platform: req.Platform,
|
|||
|
|
AccountId: req.AccountId,
|
|||
|
|
ReplyQueue: replyQueue,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查是否有 session 缓存，无缓存说明已归档，需要读取历史
|
|||
|
|
if sessionId, _ := redis.GetSessionCache(ctx, tenantId, userId); sessionId == "" {
|
|||
|
|
if history, histErr := dao.Conversation.GetRecentHistory(ctx, userId, redis.GetHistoryContextLimit()); histErr == nil && len(history) > 0 {
|
|||
|
|
msg.History = history
|
|||
|
|
glog.Infof(ctx, "用户已归档，读取 %d 轮历史对话 - 用户: %s", len(history), userId)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 写入 Redis Stream
|
|||
|
|
streamMsgId, err := redis.AddToStream(ctx, redis.RAGFlowRequestStreamKey, msg)
|
|||
|
|
if err != nil {
|
|||
|
|
jaeger.RecordError(ctx, err, "写入 Stream 失败")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
glog.Infof(ctx, "消息已写入 Stream - MessageID: %s", streamMsgId)
|
|||
|
|
|
|||
|
|
res = &dto.WebhookReceiveRes{
|
|||
|
|
Success: true,
|
|||
|
|
MsgId: streamMsgId,
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetHistory 查询用户对话记录
|
|||
|
|
func (s *webhookService) GetHistory(ctx context.Context, req *dto.ConversationListReq) (res *dto.ConversationListRes, err error) {
|
|||
|
|
list, err := dao.Conversation.FindByUserId(ctx, req.UserId, req.Limit)
|
|||
|
|
if err != nil {
|
|||
|
|
jaeger.RecordError(ctx, err, "查询对话记录失败")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
res = &dto.ConversationListRes{
|
|||
|
|
List: make([]*dto.ConversationItem, 0, len(list)),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, item := range list {
|
|||
|
|
res.List = append(res.List, &dto.ConversationItem{
|
|||
|
|
Question: item.Question,
|
|||
|
|
Answer: item.Answer,
|
|||
|
|
MsgTime: gtime.New(item.MsgTime).Format("Y-m-d H:i:s"),
|
|||
|
|
SessionId: item.SessionId,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|