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

View File

@@ -6,6 +6,7 @@ import (
"github.com/graphql-go/graphql" "github.com/graphql-go/graphql"
"github.com/thoas/go-funk" "github.com/thoas/go-funk"
"gorm.io/gorm/clause"
) )
type Collection struct { type Collection struct {
@@ -21,8 +22,7 @@ type Collection struct {
CreateTime time.Time `json:"create_time"` CreateTime time.Time `json:"create_time"`
UpdateTime time.Time `json:"update_time"` UpdateTime time.Time `json:"update_time"`
User User `json:"user" gorm:"foreignKey:UserId;references:ID"` User User `json:"user" gorm:"foreignKey:UserId;references:ID"`
PraiseCount int `json:"praise_count" gorm:"default:0"` Fan bool `json:"praise" gorm:"-"`
Praise bool `json:"praise" gorm:"-"`
} }
func (Collection) TableName() string { func (Collection) TableName() string {
@@ -78,16 +78,19 @@ var CollectionItems = &graphql.Field{
limit = p.Args["first"].(int) 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()) fmt.Println(err.Error())
return nil, err return nil, err
} }
var items = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections) var items = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections)
if funk.Contains(items, "praise_count") { if funk.Contains(items, "fan") {
for i := range collects { for i := range collects {
collects[i].PraiseCount = collects[i].Fans collects[i].Fan = false
collects[i].Praise = false
} }
} }