package logic import ( "context" "strconv" "lone-services/pkg/modelbase" "lone-services/pkg/utils" "lone-services/pkg/validate" user "lone-services/rpc/user/pb" "lone-services/services/user/internal/dao" "lone-services/services/user/internal/model" "lone-services/services/user/internal/svc" "lone-services/services/user/validator" "github.com/zeromicro/go-zero/core/logx" "google.golang.org/grpc/codes" "google.golang.org/grpc/status" ) type InternalRevokeBizClientLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger } func NewInternalRevokeBizClientLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InternalRevokeBizClientLogic { return &InternalRevokeBizClientLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *InternalRevokeBizClientLogic) InternalRevokeBizClient(in *user.InternalRevokeBizClientReq) (*user.InternalRevokeBizClientData, error) { var req validator.InternalRevokeBizClientValidator if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty { return nil, status.Error(codes.InvalidArgument, msg) } clientModel := model.UserClientModel{}.Init() var row dao.UserClient if err := clientModel.GetOne(modelbase.Params{ Eq: map[string]string{ "user_id": strconv.FormatInt(req.UserId, utils.NumberTen), "client_code": req.ClientCode, }, }, &row); err != nil { l.Errorf("internal revoke biz client get: %v", err) return nil, status.Error(codes.Internal, utils.Fail.Msg) } if row.Id < utils.NumberOne { return &user.InternalRevokeBizClientData{Revoked: false}, nil } if row.Status == dao.StatusDisabled { return &user.InternalRevokeBizClientData{Revoked: true}, nil } if _, err := clientModel.Edit(modelbase.Params{ Eq: map[string]string{"id": strconv.FormatInt(row.Id, utils.NumberTen)}, }, map[string]interface{}{"status": dao.StatusDisabled}); err != nil { l.Errorf("internal revoke biz client edit: %v", err) return nil, status.Error(codes.Internal, utils.Fail.Msg) } return &user.InternalRevokeBizClientData{Revoked: true}, nil }