This commit is contained in:
2024-11-27 20:26:18 +08:00
parent a015fca246
commit bcfde1999c
2 changed files with 216 additions and 49 deletions

View File

@@ -569,6 +569,24 @@ 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), "")
query = query.Select("web_images.id", goqu.L(
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_images.%s %s)", sort, p.Args["order"]),
).As("row_num"))
}
// 取所有数据的前N条
sql, _, _ := query.Where(goqu.Ex{"article_category_top_id": 22}).ToSQL()
fmt.Println(sql)
// 遊標截取篩選結果集的前N条
var cursor string
if p.Args["after"] != nil {
cursor = fmt.Sprintf(`WHERE row_num > (SELECT row_num FROM RankedArticles WHERE RankedArticles.id = %d)`, p.Args["after"].(int))
}
var limit int = 10 var limit int = 10
if p.Args["first"] != nil { if p.Args["first"] != nil {
limit = p.Args["first"].(int) limit = p.Args["first"].(int)
@@ -576,6 +594,14 @@ func NewSchema(config Config) (graphql.Schema, error) {
limit = p.Args["last"].(int) limit = p.Args["last"].(int)
} }
sql = fmt.Sprintf(`
WITH RankedArticles AS (%s)
SELECT * FROM web_images INNER JOIN(
SELECT id, row_num FROM RankedArticles %s
) AS LimitedRanked ON LimitedRanked.id = web_images.id
ORDER BY LimitedRanked.row_num ASC LIMIT %d
`, sql, cursor, limit)
if err := db.Limit(limit).Where("category_top_id = 22").Find(&games).Error; err != nil { if err := db.Limit(limit).Where("category_top_id = 22").Find(&games).Error; err != nil {
return nil, err return nil, err
} }
@@ -586,6 +612,176 @@ func NewSchema(config Config) (graphql.Schema, error) {
}, err }, err
}, },
}, },
"works": &graphql.Field{
Name: "works",
Description: "作品列表",
Type: graphql.NewObject(graphql.ObjectConfig{
Name: "WorkConnection",
Description: "条件筛选作品列表",
Fields: graphql.Fields{
"list": &graphql.Field{Type: graphql.NewList(article), Description: "作品列表"},
//"total": &graphql.Field{Type: graphql.Int, Description: "作品总数"},
},
}),
Args: graphql.FieldConfigArgument{
"id": &graphql.ArgumentConfig{Type: graphql.Int, Description: "筛选作品中指定ID的"},
"title": &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选作品中含有指定标题的"},
"tags": &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选作品中含有指定标签的"},
"create_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选作品中创建时间等于指定值的"},
"update_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选作品中更新时间等于指定值的"},
},
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
var works []Work
var total int
var err error
var query = goqu.Dialect("mysql").From("web_article")
// 筛选条件
for _, format := range []string{"id", "style", "device", "orientation", "era", "category_id", "tags"} {
if p.Args[format] != nil {
query = query.Where(goqu.C(format).Eq(p.Args[format]))
}
}
// 排序条件
if p.Args["sort"] != nil {
if p.Args["order"].(string) == "ASC" {
query = query.Order(goqu.C(p.Args["sort"].(string)).Asc())
}
if p.Args["order"].(string) == "DESC" {
query = query.Order(goqu.C(p.Args["sort"].(string)).Desc())
}
}
// 如果没有外部排序则使用指定排序(正则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), "")
query = query.Select("web_images.id", goqu.L(
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_images.%s %s)", sort, p.Args["order"]),
).As("row_num"))
}
// 取所有数据的前N条
sql, _, _ := query.Where(goqu.Ex{"article_category_top_id": 1}).ToSQL()
fmt.Println(sql)
// 遊標截取篩選結果集的前N条
var cursor string
if p.Args["after"] != nil {
cursor = fmt.Sprintf(`WHERE row_num > (SELECT row_num FROM RankedArticles WHERE RankedArticles.id = %d)`, p.Args["after"].(int))
}
var limit int = 10
if p.Args["first"] != nil {
limit = p.Args["first"].(int)
} else if p.Args["last"] != nil {
limit = p.Args["last"].(int)
}
sql = fmt.Sprintf(`
WITH RankedArticles AS (%s)
SELECT * FROM web_images INNER JOIN(
SELECT id, row_num FROM RankedArticles %s
) AS LimitedRanked ON LimitedRanked.id = web_images.id
ORDER BY LimitedRanked.row_num ASC LIMIT %d
`, sql, cursor, limit)
if err := db.Limit(limit).Where("category_top_id = 1").Find(&works).Error; err != nil {
return nil, err
}
return map[string]interface{}{
"list": works,
"total": total,
}, err
},
},
"articles": &graphql.Field{
Name: "articles",
Description: "文章列表",
Type: graphql.NewObject(graphql.ObjectConfig{
Name: "ArticleConnection",
Description: "条件筛选文章列表",
Fields: graphql.Fields{
"list": &graphql.Field{Type: graphql.NewList(article), Description: "文章列表"},
"total": &graphql.Field{Type: graphql.Int, Description: "文章总数"},
},
}),
Args: graphql.FieldConfigArgument{
"id": &graphql.ArgumentConfig{Type: graphql.Int, Description: "筛选文章中指定ID的"},
"title": &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选文章中含有指定标题的"},
"tags": &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选文章中含有指定标签的"},
"create_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选文章中创建时间等于指定值的"},
"update_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选文章中更新时间等于指定值的"},
},
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
var articles []Article
var total int
var err error
var query = goqu.Dialect("mysql").From("web_article")
// 筛选条件
for _, format := range []string{"id", "style", "device", "orientation", "era", "category_id", "tags"} {
if p.Args[format] != nil {
query = query.Where(goqu.C(format).Eq(p.Args[format]))
}
}
// 排序条件
if p.Args["sort"] != nil {
if p.Args["order"].(string) == "ASC" {
query = query.Order(goqu.C(p.Args["sort"].(string)).Asc())
}
if p.Args["order"].(string) == "DESC" {
query = query.Order(goqu.C(p.Args["sort"].(string)).Desc())
}
}
// 如果没有外部排序则使用指定排序(正则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), "")
query = query.Select("web_images.id", goqu.L(
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_images.%s %s)", sort, p.Args["order"]),
).As("row_num"))
}
// 取所有数据的前N条
sql, _, _ := query.Where(goqu.Ex{"article_category_top_id": 9}).ToSQL()
fmt.Println(sql)
// 遊標截取篩選結果集的前N条
var cursor string
if p.Args["after"] != nil {
cursor = fmt.Sprintf(`WHERE row_num > (SELECT row_num FROM RankedArticles WHERE RankedArticles.id = %d)`, p.Args["after"].(int))
}
var limit int = 10
if p.Args["first"] != nil {
limit = p.Args["first"].(int)
} else if p.Args["last"] != nil {
limit = p.Args["last"].(int)
}
sql = fmt.Sprintf(`
WITH RankedArticles AS (%s)
SELECT * FROM web_images INNER JOIN(
SELECT id, row_num FROM RankedArticles %s
) AS LimitedRanked ON LimitedRanked.id = web_images.id
ORDER BY LimitedRanked.row_num ASC LIMIT %d
`, sql, cursor, limit)
if err := db.Limit(limit).Where("category_top_id = 9").Find(&articles).Error; err != nil {
return nil, err
}
return map[string]interface{}{
"list": articles,
"total": total,
}, err
},
},
"images": &graphql.Field{ "images": &graphql.Field{
Name: "images", Name: "images",
Description: "图像列表", Description: "图像列表",
@@ -1017,55 +1213,6 @@ func NewSchema(config Config) (graphql.Schema, error) {
}, nil }, nil
}, },
}, },
"articles": &graphql.Field{
Name: "Articles",
Description: "文章列表",
Type: graphql.NewObject(graphql.ObjectConfig{
Name: "ArticleConnection",
Description: "条件筛选文章列表",
Fields: graphql.Fields{
"list": &graphql.Field{Type: graphql.NewList(article), Description: "文章列表"},
"total": &graphql.Field{Type: graphql.Int, Description: "文章总数"},
},
}),
Args: graphql.FieldConfigArgument{
"id": &graphql.ArgumentConfig{Type: graphql.Int, Description: "筛选文章中指定ID的"},
"title": &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选文章中含有指定标题的"},
"tags": &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选文章中含有指定标签的"},
"create_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选文章中创建时间等于指定值的"},
"update_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选文章中更新时间等于指定值的"},
},
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
first := 10 // p.Args["first"]
after := 0 // p.Args["after"]
fields := strings.Join(get_fields(p.Info.FieldASTs[0].SelectionSet.Selections), ",")
var where []string
if p.Args["id"] != nil {
where = append(where, fmt.Sprintf("id=%d", p.Args["id"]))
}
if p.Args["title"] != nil {
where = append(where, fmt.Sprintf("title='%s'", p.Args["title"]))
}
// 筛选条件
where_str := strings.Join(where, " AND ")
if where_str != "" {
where_str = "WHERE " + where_str
}
var query strings.Builder
query.WriteString(fmt.Sprintf("SELECT %s FROM web_article %s LIMIT %d OFFSET %d", fields, where_str, first, after))
// 返回翻页信息
var articles []Article
if err := connection.Select(&articles, query.String()); err != nil {
fmt.Println("获取文章列表失败", err)
return nil, err
}
return map[string]interface{}{
"list": articles,
"total": 0,
}, nil
},
},
"tags": &graphql.Field{ "tags": &graphql.Field{
Name: "tags", Name: "tags",
Description: "标签列表", Description: "标签列表",

View File

@@ -109,6 +109,26 @@ func (User) TableName() string {
return "web_member" return "web_member"
} }
type Work struct {
ID int `json:"id" gorm:"primaryKey"`
Title string `json:"title"`
Orientation string `json:"orientation"`
Device string `json:"device"`
Era string `json:"era"`
Tags string `json:"tags"`
UserId int `json:"user_id"`
Content string `json:"content"`
Image string `json:"image"`
Images string `json:"images"`
User User `json:"user" gorm:"foreignKey:UserId"`
CreateTime time.Time `json:"create_time"`
UpdateTime time.Time `json:"update_time"`
}
func (Work) TableName() string {
return "web_article"
}
type Game struct { type Game struct {
ID int `json:"id" gorm:"primaryKey"` ID int `json:"id" gorm:"primaryKey"`
Title string `json:"title"` Title string `json:"title"`