collect_id

api/image.go

@@ -115,7 +115,8 @@ var imageType = graphql.NewObject(graphql.ObjectConfig{
 			var user_id = p.Context.Value("user_id").(int)
 			if user_id != 0 {
 				var collect_id int
-				if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("image_id = ?", p.Source.(Image).ID).First(&collect_id).Error; err != nil {
+				if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("image_id = ?", p.Source.(Image).ID).Where("type = ?", 1).Select("id").Scan(&collect_id).Error; err != nil {
+					fmt.Println(user_id, p.Source.(Image).ID, collect_id, "E", err)
 					return nil, nil
 				}
 				return collect_id, nil