95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package l_ai_category
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"gitea.cdlsxd.cn/self-tools/l_ai_category/utils_gorm"
|
|
"testing"
|
|
)
|
|
|
|
const (
|
|
dns = "root:lansexiongdi6,@tcp(47.97.27.195:3306)/report?parseTime=True&loc=Local"
|
|
driver = "mysql"
|
|
table = "goods_category"
|
|
modelType = "deepseek-v3-250324"
|
|
)
|
|
|
|
func TestCategory(t *testing.T) {
|
|
path := getPath()
|
|
res, err := GetCategory(context.Background(), "MM 便携装湿厕纸 40包*10片@默认", "03320e58-6a0b-4061-a22b-902039f2190d", modelType, path)
|
|
t.Log(res, err)
|
|
}
|
|
|
|
func TestRem(t *testing.T) {
|
|
|
|
res, err := Test(context.Background(), "914ccf1d-c002-4fad-a431-f291f5e0d2ad", modelType)
|
|
t.Log(res, err)
|
|
}
|
|
|
|
//func TestFlush(t *testing.T) {
|
|
// catString := string(CategoryToJsonByte())
|
|
// res, err := FlushCate(context.Background(), "914ccf1d-c002-4fad-a431-f291f5e0d2ad", modelType, catString)
|
|
// t.Log(res, err)
|
|
//}
|
|
//
|
|
//func TestGetCateGoryJson(t *testing.T) {
|
|
// cateJsonByte := CategoryToJsonByte()
|
|
//
|
|
// t.Log(string(cateJsonByte))
|
|
//}
|
|
|
|
//func CategoryToJsonByte() (cateJsonByte []byte) {
|
|
//
|
|
// cateLevelPath := getPath()
|
|
// cate := cateIteration(cateLevelPath[0], cateIteration(cateLevelPath[1], cateLevelPath[2]))[0]
|
|
// cateJsonByte, _ = json.Marshal(cate)
|
|
// return
|
|
//}
|
|
|
|
func getPath() []map[int][]*CategoryDic {
|
|
cates := cateAll()
|
|
var (
|
|
cateLevelPath = make([]map[int][]*CategoryDic, 3)
|
|
)
|
|
for _, v := range cates {
|
|
if cateLevelPath[v.Level-1] == nil {
|
|
cateLevelPath[v.Level-1] = make(map[int][]*CategoryDic)
|
|
}
|
|
if _, ex := cateLevelPath[v.Level-1][v.Pid]; !ex {
|
|
cateLevelPath[v.Level-1][v.Pid] = make([]*CategoryDic, 0)
|
|
}
|
|
cateLevelPath[v.Level-1][v.Pid] = append(cateLevelPath[v.Level-1][v.Pid], &CategoryDic{Category: v})
|
|
}
|
|
return cateLevelPath
|
|
}
|
|
|
|
//func cateIteration(parent map[int][]*CategoryDic, child map[int][]*CategoryDic) map[int][]*CategoryDic {
|
|
// for _, v := range parent {
|
|
// for _, vv := range v {
|
|
// vv.Children = child[vv.Id]
|
|
// }
|
|
// }
|
|
// return parent
|
|
//}
|
|
|
|
func cateAll() (data []Category) {
|
|
var (
|
|
row Category
|
|
)
|
|
db, err, clean := utils_gorm.DB(driver, dns)
|
|
defer clean()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
rows, _ := db.Raw(fmt.Sprintf("SELECT id,name,level,pid FROM `%s` where `type`=2", table)).Rows()
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
err = db.ScanRows(rows, &row)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
data = append(data, row)
|
|
}
|
|
return
|
|
}
|