shanshanzhong
fcdd6ac170
All checks were successful
Build docker and publish / build (20.15.1) (push) Successful in 6m47s
添加设备删除前的检查逻辑，确保设备存在后再执行删除操作 在删除设备前先踢下线在线设备并清理相关缓存会话 最后删除数据库中的设备记录，确保数据一致性
63 lines
2.1 KiB
Go
63 lines
2.1 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/perfect-panel/server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/internal/config"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type DeleteUserDeviceLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// Delete user device
|
|
func NewDeleteUserDeviceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteUserDeviceLogic {
|
|
return &DeleteUserDeviceLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *DeleteUserDeviceLogic) DeleteUserDevice(req *types.DeleteUserDeivceRequest) error {
|
|
device, findErr := l.svcCtx.UserModel.FindOneDevice(l.ctx, req.Id)
|
|
if findErr != nil {
|
|
if errors.Is(findErr, gorm.ErrRecordNotFound) {
|
|
return nil
|
|
}
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "get Device error: %v", findErr.Error())
|
|
}
|
|
|
|
// 尝试踢下线在线设备
|
|
l.svcCtx.DeviceManager.KickDevice(device.UserId, device.Identifier)
|
|
|
|
// 清理与设备相关的缓存会话
|
|
ctx, cancel := context.WithTimeout(l.ctx, 2*time.Second)
|
|
defer cancel()
|
|
deviceCacheKey := fmt.Sprintf("%v:%v", config.DeviceCacheKeyKey, device.Identifier)
|
|
if sessionId, rerr := l.svcCtx.Redis.Get(ctx, deviceCacheKey).Result(); rerr == nil && sessionId != "" {
|
|
_ = l.svcCtx.Redis.Del(ctx, deviceCacheKey).Err()
|
|
sessionIdCacheKey := fmt.Sprintf("%v:%v", config.SessionIdKey, sessionId)
|
|
_ = l.svcCtx.Redis.Del(ctx, sessionIdCacheKey).Err()
|
|
sessionsKey := fmt.Sprintf("%s%v", config.UserSessionsKeyPrefix, device.UserId)
|
|
_ = l.svcCtx.Redis.ZRem(ctx, sessionsKey, sessionId).Err()
|
|
}
|
|
|
|
// 最后删除数据库记录
|
|
if err := l.svcCtx.UserModel.DeleteDevice(l.ctx, req.Id); err != nil {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseDeletedError), "delete user error: %v", err.Error())
|
|
}
|
|
return nil
|
|
}
|