order

api/collect.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/graphql-go/graphql"
 	"github.com/thoas/go-funk"
+	"gorm.io/gorm/clause"
 )
 
 type Collection struct {
@@ -21,8 +22,7 @@ type Collection struct {
 	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:"-"`
+	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
 			}
 		}