feat: user get mobile

services/analysis/internal/logic/analysis/internalSetRegisteredLogic.go

@@ -0,0 +1,61 @@
+package analysislogic
+
+import (
+	"context"
+	"strconv"
+
+	"lone-services/pkg/modelbase"
+	"lone-services/pkg/utils"
+	analysis "lone-services/rpc/analysis/pb"
+	"lone-services/services/analysis/internal/logic/common"
+	"lone-services/services/analysis/internal/model"
+	"lone-services/services/analysis/internal/svc"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+type InternalSetRegisteredLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewInternalSetRegisteredLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InternalSetRegisteredLogic {
+	return &InternalSetRegisteredLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *InternalSetRegisteredLogic) InternalSetRegistered(in *analysis.InternalSetRegisteredReq) (*analysis.InternalSetRegisteredData, error) {
+	if in == nil || in.Id < utils.NumberOne {
+		return nil, status.Error(codes.InvalidArgument, "用户ID不能为空")
+	}
+
+	row, err := common.LoadAnalysisUser(in.Id)
+	if err != nil {
+		l.Errorf("internal set registered get: %v", err)
+		return nil, status.Error(codes.Internal, utils.Fail.Msg)
+	}
+	if row == nil {
+		return nil, status.Error(codes.NotFound, utils.ErrorNotFund.Msg)
+	}
+	if row.IsRegistered == uint8(utils.StatusOk) {
+		return &analysis.InternalSetRegisteredData{Ok: true}, nil
+	}
+
+	_, err = model.UserModel{}.Init().Edit(modelbase.Params{
+		Eq: map[string]string{"id": strconv.FormatInt(in.Id, utils.NumberTen)},
+	}, map[string]interface{}{
+		"is_registered": uint8(utils.StatusOk),
+	})
+	if err != nil {
+		l.Errorf("internal set registered edit: %v", err)
+		return nil, status.Error(codes.Internal, utils.Fail.Msg)
+	}
+
+	return &analysis.InternalSetRegisteredData{Ok: true}, nil
+}