diff --git a/api/collect.go b/api/collect.go index 6eaff86..ebb582d 100644 --- a/api/collect.go +++ b/api/collect.go @@ -127,11 +127,13 @@ var CollectionItems = &graphql.Field{ 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("collect_id AS id").Limit(3).Where("explorer_id = ?", item.ID).Find(&data).Error; err != nil { + if err := db.Table("web_collect").Select(s).Limit(3).Where("explorer_id = ?", item.ID).Find(&data).Error; err != nil { fmt.Println("获取封面ID失败", err) }