86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
// Package service - 会话服务
|
||
// 功能：用户会话管理、状态维护
|
||
package service
|
||
|
||
import (
|
||
"context"
|
||
"customer-server/dao"
|
||
|
||
"gitea.com/red-future/common/jaeger"
|
||
"gitea.com/red-future/common/rabbitmq"
|
||
"gitea.com/red-future/common/redis"
|
||
"github.com/gogf/gf/v2/encoding/gjson"
|
||
"github.com/gogf/gf/v2/os/glog"
|
||
)
|
||
|
||
// sessionService 会话服务（操作 session 表）
|
||
type sessionService struct{}
|
||
|
||
// SessionService 会话服务单例
|
||
var SessionService = new(sessionService)
|
||
|
||
// ============== RabbitMQ 消费者（归档延时消息）==============
|
||
|
||
// SessionArchiveConsumer 会话归档消费者
|
||
type SessionArchiveConsumer struct {
|
||
consumer *rabbitmq.Consumer
|
||
}
|
||
|
||
// NewSessionArchiveConsumer 创建会话归档消费者
|
||
func NewSessionArchiveConsumer(ctx context.Context) *SessionArchiveConsumer {
|
||
queueName := GetConfigString(ctx, "archive.queue")
|
||
return &SessionArchiveConsumer{
|
||
consumer: rabbitmq.NewConsumer(queueName, handleSessionArchive),
|
||
}
|
||
}
|
||
|
||
// Start 启动消费者
|
||
func (c *SessionArchiveConsumer) Start(ctx context.Context) (err error) {
|
||
glog.Info(ctx, "会话归档消费者启动...")
|
||
return c.consumer.Start(ctx)
|
||
}
|
||
|
||
// Stop 停止消费者
|
||
func (c *SessionArchiveConsumer) Stop(ctx context.Context) {
|
||
c.consumer.Stop(ctx)
|
||
}
|
||
|
||
// handleSessionArchive 处理会话归档消息
|
||
func handleSessionArchive(ctx context.Context, body []byte) (err error) {
|
||
ctx, span := jaeger.NewSpan(ctx, "consumer.session.archive")
|
||
defer span.End()
|
||
|
||
var msg redis.ArchiveMessage
|
||
if err = gjson.DecodeTo(body, &msg); err != nil {
|
||
jaeger.RecordError(ctx, err, "解析归档消息失败")
|
||
return
|
||
}
|
||
|
||
glog.Infof(ctx, "收到归档消息 - 用户: %s, Session: %s", msg.UserId, msg.SessionId)
|
||
|
||
// 检查用户是否在归档发送后有活跃（60分钟内）
|
||
isActive, err := redis.IsUserActive(ctx, msg.UserId, int64(redis.GetArchiveDelay()))
|
||
if err != nil {
|
||
jaeger.RecordError(ctx, err, "检查用户活跃状态失败")
|
||
return
|
||
}
|
||
|
||
if isActive {
|
||
glog.Infof(ctx, "用户 %s 在归档期间有活跃，跳过归档", msg.UserId)
|
||
return
|
||
}
|
||
|
||
// 执行归档
|
||
if err = dao.Session.Archive(ctx, msg.UserId, msg.SessionId); err != nil {
|
||
jaeger.RecordError(ctx, err, "归档会话失败")
|
||
return
|
||
}
|
||
|
||
// 清除 Session 缓存（需要tenantId）
|
||
// TODO: ArchiveMessage需要添加TenantId字段
|
||
redis.DelSessionCache(ctx, msg.TenantId, msg.UserId)
|
||
|
||
glog.Infof(ctx, "会话已归档 - 用户: %s, Session: %s", msg.UserId, msg.SessionId)
|
||
return
|
||
}
|