补空参

This commit is contained in:
2024-11-17 19:05:14 +08:00
parent c1f953ce72
commit 678e05ede4
2 changed files with 5 additions and 5 deletions

View File

@@ -92,14 +92,13 @@ func PutPraises(page int) error {
// 获取推荐ID // 获取推荐ID
func GetRecommend(user_id int, category string) ([]int, error) { func GetRecommend(user_id int, category string) ([]int, error) {
var ctx context.Context = context.Background() var ctx context.Context = context.Background()
var ids []int
data, err := gorse.GetItemLatestWithCategory(ctx, fmt.Sprintf("%d", user_id), category, 100, 0) data, err := gorse.GetItemLatestWithCategory(ctx, fmt.Sprintf("%d", user_id), category, 100, 0)
if err != nil { if err != nil {
fmt.Println("获取推荐失败", err) fmt.Println("获取推荐失败", err)
return nil, err return ids, err
} }
var ids []int
for _, item := range data { for _, item := range data {
id, _ := strconv.Atoi(item.Id) id, _ := strconv.Atoi(item.Id)
ids = append(ids, id) ids = append(ids, id)

View File

@@ -635,16 +635,17 @@ func NewSchema(config Config) (graphql.Schema, error) {
// 筛选:兴趣推荐 // 筛选:兴趣推荐
if p.Args["interest"] != nil { if p.Args["interest"] != nil {
fmt.Println("interest:", p.Args["interest"])
user_id := p.Args["interest"].(int) user_id := p.Args["interest"].(int)
list, err := GetRecommend(user_id, "image") list, err := GetRecommend(user_id, "image")
if err != nil { if err != nil {
fmt.Println("GetRecommend 获取兴趣推荐失败", err) fmt.Println("GetRecommend 获取兴趣推荐失败", err)
return nil, err return map[string]interface{}{"list": []Image{}, "total": 0}, nil
} }
fmt.Println("Interest:", user_id, list)
if len(list) == 0 { if len(list) == 0 {
return map[string]interface{}{"list": []Image{}, "total": 0}, nil return map[string]interface{}{"list": []Image{}, "total": 0}, nil
} }
fmt.Println("Interest:", user_id, list)
query = query.Where(goqu.Ex{"web_images.id": goqu.Op{"in": list}}).Select("web_images.id", goqu.L( query = query.Where(goqu.Ex{"web_images.id": goqu.Op{"in": list}}).Select("web_images.id", goqu.L(
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY FIELD(%s, %s))", "web_images.id", regexp.MustCompile(`[\[\]]`).ReplaceAllString(strings.Join(strings.Fields(fmt.Sprint(list)), ", "), "")), fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY FIELD(%s, %s))", "web_images.id", regexp.MustCompile(`[\[\]]`).ReplaceAllString(strings.Join(strings.Fields(fmt.Sprint(list)), ", "), "")),
).As("row_num")) ).As("row_num"))