62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
"lone-services/pkg/modelbase"
|
|
"lone-services/pkg/utils"
|
|
"lone-services/services/equipment/internal/dao"
|
|
"lone-services/services/equipment/internal/model"
|
|
"strconv"
|
|
|
|
"lone-services/rpc/equipment/pb"
|
|
"lone-services/services/equipment/internal/svc"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
type InternalUserEditNumberLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
}
|
|
|
|
func NewInternalUserEditNumberLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InternalUserEditNumberLogic {
|
|
return &InternalUserEditNumberLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
// 更新数量
|
|
func (l *InternalUserEditNumberLogic) InternalUserEditNumber(in *equipment.InternalUserInfoRequest) (*equipment.InternalResponse, error) {
|
|
if in.UserId < utils.NumberOne || in.Number < utils.NumberOne {
|
|
return nil, status.Error(codes.InvalidArgument, "参数不全")
|
|
}
|
|
|
|
var user dao.MaskUserCreate
|
|
w := modelbase.Params{Eq: map[string]string{
|
|
"user_id": strconv.FormatInt(in.UserId, utils.NumberTen),
|
|
}}
|
|
err := model.MaskUserModel{}.Init().GetOne(w, &user)
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
return nil, status.Error(codes.InvalidArgument, "错误")
|
|
}
|
|
if user.Id < utils.NumberOne {
|
|
return nil, status.Error(codes.InvalidArgument, "用户已存在")
|
|
}
|
|
|
|
user.UserId = in.UserId
|
|
user.OnTrialNum = int8(in.Number)
|
|
_, err = model.MaskUserModel{}.Init().Edit(w, &user)
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
return nil, status.Error(codes.InvalidArgument, "错误")
|
|
}
|
|
|
|
return &equipment.InternalResponse{Id: user.Id}, nil
|
|
}
|