This commit is contained in:
2024-12-03 05:49:06 +08:00
parent 8f5486b0cf
commit 92fa9e8a2c

View File

@@ -6,23 +6,23 @@ import (
"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"`
PraiseCount int `json:"praise_count" gorm:"default:0"`
Praise bool `json:"praise" gorm:"-"`
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 {
@@ -78,16 +78,19 @@ var CollectionItems = &graphql.Field{
limit = p.Args["first"].(int)
}
if err := db.Limit(limit).Preload("User").Find(&collects).Error; err != nil {
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, "praise_count") {
if funk.Contains(items, "fan") {
for i := range collects {
collects[i].PraiseCount = collects[i].Fans
collects[i].Praise = false
collects[i].Fan = false
}
}