group

models/mysql.go

@@ -29,9 +29,19 @@ func (m *MysqlConnection) Init() (err error) {
 	return
 }
 
-// 根据 id 获取图片的 content
-func (m *MysqlConnection) GetImageContent(id string) (key string, err error) {
-	var content string
-	err = m.db.QueryRow("SELECT content FROM web_images WHERE id=" + id).Scan(&content)
+// 根据图片 id 获取图片的 content
+func (m *MysqlConnection) GetImageContent(group string, id string) (content string, err error) {
+	switch group {
+	case "article":
+		err = m.db.QueryRow("SELECT image FROM web_article WHERE id=" + id).Scan(&content)
+	case "article_attribute":
+		err = m.db.QueryRow("SELECT image FROM web_article_attribute WHERE id=" + id).Scan(&content)
+	case "ad":
+		err = m.db.QueryRow("SELECT image FROM web_ad WHERE id=" + id).Scan(&content)
+	case "user":
+		err = m.db.QueryRow("SELECT avatar FROM web_member WHERE id=" + id).Scan(&content)
+	case "image":
+		err = m.db.QueryRow("SELECT content FROM web_images WHERE id=" + id).Scan(&content)
+	}
 	return content, err
 }