sample/internal/logic/register/do.go

82 lines
2.1 KiB
Go

/**
* @Author: ZhaoYadong
* @Date: 2024-02-27 21:09:45
* @LastEditors: ZhaoYadong
* @LastEditTime: 2024-02-28 11:50:05
* @FilePath: /server/Users/edy/go/src/sample/internal/logic/register/do.go
*/
package register
import (
"git.apinb.com/bsm-sdk/core/errcode"
"git.apinb.com/bsm-sdk/core/print"
"git.apinb.com/bsm-sdk/core/types"
"git.apinb.com/bsm-sdk/core/utils"
_vars "git.apinb.com/bsm-sdk/core/vars"
"git.apinb.com/bsm-sdk/sample/internal/config"
"git.apinb.com/bsm-sdk/sample/internal/impl"
"git.apinb.com/bsm-sdk/sample/internal/logic/common"
"git.apinb.com/bsm-sdk/sample/internal/vars"
pb "git.apinb.com/bsm-sdk/sample/pb"
"golang.org/x/crypto/bcrypt"
)
// Do .
func Do(in *pb.RegisterRequest) (*pb.RegisterReply, error) {
var salt string
if in.Password != "" {
salt = utils.UUID()
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(in.Password+salt), bcrypt.MinCost)
if err != nil {
return nil, errcode.ErrInternal
}
in.Password = string(hashedPassword)
}
var account = in.Account
if account == "" {
account = in.Phone
}
pa := models.sampleAccount{
Std_IICUDS: types.Std_IICUDS{
Identity: utils.UUID(),
Status: vars.Status_Normal,
},
Account: account,
Phone: in.Phone,
Password: in.Password,
Salt: salt,
AgencyId: uint(in.AgencyId),
StaffId: uint(in.StaffId),
OwnerId: uint(in.OwnerId),
}
// 插入sample 基础表
if err := impl.DBService.Create(&pa).Error; err != nil {
print.Error("create sample account and password extend by data %+v error:%v", pa, err)
return nil, errcode.ErrDB
}
extend := map[string]string{
"rights": pa.Rights,
}
token, err := common.GenerateTokenAes(uint(pa.ID), pa.Identity, "", pa.Rights, extend)
if err != nil {
return nil, err
}
//save token to cache.
err = impl.RedisService.Client.Set(impl.RedisService.Ctx, config.Spec.Token.Prefix+pa.Identity, token, _vars.JwtExpire).Err()
if err != nil {
print.Error("Set redis cache by key %v error:%v", vars.TokenPrefix+pa.Identity, err)
return nil, errcode.ErrInternal
}
return &pb.RegisterReply{
Id: int64(pa.ID),
Identity: pa.Identity,
Token: token,
Extend: extend,
}, nil
}