This commit is contained in:
2024-11-19 21:09:25 +08:00
parent 4b856887a0
commit 38253f0ce5

View File

@@ -673,15 +673,6 @@ func NewSchema(config Config) (graphql.Schema, error) {
).As("row_num")) ).As("row_num"))
} }
// 如果没有外部排序则使用指定排序(正则sort只能是字母数字下划下)
if p.Args["text"] == nil && p.Args["similar"] == nil && p.Args["interest"] == nil {
sort := regexp.MustCompile(`[^a-zA-Z0-9_]`).ReplaceAllString(p.Args["sort"].(string), "")
order := regexp.MustCompile(`[^a-zA-Z0-9_]`).ReplaceAllString(p.Args["order"].(string), "")
query = query.Select("web_images.id", goqu.L(
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_images.%s %s)", sort, order),
).As("row_num"))
}
// 筛选:时间段 // 筛选:时间段
applyTimeCondition := func(name string, str string) { applyTimeCondition := func(name string, str string) {
parts := strings.Split(str, "-") parts := strings.Split(str, "-")
@@ -874,10 +865,19 @@ func NewSchema(config Config) (graphql.Schema, error) {
} }
} }
// 如果没有外部排序则使用指定排序(正则sort只能是字母数字下划下)
if p.Args["text"] == nil && p.Args["similar"] == nil && p.Args["interest"] == nil {
sort := regexp.MustCompile(`[^a-zA-Z0-9_]`).ReplaceAllString(p.Args["sort"].(string), "")
order := regexp.MustCompile(`[^a-zA-Z0-9_]`).ReplaceAllString(p.Args["order"].(string), "")
query = query.Select("web_images.id", goqu.L(
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_images.%s %s)", sort, order),
).As("row_num"))
}
// 取所有数据的前N条 // 取所有数据的前N条
sql, _, _ := query.Where(goqu.Ex{"article_category_top_id": 22}).ToSQL() sql, _, _ := query.Where(goqu.Ex{"article_category_top_id": 22}).ToSQL()
// 取所有数据的前N条 // 遊標截取篩選結果集的前N条
var cursor string var cursor string
if p.Args["after"] != nil { if p.Args["after"] != nil {
cursor = fmt.Sprintf(`WHERE row_num > (SELECT row_num FROM RankedArticles WHERE RankedArticles.id = %d)`, p.Args["after"].(int)) cursor = fmt.Sprintf(`WHERE row_num > (SELECT row_num FROM RankedArticles WHERE RankedArticles.id = %d)`, p.Args["after"].(int))
@@ -892,16 +892,15 @@ func NewSchema(config Config) (graphql.Schema, error) {
//fmt.Println("SQL:", sql) //fmt.Println("SQL:", sql)
// 優化查詢: 如果是第一頁(未傳入遊標), 可以在分頁前(窗口函數)立即截取長度以大幅降低排序選取耗時
sql = fmt.Sprintf(` sql = fmt.Sprintf(`
WITH RankedArticles AS (%s) WITH RankedArticles AS (%s)
SELECT * FROM web_images INNER JOIN( SELECT * FROM web_images INNER JOIN(
SELECT id, row_num FROM RankedArticles %s SELECT id, row_num FROM RankedArticles %s
) AS LimitedRanked ON LimitedRanked.id = web_images.id ) AS LimitedRanked ON LimitedRanked.id = web_images.id
ORDER BY LimitedRanked.row_num LIMIT %d ORDER BY LimitedRanked.row_num ASC LIMIT %d
`, sql, cursor, limit) `, sql, cursor, limit)
//fmt.Println("cursor:", cursor, limit)
if err := db.Raw(sql).Scan(&images).Error; err != nil { if err := db.Raw(sql).Scan(&images).Error; err != nil {
fmt.Println("获取图像列表失败", err) fmt.Println("获取图像列表失败", err)
return nil, err return nil, err