change layout

services/user/internal/logic/registerLogic.go

@@ -0,0 +1,98 @@
+package logic
+
+import (
+	"context"
+	"lone-services/pkg/log"
+	"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"
+)
+
+type RegisterLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterLogic {
+	return &RegisterLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *RegisterLogic) Register(in *user.RegisterReq) (*user.Response, error) {
+	var v validator.RegisterValidator
+	if msg := validate.ValidateFromProto(in, &v); msg != utils.StringEmpty {
+		return outResponse(utils.ErrorParams, msg), nil
+	}
+
+	openid := v.Openid
+	if openid == utils.StringEmpty {
+		openid = "mock_" + utils.MD5Encrypt(v.Mobile+utils.Now().String())
+	}
+
+	var exist dao.UserExist
+	err := model.UserModel{}.Init().GetOne(modelbase.Params{
+		Eq: map[string]string{"openid": openid},
+	}, &exist)
+	if err != nil {
+		log.Errorf("register check openid: %v", err)
+		return failResponse(utils.Fail), nil
+	}
+	if exist.Id > utils.NumberZero {
+		return failResponse(utils.ErrorExist), nil
+	}
+
+	encryptMobile, cErr := utils.Crypto{}.AESEncryptECB(v.Mobile)
+	if cErr != nil {
+		log.Errorf("register encrypt mobile: %v", cErr)
+		return failResponse(utils.ErrorEncryptAesError), nil
+	}
+
+	add := dao.UserCreate{
+		Openid:             openid,
+		Username:           v.Username,
+		Nickname:           v.Nickname,
+		Avatar:             v.Avatar,
+		Mobile:             encryptMobile,
+		Gender:             uint8(v.Gender),
+		Birthday:           v.Birthday,
+		AppId:              v.AppId,
+		Status:             utils.NumberOne,
+		IsRegistered:       utils.NumberOne,
+		IsFaceVerified:     utils.NumberTwo,
+		IsProfileCompleted: utils.NumberTwo,
+		OnTrialNum:         utils.NumberOne,
+	}
+	err = model.UserModel{}.Init().Create(&add)
+	if err != nil {
+		log.Errorf("register create: %v", err)
+		return failResponse(utils.Fail), nil
+	}
+
+	session := toSession(dao.UserInfo{
+		Id:       add.Id,
+		Openid:   add.Openid,
+		Username: add.Username,
+		Nickname: add.Nickname,
+		Avatar:   add.Avatar,
+		Gender:   add.Gender,
+		Birthday: add.Birthday,
+		AppId:    add.AppId,
+		Status:   add.Status,
+	}, v.Mobile)
+	ret := buildToken(session)
+	if status := setLogin(ret.Token, ret.Refresh, session); status.Code != utils.Ok.Code {
+		return failResponse(status), nil
+	}
+	return okResponse(ret), nil
+}