129 lines
2.6 KiB
Go
129 lines
2.6 KiB
Go
// Code generated by hertz generator.
|
|
|
|
package auth
|
|
|
|
import (
|
|
auth "cn/oneao/base-go/biz/model/auth"
|
|
"cn/oneao/base-go/db"
|
|
"cn/oneao/base-go/db/repo"
|
|
"cn/oneao/base-go/utils"
|
|
"cn/oneao/base-go/utils/errorx"
|
|
"cn/oneao/base-go/utils/idgen"
|
|
"cn/oneao/base-go/utils/pgtypex"
|
|
"cn/oneao/base-go/utils/response"
|
|
"context"
|
|
|
|
"github.com/cloudwego/hertz/pkg/app"
|
|
"github.com/cloudwego/hertz/pkg/protocol/consts"
|
|
"github.com/jackc/pgx/v5"
|
|
)
|
|
|
|
// RegisterDefault .
|
|
// @router /auth/register/default [POST]
|
|
func RegisterDefault(ctx context.Context, c *app.RequestContext) {
|
|
var err error
|
|
var req auth.RegisterDefaultReq
|
|
err = c.BindAndValidate(&req)
|
|
if err != nil {
|
|
c.String(consts.StatusBadRequest, err.Error())
|
|
return
|
|
}
|
|
|
|
account := req.Account
|
|
password := req.Password
|
|
|
|
_, err = db.Queries.GetUserByAccount(ctx, account)
|
|
if err != nil {
|
|
if errorx.IsNotFound(err) {
|
|
// 忽略
|
|
} else {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
} else {
|
|
response.Fail(c).Message("该账号已被注册").Send()
|
|
return
|
|
}
|
|
|
|
// 开启事务
|
|
tx, err := db.DB.BeginTx(ctx, pgx.TxOptions{})
|
|
if err != nil {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
|
|
defer func() {
|
|
if err != nil {
|
|
err := tx.Rollback(ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
} else {
|
|
err := tx.Commit(ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
generateCode, err := utils.GenerateCode(8)
|
|
if err != nil {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
|
|
// 插入用户
|
|
params := repo.InsertUserParams{
|
|
ID: idgen.NextId(),
|
|
Account: account,
|
|
Password: password,
|
|
Username: pgtypex.StringToText("a_" + generateCode),
|
|
}
|
|
|
|
insert, err := db.Queries.WithTx(tx).InsertUser(ctx, params)
|
|
if err != nil || insert == 0 {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
|
|
// 初始化用户分类
|
|
err = initMoneyCategoryTx(ctx, tx, params.ID)
|
|
if err != nil {
|
|
errorx.AddError(c, err)
|
|
return
|
|
}
|
|
|
|
// 注册成功
|
|
response.Success(c).Message("注册成功").Send()
|
|
}
|
|
|
|
func initMoneyCategoryTx(ctx context.Context, tx pgx.Tx, userId int64) error {
|
|
sysCategories, err := db.Queries.ListMoneySysCategory(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(sysCategories) == 0 {
|
|
return nil
|
|
}
|
|
|
|
userCategories := make([]repo.BatchInsertMoneyUserCategoriesParams, 0, len(sysCategories))
|
|
for _, sysCat := range sysCategories {
|
|
userCategories = append(userCategories, repo.BatchInsertMoneyUserCategoriesParams{
|
|
ID: idgen.NextId(),
|
|
UserID: userId,
|
|
Name: sysCat.Name,
|
|
Icon: sysCat.Icon,
|
|
Type: sysCat.Type,
|
|
SortNumber: sysCat.SortNumber,
|
|
})
|
|
}
|
|
|
|
_, err = db.Queries.WithTx(tx).BatchInsertMoneyUserCategories(ctx, userCategories)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|