gameui/webp
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

图片列表查询

Browse Source
This commit is contained in:
桜華
2023-04-10 17:10:33 +08:00
parent 3a90035be3
commit fd96b751bf
2 changed files with 115 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
models/mysql.go
View File

@@ -9,8 +9,13 @@ import (
_ "github.com/go-sql-driver/mysql"
)
// 初始化数据库连接
//func (m *MysqlConnectionx) Init() {
// fmt.Println("初始化数据库连接")
//}
type MysqlConnection struct {
db *sql.DB
Database *sql.DB
}
// 初始化数据库连接
@@ -22,7 +27,7 @@ func (m *MysqlConnection) Init() (err error) {
port := viper.Get("mysql.port").(int)
database := viper.Get("mysql.database").(string)
sqlconf := user + ":" + password + "@tcp(" + host + ":" + strconv.Itoa(port) + ")/" + database + "?charset=utf8mb4&parseTime=True&loc=Local"
m.db, err = sql.Open("mysql", sqlconf) // 连接数据库
m.Database, err = sql.Open("mysql", sqlconf) // 连接数据库
if err != nil {
log.Println("连接数据库失败", err)
return
@@ -34,17 +39,43 @@ func (m *MysqlConnection) Init() (err error) {
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)
err = m.Database.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)
err = m.Database.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)
err = m.Database.QueryRow("SELECT image FROM web_ad WHERE id=" + id).Scan(&content)
case "avatar":
err = m.db.QueryRow("SELECT avatar FROM web_member WHERE id=" + id).Scan(&content)
err = m.Database.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)
err = m.Database.QueryRow("SELECT content FROM web_images WHERE id=" + id).Scan(&content)
default:
err = errors.New("group 参数错误")
}
return content, err
}
// 获取图片列表
func (m *MysqlConnection) GetImages(page int, size int) (images []byte, err error) {
rows, err := m.Database.Query("SELECT id, group, content FROM web_images LIMIT ?, ?", (page-1)*size, size)
if err != nil {
log.Println("获取图片列表失败", err)
return
}
defer rows.Close()
images = []byte("[")
for rows.Next() {
var id int
var group string
var content string
err = rows.Scan(&id, &group, &content)
if err != nil {
log.Println("获取图片列表失败", err)
return
}
images = append(images, []byte("{\"id\":"+strconv.Itoa(id)+",\"group\":\""+group+"\",\"content\":\""+content+"\"},")...)
}
images = images[:len(images)-1]
images = append(images, []byte("]")...)
return
}