wwweww
87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
"juwan-backend/app/snowflake/rpc/snowflake"
|
|
"juwan-backend/app/users/rpc/internal/svc"
|
|
"juwan-backend/app/users/rpc/pb"
|
|
"juwan-backend/common/redisx"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"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 mustNewRandomNickname() string {
|
|
bytes := make([]byte, 5)
|
|
_, err := rand.Read(bytes)
|
|
if err != nil {
|
|
return "NewUser"
|
|
}
|
|
nickname := strings.Builder{}
|
|
nickname.WriteString("user_")
|
|
nickname.WriteString(hex.EncodeToString(bytes))
|
|
return nickname.String()
|
|
}
|
|
|
|
func (l *RegisterLogic) Register(in *pb.RegisterReq) (*pb.RegisterResp, error) {
|
|
if in.Username == "" || in.Passwd == "" {
|
|
logx.Error("invalid input")
|
|
return nil, errors.New("invalid input")
|
|
}
|
|
|
|
redisKey := fmt.Sprintf(redisx.VCODE_KEY_PREFIX, in.RequestId, redisx.SCENE_REG, in.Email)
|
|
vcode, err := l.svcCtx.RedisCluster.Get(l.ctx, redisKey).Result()
|
|
logx.Infof("vcode:%s, err:%v", vcode, err)
|
|
if err != nil {
|
|
logx.Error("invalid verification code")
|
|
return nil, errors.New("invalid verification code")
|
|
}
|
|
|
|
code, err := strconv.ParseInt(vcode, 10, 32)
|
|
if err != nil || int32(code) != in.Vcode {
|
|
logx.Error("invalid verification code")
|
|
return nil, errors.New("invalid verification code")
|
|
}
|
|
|
|
resp, err := l.svcCtx.Snowflake.NextId(l.ctx, &snowflake.NextIdReq{})
|
|
if err != nil {
|
|
return nil, errors.New("generate user ID failed")
|
|
}
|
|
|
|
_, err = l.svcCtx.UsersModelRW.Create().
|
|
SetID(resp.Id).
|
|
SetUsername(in.Username).
|
|
SetPasswordHash(in.Passwd).
|
|
SetPhone(in.Phone).
|
|
SetEmail(in.Email).
|
|
SetNickname(mustNewRandomNickname()).
|
|
Save(l.ctx)
|
|
if err != nil {
|
|
logx.Errorf("create user err:%v", err)
|
|
return nil, err
|
|
}
|
|
|
|
return &pb.RegisterResp{
|
|
Res: "user registered successfully",
|
|
}, nil
|
|
}
|