Files
webp/api/collect.go
2024-12-06 08:51:31 +08:00

110 lines
4.6 KiB
Go

package api
import (
"fmt"
"time"
"github.com/graphql-go/graphql"
"github.com/thoas/go-funk"
"gorm.io/gorm/clause"
)
type Collection struct {
ID int `json:"id" gorm:"primaryKey"`
Title string `json:"title"`
Content string `json:"content"`
Type int `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:"-"`
}
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: "用户"},
},
})), 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.Int, 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.String, Description: "翻页参数(傳回清單中指定遊標之後的元素)"},
"before": &graphql.ArgumentConfig{Type: graphql.String, 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
}
}
return map[string]interface{}{
"list": collects,
"total": total,
}, nil
},
}