diff --git a/api/collect.go b/api/collect.go index adbe334..1d92fa0 100644 --- a/api/collect.go +++ b/api/collect.go @@ -89,8 +89,14 @@ var CollectionItems = &graphql.Field{ var items = ListItem(p.Info.FieldASTs[0].SelectionSet.Selections) if funk.Contains(items, "fan") { - for i := range collects { - collects[i].Fan = false + 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 } }