collect_id

api/image.go

@@ -100,7 +100,7 @@ var imageType = graphql.NewObject(graphql.ObjectConfig{
 			var user_id = p.Context.Value("user_id").(int)
 			if user_id != 0 {
 				var collect int64
-				if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("image_id = ?", p.Source.(Image).ID).Count(&collect); err != nil {
+				if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("collect_id = ?", p.Source.(Image).ID).Where("type = ?", 1).Count(&collect); err != nil {
 					return false, nil
 				}
 				if collect > 0 {