合并查询 fan
This commit is contained in:
125
api/collect.go
125
api/collect.go
@@ -5,12 +5,13 @@ import (
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/doug-martin/goqu/v9"
|
||||
"github.com/graphql-go/graphql"
|
||||
"github.com/thoas/go-funk"
|
||||
"gorm.io/gorm/clause"
|
||||
)
|
||||
|
||||
type Cover struct {
|
||||
@@ -101,29 +102,123 @@ var CollectionItems = &graphql.Field{
|
||||
limit = p.Args["first"].(int)
|
||||
}
|
||||
|
||||
order := clause.OrderByColumn{
|
||||
Column: clause.Column{Name: p.Args["sort"].(string)},
|
||||
Desc: p.Args["order"].(string) == "DESC",
|
||||
//order := clause.OrderByColumn{
|
||||
// Column: clause.Column{Name: p.Args["sort"].(string)},
|
||||
// Desc: p.Args["order"].(string) == "DESC",
|
||||
//}
|
||||
|
||||
var query = goqu.Dialect("mysql").From("web_member_explorer")
|
||||
|
||||
// 筛选条件
|
||||
for _, format := range []string{"id", "title", "type"} {
|
||||
if p.Args[format] != nil {
|
||||
query = query.Where(goqu.C(format).Eq(p.Args[format]))
|
||||
}
|
||||
}
|
||||
|
||||
if err := db.Limit(limit).Preload("User").Order(order).Find(&collects).Error; err != nil {
|
||||
fmt.Println(err.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_member_explorer.id", goqu.L(
|
||||
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_member_explorer.%s %s)", sort, p.Args["order"]),
|
||||
).As("row_num"))
|
||||
} else {
|
||||
// 排序条件
|
||||
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())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 取所有数据的前N条
|
||||
sql, _, _ := query.ToSQL()
|
||||
|
||||
// 遊標截取篩選結果集的前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 user_id = p.Context.Value("user_id").(int)
|
||||
//var fields = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections)
|
||||
//var praise, praise_join, collect, collect_join, text_count string
|
||||
//if funk.Contains(fields, "praise") {
|
||||
// praise = ",CASE WHEN web_praise.id IS NOT NULL THEN TRUE ELSE FALSE END AS is_praise"
|
||||
// praise_join = fmt.Sprintf("LEFT JOIN web_praise ON web_praise.praise_id = web_article.id AND web_praise.user_id = %d AND web_praise.type = 0", user_id)
|
||||
//}
|
||||
//if funk.Contains(fields, "collect") {
|
||||
// collect = ",CASE WHEN web_collect.id IS NOT NULL THEN TRUE ELSE FALSE END AS is_collect"
|
||||
// collect_join = fmt.Sprintf("LEFT JOIN web_collect ON web_collect.collect_id = web_article.id AND web_collect.user_id = %d AND web_collect.type = 0", user_id)
|
||||
//}
|
||||
|
||||
var items = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections)
|
||||
var fan, fan_join string
|
||||
if funk.Contains(items, "fan") {
|
||||
fan = ",CASE WHEN web_fans.id IS NOT NULL THEN TRUE ELSE FALSE END AS fan"
|
||||
fan_join = fmt.Sprintf("LEFT JOIN web_fans ON web_fans.follower_id = %d AND web_fans.blogger_id = web_member_explorer.id AND web_fans.type = 3", user_id)
|
||||
//fan = fmt.Sprintf(",(SELECT COUNT(*) FROM web_fans WHERE follower_id = %d AND blogger_id = web_member_explorer.id AND type = 3) AS fan", user_id)
|
||||
}
|
||||
|
||||
sql = fmt.Sprintf(`
|
||||
WITH RankedArticles AS (%s)
|
||||
SELECT web_member_explorer.* %s FROM web_member_explorer INNER JOIN(
|
||||
SELECT id, row_num FROM RankedArticles %s
|
||||
) AS LimitedRanked ON LimitedRanked.id = web_member_explorer.id %s
|
||||
ORDER BY LimitedRanked.row_num ASC LIMIT %d
|
||||
`, sql, fan, cursor, fan_join, limit)
|
||||
|
||||
fmt.Println(sql)
|
||||
|
||||
if err := db.Raw(sql).Scan(&collects).Error; err != nil {
|
||||
fmt.Println("获取游戏列表失败", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var items = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections)
|
||||
if funk.Contains(items, "fan") {
|
||||
var user_id = p.Context.Value("user_id").(int)
|
||||
for index, item := range collects {
|
||||
var total int64
|
||||
if err := db.Table("web_fans").Where("follower_id = ? AND blogger_id = ? AND type = 3", user_id, item.ID).Count(&total).Error; err != nil {
|
||||
fmt.Println(index, err.Error())
|
||||
return nil, err
|
||||
if funk.Contains(items, "user") {
|
||||
var ids []int
|
||||
for _, collect := range collects {
|
||||
ids = append(ids, collect.UserId)
|
||||
}
|
||||
ids = funk.UniqInt(ids)
|
||||
|
||||
var users []User
|
||||
if err := db.Table("web_member").Where("id in (?)", ids).Find(&users).Error; err != nil {
|
||||
fmt.Println("获取用户信息失败", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for index, game := range collects {
|
||||
for _, user := range users {
|
||||
if game.UserId == user.ID {
|
||||
collects[index].User = user
|
||||
break
|
||||
}
|
||||
}
|
||||
collects[index].Fan = total > 0
|
||||
}
|
||||
}
|
||||
|
||||
//if err := db.Limit(limit).Preload("User").Order(order).Find(&collects).Error; err != nil {
|
||||
// fmt.Println(err.Error())
|
||||
// return nil, err
|
||||
//}
|
||||
|
||||
//if funk.Contains(items, "fan") {
|
||||
// var user_id = p.Context.Value("user_id").(int)
|
||||
// for index, item := range collects {
|
||||
// var total int64
|
||||
// if err := db.Table("web_fans").Where("follower_id = ? AND blogger_id = ? AND type = 3", user_id, item.ID).Count(&total).Error; err != nil {
|
||||
// fmt.Println(index, err.Error())
|
||||
// return nil, err
|
||||
// }
|
||||
// collects[index].Fan = total > 0
|
||||
// }
|
||||
//}
|
||||
|
||||
for index, item := range collects {
|
||||
var data []Cover
|
||||
var t string = "article"
|
||||
|
||||
Reference in New Issue
Block a user