package models

import (
	"database/sql"
	"log"
	"strconv"

	_ "github.com/go-sql-driver/mysql"
)

type MysqlConnection struct {
	db *sql.DB
}

// 初始化数据库连接
func (m *MysqlConnection) Init() (err error) {
	viper := Viper // 从 models/config.go 中获取 viper 对象
	user := viper.Get("mysql.user").(string)
	password := viper.Get("mysql.password").(string)
	host := viper.Get("mysql.host").(string)
	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) // 连接数据库
	if err != nil {
		log.Println("连接数据库失败", err)
		return
	}
	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)
	return content, err
}