shanshanzhong
4ffccd5ad8
All checks were successful
Build docker and publish / build (20.15.1) (push) Successful in 5m40s
75 lines
2.4 KiB
Go
75 lines
2.4 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/config"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"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()
|
|
}
|
|
|
|
// 使用事务同时删除设备记录和关联的认证方式
|
|
err := l.svcCtx.UserModel.Transaction(l.ctx, func(db *gorm.DB) error {
|
|
// 删除设备记录
|
|
if err := l.svcCtx.UserModel.DeleteDevice(l.ctx, req.Id, db); err != nil {
|
|
return err
|
|
}
|
|
// 删除关联的 AuthMethod (type="device", identifier=device.Identifier)
|
|
if err := l.svcCtx.UserModel.DeleteUserAuthMethodByIdentifier(l.ctx, device.UserId, "device", device.Identifier, db); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseDeletedError), "delete user device transaction error: %v", err.Error())
|
|
}
|
|
return nil
|
|
}
|