110 lines
4.1 KiB
Go
110 lines
4.1 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
"tyc-server/app/user/cmd/api/internal/svc"
|
|
"tyc-server/app/user/cmd/api/internal/types"
|
|
"tyc-server/app/user/model"
|
|
jwtx "tyc-server/common/jwt"
|
|
"tyc-server/common/tool"
|
|
"tyc-server/common/xerr"
|
|
"tyc-server/pkg/lzkit/crypto"
|
|
"tyc-server/pkg/lzkit/lzUtils"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/zeromicro/go-zero/core/stores/redis"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type RegisterLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterLogic {
|
|
return &RegisterLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *RegisterLogic) Register(req *types.RegisterReq) (resp *types.RegisterResp, err error) {
|
|
secretKey := l.svcCtx.Config.Encrypt.SecretKey
|
|
encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "手机注册, 加密手机号失败: %+v", err)
|
|
}
|
|
// 检查手机号是否在一分钟内已发送过验证码
|
|
redisKey := fmt.Sprintf("%s:%s", "register", encryptedMobile)
|
|
cacheCode, err := l.svcCtx.Redis.Get(redisKey)
|
|
if err != nil {
|
|
if errors.Is(err, redis.Nil) {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "手机注册, 验证码过期: %s", encryptedMobile)
|
|
}
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机注册, 读取验证码redis缓存失败, mobile: %s, err: %+v", encryptedMobile, err)
|
|
}
|
|
if cacheCode != req.Code {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "手机注册, 验证码不正确: %s", encryptedMobile)
|
|
}
|
|
hasUser, findUserErr := l.svcCtx.UserModel.FindOneByMobile(l.ctx, encryptedMobile)
|
|
if findUserErr != nil && findUserErr != model.ErrNotFound {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机注册, 读取数据库获取用户失败, mobile%s, err: %+v", encryptedMobile, err)
|
|
}
|
|
if hasUser != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("该手机号码已注册"), "手机注册, 手机号码已注册, mobile:%s", encryptedMobile)
|
|
}
|
|
var userId int64
|
|
if transErr := l.svcCtx.UserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error {
|
|
user := new(model.User)
|
|
user.Mobile = encryptedMobile
|
|
if user.Nickname.Valid && user.Nickname.String != "" {
|
|
user.Nickname = sql.NullString{
|
|
String: encryptedMobile,
|
|
Valid: true,
|
|
}
|
|
}
|
|
if len(req.Password) > 0 {
|
|
user.Password = lzUtils.StringToNullString(tool.Md5ByString(req.Password))
|
|
}
|
|
insertResult, userInsertErr := l.svcCtx.UserModel.Insert(ctx, session, user)
|
|
if userInsertErr != nil {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机注册, 数据库插入新用户失败, mobile%s, err: %+v", encryptedMobile, err)
|
|
}
|
|
lastId, lastInsertIdErr := insertResult.LastInsertId()
|
|
if lastInsertIdErr != nil {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机注册, 获取新用户ID失败, err:%+v, user:%+v", lastInsertIdErr, user)
|
|
}
|
|
userId = lastId
|
|
|
|
userAuth := new(model.UserAuth)
|
|
userAuth.UserId = lastId
|
|
userAuth.AuthKey = encryptedMobile
|
|
userAuth.AuthType = model.UserAuthTypeAppMobile
|
|
if _, userAuthInsertErr := l.svcCtx.UserAuthModel.Insert(ctx, session, userAuth); userAuthInsertErr != nil {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机注册, 数据库插入用户认证失败, err:%+v", userAuthInsertErr)
|
|
}
|
|
return nil
|
|
}); transErr != nil {
|
|
return nil, transErr
|
|
}
|
|
|
|
token, generaErr := jwtx.GenerateJwtToken(userId, l.svcCtx.Config.JwtAuth.AccessSecret, l.svcCtx.Config.JwtAuth.AccessExpire)
|
|
if generaErr != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "手机注册, 生成jwt token失败, userid: %d, err:%+v", userId, generaErr)
|
|
}
|
|
// 获取当前时间戳
|
|
now := time.Now().Unix()
|
|
return &types.RegisterResp{
|
|
AccessToken: token,
|
|
AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire,
|
|
RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter,
|
|
}, nil
|
|
}
|