add check mask and change bind name

2026-10-21 13:56:23 +08:00
parent 746bde71de
commit 80644b89a6
14 changed files with 1009 additions and 171 deletions
@@ -0,0 +1,60 @@
+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 InternalUserAddLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewInternalUserAddLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InternalUserAddLogic {
+	return &InternalUserAddLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+// 添加用户信息
+func (l *InternalUserAddLogic) InternalUserAdd(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.NumberZero {
+		return nil, status.Error(codes.InvalidArgument, "用户已存在")
+	}
+	user.UserId = in.UserId
+	user.OnTrialNum = int8(in.Number)
+	err = model.MaskUserModel{}.Init().Create(&user)
+	if err != nil {
+		l.Logger.Error(err)
+		return nil, status.Error(codes.InvalidArgument, "错误")
+	}
+
+	return &equipment.InternalResponse{Id: user.Id}, nil
+}