shanshanzhong
384c8df506
All checks were successful
Build docker and publish / build (20.15.1) (push) Successful in 7m43s
- kickOfflineByUserDeviceLogic: 管理员踢设备后新增 clearAllSessions， 之前只清单个 WebSocket session，用户可用旧 token 继续访问 - unbindDeviceLogic: 家庭成员被踢时增加踢设备+清 session； 补全 session detail key 清理 Co-Authored-By: claude-flow <ruv@ruv.net>
87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/perfect-panel/server/internal/config"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"github.com/perfect-panel/server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type KickOfflineByUserDeviceLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// kick offline user device
|
|
func NewKickOfflineByUserDeviceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *KickOfflineByUserDeviceLogic {
|
|
return &KickOfflineByUserDeviceLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *KickOfflineByUserDeviceLogic) KickOfflineByUserDevice(req *types.KickOfflineRequest) error {
|
|
device, err := l.svcCtx.UserModel.FindOneDevice(l.ctx, req.Id)
|
|
if err != nil {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "get Device error: %v", err.Error())
|
|
}
|
|
l.svcCtx.DeviceManager.KickDevice(device.UserId, device.Identifier)
|
|
device.Online = false
|
|
err = l.svcCtx.UserModel.UpdateDevice(l.ctx, device)
|
|
if err != nil {
|
|
l.Logger.Error("[KickOfflineByUserDeviceLogic] Update Device Error:", logger.Field("err", err.Error()))
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "update Device error: %v", err.Error())
|
|
}
|
|
|
|
// 清除该用户的所有会话，确保旧 token 失效
|
|
l.clearAllSessions(device.UserId)
|
|
|
|
return nil
|
|
}
|
|
|
|
// clearAllSessions 清除指定用户的所有会话
|
|
func (l *KickOfflineByUserDeviceLogic) clearAllSessions(userId int64) {
|
|
sessionsKey := fmt.Sprintf("%s%v", config.UserSessionsKeyPrefix, userId)
|
|
sessions, err := l.svcCtx.Redis.ZRange(l.ctx, sessionsKey, 0, -1).Result()
|
|
if err != nil {
|
|
l.Errorw("获取用户会话列表失败",
|
|
logger.Field("user_id", userId),
|
|
logger.Field("error", err.Error()),
|
|
)
|
|
return
|
|
}
|
|
|
|
if len(sessions) == 0 {
|
|
return
|
|
}
|
|
|
|
pipe := l.svcCtx.Redis.TxPipeline()
|
|
for _, sessionID := range sessions {
|
|
if sessionID == "" {
|
|
continue
|
|
}
|
|
pipe.Del(l.ctx, fmt.Sprintf("%v:%v", config.SessionIdKey, sessionID))
|
|
pipe.Del(l.ctx, fmt.Sprintf("%s:detail:%s", config.SessionIdKey, sessionID))
|
|
}
|
|
pipe.Del(l.ctx, sessionsKey)
|
|
|
|
if _, err = pipe.Exec(l.ctx); err != nil {
|
|
l.Errorw("清理会话缓存失败",
|
|
logger.Field("user_id", userId),
|
|
logger.Field("error", err.Error()),
|
|
)
|
|
}
|
|
|
|
l.Infow("[KickOffline] 管理员踢设备-清除所有Session",
|
|
logger.Field("user_id", userId),
|
|
logger.Field("count", len(sessions)),
|
|
)
|
|
}
|