139 lines
2.7 KiB
Go
139 lines
2.7 KiB
Go
// Code generated by hertz generator.
|
|
|
|
package auth
|
|
|
|
import (
|
|
auth "allapp/biz/model/auth"
|
|
"allapp/db"
|
|
"allapp/db/repo"
|
|
"allapp/utils"
|
|
"allapp/utils/errorx"
|
|
"allapp/utils/idgen"
|
|
"allapp/utils/pgtypex"
|
|
"allapp/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 {
|
|
|
|
q := db.Queries.WithTx(tx)
|
|
|
|
sysCategories, err := q.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,
|
|
})
|
|
}
|
|
|
|
// ⚠️ 注意：这里是 DeleteMoneyCategoryByUserId，不是 ById（你代码里疑似写错）
|
|
if err := q.DeleteMoneyCategoryByUserId(ctx, userId); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := q.BatchInsertMoneyUserCategories(ctx, userCategories); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|