53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package databaseInit
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
_ "github.com/go-sql-driver/mysql"
|
||
"gopkg.in/yaml.v3"
|
||
"os"
|
||
"toutoukan/model/config"
|
||
)
|
||
|
||
var UserDB *sql.DB
|
||
|
||
func DbInit() {
|
||
data, err := os.ReadFile("./config.yaml")
|
||
if err != nil {
|
||
panic(fmt.Errorf("读取配置文件失败: %w", err))
|
||
}
|
||
|
||
// 解析配置
|
||
var config config.Config
|
||
if err := yaml.Unmarshal(data, &config); err != nil {
|
||
panic(fmt.Errorf("解析YAML失败: %w", err))
|
||
}
|
||
|
||
// 构建DSN
|
||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/toutoukan?%s",
|
||
config.Database.Username,
|
||
config.Database.Password,
|
||
config.Database.Host,
|
||
config.Database.Port,
|
||
config.Database.Params,
|
||
)
|
||
|
||
fmt.Println("DSN:", dsn)
|
||
|
||
UserDB, err = sql.Open("mysql", dsn)
|
||
if err != nil {
|
||
panic(fmt.Errorf("数据库连接失败1: %w", err))
|
||
}
|
||
|
||
// 验证连接是否可用
|
||
if err := UserDB.Ping(); err != nil {
|
||
panic("数据库连接不可用2")
|
||
}
|
||
|
||
// 设置连接池参数
|
||
UserDB.SetMaxOpenConns(10) // 最大打开连接数
|
||
UserDB.SetMaxIdleConns(5) // 最大空闲连接数
|
||
UserDB.SetConnMaxLifetime(0) // 连接最大存活时间（0表示不限制）
|
||
|
||
}
|