66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
|
package tokenservicelogic
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"ark-permission/gen_result/pb/permission"
|
|||
|
|
"ark-permission/internal/svc"
|
|||
|
|
ers "code.30cm.net/wanderland/library-go/errors"
|
|||
|
|
"context"
|
|||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type CancelTokenLogic struct {
|
|||
|
|
ctx context.Context
|
|||
|
|
svcCtx *svc.ServiceContext
|
|||
|
|
logx.Logger
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewCancelTokenLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CancelTokenLogic {
|
|||
|
|
return &CancelTokenLogic{
|
|||
|
|
ctx: ctx,
|
|||
|
|
svcCtx: svcCtx,
|
|||
|
|
Logger: logx.WithContext(ctx),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type cancelTokenReq struct {
|
|||
|
|
Token string `json:"token" validate:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CancelToken 取消 Token，也包含他裡面的 One Time Toke
|
|||
|
|
func (l *CancelTokenLogic) CancelToken(in *permission.CancelTokenReq) (*permission.OKResp, error) {
|
|||
|
|
// 驗證所需
|
|||
|
|
if err := l.svcCtx.Validate.ValidateAll(&cancelTokenReq{
|
|||
|
|
Token: in.GetToken(),
|
|||
|
|
}); err != nil {
|
|||
|
|
return nil, ers.InvalidFormat(err.Error())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
claims, err := parseClaims(in.GetToken(), l.svcCtx.Config.Token.Secret, false)
|
|||
|
|
if err != nil {
|
|||
|
|
logx.WithCallerSkip(1).WithFields(
|
|||
|
|
logx.Field("func", "parseClaims"),
|
|||
|
|
).Error(err.Error())
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
token, err := l.svcCtx.TokenRedisRepo.GetAccessTokenByID(l.ctx, claims.ID())
|
|||
|
|
if err != nil {
|
|||
|
|
logx.WithCallerSkip(1).WithFields(
|
|||
|
|
logx.Field("func", "TokenRedisRepo.GetByAccess"),
|
|||
|
|
logx.Field("claims", claims),
|
|||
|
|
).Error(err.Error())
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = l.svcCtx.TokenRedisRepo.Delete(l.ctx, token)
|
|||
|
|
if err != nil {
|
|||
|
|
logx.WithCallerSkip(1).WithFields(
|
|||
|
|
logx.Field("func", "TokenRedisRepo.Delete"),
|
|||
|
|
logx.Field("req", token),
|
|||
|
|
).Error(err.Error())
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &permission.OKResp{}, nil
|
|||
|
|
}
|