共用配置文件导入

models/mysql.go

@@ -10,38 +10,24 @@ import (
 	"time"
 
 	_ "github.com/go-sql-driver/mysql"
-	"github.com/jmoiron/sqlx"
 )
 
-var connection *sqlx.DB
 var connectionx *sql.DB
 
-func init() {
-	var err error
-	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)
-	connection, err = sqlx.Connect("mysql", fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", user, password, host, port, database))
-	if err != nil {
-		log.Fatalln("连接数据库失败", err)
-	}
-}
-
 type MysqlConnection struct {
 	Database *sql.DB
 }
 
 // 初始化数据库连接
 func (m *MysqlConnection) Init() (err error) {
-	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.Database, err = sql.Open("mysql", sqlconf) // 连接数据库
+	m.Database, err = sql.Open("mysql", fmt.Sprintf(
+		"%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		config.GetString("mysql.user"),
+		config.GetString("mysql.password"),
+		config.GetString("mysql.host"),
+		config.GetInt("mysql.port"),
+		config.GetString("mysql.database"),
+	))
 	if err != nil {
 		log.Println("连接数据库失败", err)
 		return