205 lines
7.0 KiB
Go
205 lines
7.0 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/graphql-go/graphql"
|
|
"github.com/thoas/go-funk"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
type Cover struct {
|
|
ID int `json:"id"`
|
|
Type string `json:"type"`
|
|
}
|
|
|
|
type Collection struct {
|
|
ID int `json:"id" gorm:"primaryKey"`
|
|
Title string `json:"title"`
|
|
Content string `json:"content"`
|
|
Type string `json:"type"`
|
|
Thumbnail string `json:"thumbnail"`
|
|
Num int `json:"num"`
|
|
Fans int `json:"fans"`
|
|
UserId int `json:"user_id"`
|
|
ArticleId int `json:"article_id"`
|
|
CreateTime time.Time `json:"create_time"`
|
|
UpdateTime time.Time `json:"update_time"`
|
|
User User `json:"user" gorm:"foreignKey:UserId;references:ID"`
|
|
Fan bool `json:"praise" gorm:"-"`
|
|
Covers []Cover `json:"covers" gorm:"-"`
|
|
Views int `json:"views"`
|
|
}
|
|
|
|
func (Collection) TableName() string {
|
|
return "web_member_explorer"
|
|
}
|
|
|
|
var CollectionItems = &graphql.Field{
|
|
Name: "collections",
|
|
Description: "收藏列表",
|
|
Type: graphql.NewObject(graphql.ObjectConfig{
|
|
Name: "CollectConnection",
|
|
Description: "条件筛选收藏夹列表",
|
|
Fields: graphql.Fields{
|
|
"list": &graphql.Field{Type: graphql.NewList(graphql.NewObject(graphql.ObjectConfig{
|
|
Name: "collection",
|
|
Description: "收藏夹",
|
|
Fields: graphql.Fields{
|
|
"id": &graphql.Field{Type: graphql.Int, Description: "ID"},
|
|
"type": &graphql.Field{Type: graphql.Int, Description: "类型"},
|
|
"title": &graphql.Field{Type: graphql.String, Description: "标题"},
|
|
"content": &graphql.Field{Type: graphql.String, Description: "内容"},
|
|
"thumbnail": &graphql.Field{Type: graphql.String, Description: "缩略图"},
|
|
"num": &graphql.Field{Type: graphql.Int, Description: "收藏数量"},
|
|
"create_time": &graphql.Field{Type: graphql.DateTime, Description: "创建时间"},
|
|
"update_time": &graphql.Field{Type: graphql.DateTime, Description: "更新时间"},
|
|
"fans": &graphql.Field{Type: graphql.Int, Description: "关注数"},
|
|
"fan": &graphql.Field{Type: graphql.Boolean, Description: "当前用户是否关注"},
|
|
"user": &graphql.Field{Type: userType, Description: "用户"},
|
|
"covers": &graphql.Field{
|
|
Type: graphql.NewList(graphql.NewObject(graphql.ObjectConfig{
|
|
Name: "cover",
|
|
Description: "封面",
|
|
Fields: graphql.Fields{
|
|
"id": &graphql.Field{Type: graphql.Int, Description: "ID"},
|
|
"type": &graphql.Field{Type: graphql.String, Description: "类型"},
|
|
},
|
|
})),
|
|
Description: "封面集",
|
|
},
|
|
"views": &graphql.Field{Type: graphql.Int, Description: "浏览量"},
|
|
},
|
|
})), 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: "筛选收藏中含有指定标题的"},
|
|
"type": &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 collects []Collection
|
|
var total int
|
|
var limit int = 10
|
|
|
|
if p.Args["first"] != nil {
|
|
limit = p.Args["first"].(int)
|
|
}
|
|
|
|
order := clause.OrderByColumn{
|
|
Column: clause.Column{Name: p.Args["sort"].(string)},
|
|
Desc: p.Args["order"].(string) == "DESC",
|
|
}
|
|
|
|
if err := db.Limit(limit).Preload("User").Order(order).Find(&collects).Error; err != nil {
|
|
fmt.Println(err.Error())
|
|
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
|
|
}
|
|
collects[index].Fan = total > 0
|
|
}
|
|
}
|
|
|
|
for index, item := range collects {
|
|
var data []Cover
|
|
var t string = "article"
|
|
var s string = "collect_id AS id"
|
|
if item.Type == "0" {
|
|
t = "image"
|
|
s = "image_id AS id"
|
|
}
|
|
|
|
if err := db.Table("web_collect").Select(s).Limit(3).Where("explorer_id = ?", item.ID).Find(&data).Error; err != nil {
|
|
fmt.Println("获取封面ID失败", err)
|
|
}
|
|
|
|
for i := range data {
|
|
data[i].Type = t
|
|
}
|
|
collects[index].Covers = data
|
|
}
|
|
|
|
if funk.Contains(items, "views") {
|
|
type ApiResponse struct {
|
|
ID int `json:"id"`
|
|
Count int `json:"count"`
|
|
}
|
|
|
|
// 0. 收集要查询的 ID
|
|
var ids []int
|
|
for x := range collects {
|
|
ids = append(ids, collects[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 collects {
|
|
if collects[i].ID == item.ID {
|
|
collects[i].Views = item.Count
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return map[string]interface{}{
|
|
"list": collects,
|
|
"total": total,
|
|
}, nil
|
|
},
|
|
}
|