u

code/allapp/allapp-go/biz/handler/money/money_category_service.go

@@ -7,6 +7,7 @@ import (
 	"allapp/db"
 	"allapp/db/repo"
 	"allapp/utils/errorx"
+	"allapp/utils/idgen"
 	"allapp/utils/response"
 	"context"
 
@@ -16,11 +17,11 @@ import (
 	"github.com/cloudwego/hertz/pkg/protocol/consts"
 )
 
-// QueryUserMoneyCategory .
+// List .
 // @router /money/category [GET]
-func QueryUserMoneyCategory(ctx context.Context, c *app.RequestContext) {
+func List(ctx context.Context, c *app.RequestContext) {
 	var err error
-	var req money.QueryMoneyCategoryReq
+	var req money.ListMoneyCategoryReq
 	err = c.BindAndValidate(&req)
 	if err != nil {
 		c.String(consts.StatusBadRequest, err.Error())
@@ -37,7 +38,7 @@ func QueryUserMoneyCategory(ctx context.Context, c *app.RequestContext) {
 	}
 
 	// 构造返回数据，切片类型是 []*UserMoneyCategory
-	resp := &money.QueryMoneyCategoryResp{
+	resp := &money.ListMoneyCategoryResp{
 		Categories: make([]*money.MoneyCategory, 0, len(moneyUserCategories)),
 	}
 
@@ -116,3 +117,48 @@ func UpdateMoneyCategorySort(ctx context.Context, c *app.RequestContext) {
 	// 返回成功
 	response.Success(c).Send()
 }
+
+// CreateMoneyCategory .
+// @router /money/category/create [POST]
+func CreateMoneyCategory(ctx context.Context, c *app.RequestContext) {
+	var err error
+	var req money.CreateMoneyCategoryReq
+	err = c.BindAndValidate(&req)
+	if err != nil {
+		c.String(consts.StatusBadRequest, err.Error())
+		return
+	}
+
+	resp := new(money.CreateMoneyCategoryResp)
+
+	ID := idgen.NextId()
+
+	params := repo.InsertMoneyUserCategoryParams{
+		ID:     ID,
+		UserID: middleware.GetUserID(ctx),
+		Name:   req.Name,
+		Icon:   req.Icon,
+		Type:   req.Type,
+		// 默认排序号为ID
+		SortNumber: 0,
+	}
+
+	// 插入
+	db.WithTx(ctx, func(q *repo.Queries) error {
+		err = q.InsertMoneyUserCategory(ctx, params)
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+
+	if err != nil {
+		errorx.AddError(c, err)
+		return
+	}
+
+	resp.Id = ID
+
+	response.Success(c).Data(resp).Send()
+}
+