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 }