311 lines
12 KiB
Go
311 lines
12 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/doug-martin/goqu/v9"
|
|
"github.com/graphql-go/graphql"
|
|
"github.com/thoas/go-funk"
|
|
)
|
|
|
|
type Article struct {
|
|
ID int `json:"id" db:"id" gorm:"primaryKey"`
|
|
Title string `json:"title" db:"title"`
|
|
Orientation string `json:"orientation" db:"orientation"`
|
|
Device string `json:"device" db:"device"`
|
|
Era string `json:"era" db:"era"`
|
|
Tags string `json:"tags" db:"tags"`
|
|
CategoryID int `json:"category_id"`
|
|
UserId int `json:"user_id" db:"user_id"`
|
|
User User `json:"user" gorm:"foreignKey:UserId"`
|
|
Collect bool `json:"collect" gorm:"column:is_collect"`
|
|
Praise bool `json:"praise" gorm:"column:is_praise"`
|
|
CollectCount int `json:"collect_count" gorm:"column:collect_num"`
|
|
PraiseCount int `json:"praise_count" gorm:"column:praise"`
|
|
CreateTime time.Time `json:"create_time" db:"create_time"`
|
|
UpdateTime time.Time `json:"update_time" db:"update_time"`
|
|
Emoji1 int `json:"emoji1"`
|
|
Emoji2 int `json:"emoji2"`
|
|
Emoji3 int `json:"emoji3"`
|
|
Emoji4 int `json:"emoji4"`
|
|
Emoji5 int `json:"emoji5"`
|
|
Views int `json:"views"`
|
|
}
|
|
|
|
func (Article) TableName() string {
|
|
return "web_article"
|
|
}
|
|
|
|
var articleType = graphql.NewObject(graphql.ObjectConfig{
|
|
Name: "Article",
|
|
Description: "文章",
|
|
Fields: graphql.Fields{
|
|
"id": &graphql.Field{Type: graphql.Int, Description: "ID"},
|
|
"title": &graphql.Field{Type: graphql.String, Description: "标题"},
|
|
"orientation": &graphql.Field{Type: graphql.String, Description: "方向"},
|
|
"device": &graphql.Field{Type: graphql.String, Description: "设备"},
|
|
"era": &graphql.Field{Type: graphql.String, Description: "游戏上线年份"},
|
|
"tags": &graphql.Field{Type: graphql.String, Description: "标签"},
|
|
"category_id": &graphql.Field{Type: graphql.Int, Description: "分类ID"},
|
|
"user": &graphql.Field{Type: userType, Description: "所属用户"},
|
|
"create_time": &graphql.Field{Type: graphql.DateTime, Description: "创建时间"},
|
|
"update_time": &graphql.Field{Type: graphql.DateTime, Description: "更新时间"},
|
|
"praise": &graphql.Field{Type: graphql.Boolean, Description: "当前用户是否点赞"},
|
|
"collect": &graphql.Field{Type: graphql.Boolean, Description: "当前用户是否收藏"},
|
|
"praise_count": &graphql.Field{Type: graphql.Int, Description: "点赞数"},
|
|
"collect_count": &graphql.Field{Type: graphql.Int, Description: "收藏数"},
|
|
"emoji1": &graphql.Field{Type: graphql.Int, Description: "表情1数量"},
|
|
"emoji2": &graphql.Field{Type: graphql.Int, Description: "表情2数量"},
|
|
"emoji3": &graphql.Field{Type: graphql.Int, Description: "表情3数量"},
|
|
"emoji4": &graphql.Field{Type: graphql.Int, Description: "表情4数量"},
|
|
"emoji5": &graphql.Field{Type: graphql.Int, Description: "表情5数量"},
|
|
"views": &graphql.Field{Type: graphql.Int, Description: "浏览量"},
|
|
},
|
|
})
|
|
|
|
var ArticleItem = &graphql.Field{
|
|
Name: "article",
|
|
Description: "单篇文章",
|
|
Type: articleType,
|
|
Args: graphql.FieldConfigArgument{
|
|
"id": &graphql.ArgumentConfig{Type: graphql.Int, Description: "根据ID获取文章"},
|
|
},
|
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
|
article := Article{ID: p.Args["id"].(int)}
|
|
if err := db.First(&article).Error; err != nil {
|
|
log.Println("获取文章失败", err)
|
|
return nil, err
|
|
}
|
|
return article, nil
|
|
},
|
|
}
|
|
|
|
var ArticleItems = &graphql.Field{
|
|
Name: "articles",
|
|
Description: "文章列表",
|
|
Type: graphql.NewObject(graphql.ObjectConfig{
|
|
Name: "ArticleConnection",
|
|
Description: "条件筛选文章列表",
|
|
Fields: graphql.Fields{
|
|
"list": &graphql.Field{Type: graphql.NewList(articleType), 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: "筛选文章中更新时间等于指定值的"},
|
|
"sort": &graphql.ArgumentConfig{Type: graphql.String, Description: "按指定字段排序", DefaultValue: "id"},
|
|
"order": &graphql.ArgumentConfig{Type: orderType, Description: "排序类型(升序或降序)", DefaultValue: "ASC"},
|
|
"first": &graphql.ArgumentConfig{Type: graphql.Int, Description: "翻页参数(傳回清單中的前n個元素)"},
|
|
"last": &graphql.ArgumentConfig{Type: graphql.Int, Description: "翻页参数(傳回清單中的最後n個元素)"},
|
|
"after": &graphql.ArgumentConfig{Type: graphql.Int, Description: "翻页参数(傳回清單中指定遊標之後的元素)"},
|
|
"before": &graphql.ArgumentConfig{Type: graphql.Int, Description: "翻页参数(傳回清單中指定遊標之前的元素)"},
|
|
},
|
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
|
var articles []Article
|
|
var total int
|
|
var err error
|
|
|
|
// 文章的顶级ID为9
|
|
var query = goqu.Dialect("mysql").From("web_article").Where(goqu.Ex{"category_id": 10})
|
|
|
|
// 筛选条件
|
|
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["praise"] != nil {
|
|
query = query.Join(goqu.T("web_praise"), goqu.On(
|
|
goqu.I("web_article.id").Eq(goqu.I("web_praise.praise_id")),
|
|
goqu.I("web_praise.user_id").Eq(p.Args["praise"]),
|
|
goqu.I("web_praise.type").Eq(0),
|
|
))
|
|
}
|
|
|
|
// 按指定用户收藏筛选
|
|
if p.Args["collect"] != nil {
|
|
query = query.Join(goqu.T("web_collect"), goqu.On(
|
|
goqu.I("web_article.id").Eq(goqu.I("web_collect.collect_id")),
|
|
goqu.I("web_collect.user_id").Eq(p.Args["collect"]),
|
|
goqu.I("web_collect.type").Eq(0),
|
|
))
|
|
}
|
|
|
|
// 如果查询了 total 字段
|
|
if existField(p.Info.FieldASTs[0].SelectionSet.Selections, "total") {
|
|
sql, _, _ := query.ToSQL()
|
|
sql = strings.Replace(sql, "SELECT *", "SELECT COUNT(*)", 1)
|
|
if err := db.Raw(sql).Scan(&total).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// 如果没有外部排序则使用指定排序(正则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_article.id", goqu.L(
|
|
fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_article.%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 limit int = 10
|
|
if p.Args["first"] != nil {
|
|
limit = p.Args["first"].(int)
|
|
} else if p.Args["last"] != nil {
|
|
limit = p.Args["last"].(int)
|
|
}
|
|
|
|
// 字段选择
|
|
var user_id = p.Context.Value("user_id").(int)
|
|
var fields = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections)
|
|
var praise, collect, text_count string
|
|
if funk.Contains(fields, "praise") {
|
|
praise = fmt.Sprintf(",CASE WHEN EXISTS (SELECT 1 FROM web_praise WHERE web_praise.praise_id = web_article.id AND web_praise.user_id = %d AND web_praise.type = 0) THEN TRUE ELSE FALSE END AS is_praise", user_id)
|
|
}
|
|
if funk.Contains(fields, "collect") {
|
|
collect = fmt.Sprintf(",CASE WHEN EXISTS (SELECT 1 FROM web_collect WHERE web_collect.collect_id = web_article.id AND web_collect.user_id = %d AND web_collect.type = 0) THEN TRUE ELSE FALSE END AS is_collect", user_id)
|
|
}
|
|
if funk.Contains(fields, "text_count") {
|
|
text_count = ",(SELECT COUNT(*) FROM web_images wi WHERE wi.article_id = web_article.id AND wi.text != '') AS text_count"
|
|
}
|
|
sql = fmt.Sprintf(`
|
|
WITH RankedArticles AS (%s)
|
|
SELECT * %s %s %s FROM web_article INNER JOIN(
|
|
SELECT id, row_num FROM RankedArticles %s
|
|
) AS LimitedRanked ON LimitedRanked.id = web_article.id
|
|
ORDER BY LimitedRanked.row_num ASC LIMIT %d
|
|
`, sql, praise, collect, text_count, cursor, limit)
|
|
fmt.Println(sql)
|
|
|
|
if err := db.Raw(sql).Scan(&articles).Error; err != nil {
|
|
fmt.Println("获取文章列表失败", err)
|
|
return nil, err
|
|
}
|
|
|
|
//// 试图使用GORM语法整理
|
|
//subQuery := db.Table("web_article").Select([]string{
|
|
// "web_article.*",
|
|
// fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_article.%s %s) AS row_num", p.Args["sort"], p.Args["order"]),
|
|
//}).Where("category_id = 10").Limit(limit)
|
|
//// 筛选条件
|
|
//for _, format := range []string{"id", "style", "device", "orientation", "era", "category_id", "tags"} {
|
|
// if p.Args[format] != nil {
|
|
// subQuery = subQuery.Where(fmt.Sprintf("%s = ?", format), p.Args[format])
|
|
// }
|
|
//}
|
|
//// 按指定用户点赞筛选
|
|
//if p.Args["praise"] != nil {
|
|
// subQuery = subQuery.Joins("JOIN web_praise AS wp ON wp.praise_id = web_article.id AND wp.user_id = ? AND wp.type = 0", p.Args["praise"])
|
|
//}
|
|
//// 按指定用户收藏筛选
|
|
//if p.Args["collect"] != nil {
|
|
// subQuery = subQuery.Joins("JOIN web_collect AS wc ON wc.collect_id = web_article.id AND wc.user_id = ? AND wc.type = 0", p.Args["collect"])
|
|
//}
|
|
//var _select []string = []string{"*"}
|
|
//if funk.Contains(fields, "text_count") {
|
|
// _select = append(_select, "(SELECT COUNT(*) FROM web_images wi WHERE wi.article_id = web_article.id AND wi.text != '') AS text_count")
|
|
//}
|
|
//if funk.Contains(fields, "collect") {
|
|
// _select = append(_select, fmt.Sprintf("(SELECT COUNT(*) FROM web_collect wc WHERE wc.collect_id = web_article.id AND wc.user_id = %d AND wc.type = 0) AS is_collect", user_id))
|
|
//}
|
|
//if funk.Contains(fields, "praise") {
|
|
// _select = append(_select, fmt.Sprintf("(SELECT COUNT(*) FROM web_praise wp WHERE wp.praise_id = web_article.id AND wp.user_id = %d AND wp.type = 0) AS is_praise", user_id))
|
|
//}
|
|
//if err := db.Table("(?) AS limited_imgs", subQuery).Select(_select).Scan(&articles).Error; err != nil {
|
|
// fmt.Println("获取文章列表失败", err)
|
|
// return nil, err
|
|
//}
|
|
|
|
var items = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections)
|
|
if funk.Contains(items, "views") {
|
|
type ApiResponse struct {
|
|
ID int `json:"id"`
|
|
Count int `json:"count"`
|
|
}
|
|
|
|
// 0. 收集要查询的 ID
|
|
var ids []int
|
|
for x := range articles {
|
|
ids = append(ids, articles[x].ID)
|
|
}
|
|
idx := strings.Trim(strings.Replace(fmt.Sprint(ids), " ", ",", -1), "[]")
|
|
|
|
// 1. 发送 GET 请求
|
|
resp, err := http.Get("http://localhost:6005/api/get_views/文章?ids=" + idx)
|
|
if err != nil {
|
|
fmt.Println("Error making GET request:", err)
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// 2. 检查 HTTP 状态码
|
|
if resp.StatusCode != http.StatusOK {
|
|
fmt.Printf("Request failed with status code: %d\n", resp.StatusCode)
|
|
return nil, err
|
|
}
|
|
|
|
// 3. 读取响应体
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
fmt.Println("Error reading response body:", err)
|
|
return nil, err
|
|
}
|
|
|
|
// 4. 解析 JSON 数据到结构体
|
|
var data []ApiResponse
|
|
err = json.Unmarshal(body, &data)
|
|
if err != nil {
|
|
fmt.Println("Error unmarshalling JSON:", err)
|
|
return nil, err
|
|
}
|
|
|
|
// 5. 赋值到数据集
|
|
for _, item := range data {
|
|
for i := range articles {
|
|
if articles[i].ID == item.ID {
|
|
articles[i].Views = item.Count
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return map[string]interface{}{
|
|
"list": articles,
|
|
"total": total,
|
|
}, err
|
|
},
|
|
}
|