83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
package money
|
|
|
|
import (
|
|
"base-go-v2/internal/db"
|
|
response "base-go-v2/internal/reponse"
|
|
"base-go-v2/internal/router"
|
|
"base-go-v2/internal/utils/idgen"
|
|
"base-go-v2/internal/utils/mapx"
|
|
"fmt"
|
|
)
|
|
|
|
func InitMoneyRouter(r *router.Router) {
|
|
group := r.Group("money")
|
|
group.GET("/category", getUserCategory)
|
|
//group := r.Group("auth")
|
|
}
|
|
|
|
// InitDefaultCategories 初始化系统默认分类到用户分类表
|
|
func InitDefaultCategories(userID int64) error {
|
|
fmt.Println("进入 InitDefaultCategories ===")
|
|
|
|
// 1. 查询系统分类表
|
|
sysCategories, err := db.FindAll("money_sys_category", "sort_number")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("系统分类数量: %d\n", len(sysCategories))
|
|
|
|
// 2. 构建批量插入数据
|
|
//userCategories := make([]mapx.M, 0, len(sysCategories))
|
|
userCategories := mapx.NewArr(len(sysCategories))
|
|
|
|
for i, row := range sysCategories {
|
|
m := mapx.New().SetKV(
|
|
"id", idgen.NextId(),
|
|
"user_id", userID,
|
|
"name", row.GetString("name"),
|
|
"icon", row.GetString("icon"),
|
|
"type", row.GetInt("type"),
|
|
"sort_number", row.GetInt("sort_number"),
|
|
)
|
|
fmt.Printf("用户分类 %d: %+v\n", i+1, m)
|
|
userCategories = append(userCategories, m)
|
|
}
|
|
|
|
// 3. 批量插入到用户分类表
|
|
if _, err := db.InsertBatch("money_user_category", userCategories); err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println("初始化用户分类成功！")
|
|
return nil
|
|
}
|
|
|
|
func getUserCategory(c *router.Context) error {
|
|
fmt.Println("请求进入了====")
|
|
userId := c.GetUserID()
|
|
|
|
// 查询当前用户的分类
|
|
userCategories, err := db.Find("money_user_category", mapx.New().Set("user_id", userId), "sort_number ASC")
|
|
if err != nil {
|
|
return c.AddError(err)
|
|
}
|
|
|
|
// 按 type 分组
|
|
result := map[string][]mapx.M{
|
|
"income": {},
|
|
"expense": {},
|
|
}
|
|
|
|
for _, cat := range userCategories {
|
|
switch cat.GetInt("type") {
|
|
case 0:
|
|
result["income"] = append(result["income"], cat)
|
|
case 1:
|
|
result["expense"] = append(result["expense"], cat)
|
|
}
|
|
}
|
|
|
|
return response.Success(c).Data(result).Send()
|
|
}
|