Golang初始化MySQL数据库方法浅析

package common
import (
	"fmt"
	// 看清楚引入包，包的差异存在方法差异
	"github.com/jinzhu/gorm" 
	"github.com/spf13/viper"
)
// 用大写声明(可以全局访问)
var DB *gorm.DB
// 创建数据库连接池
func InitDB() *gorm.DB {
	driverName := viper.GetString("datasource.driverName")
	host := viper.GetString("datasource.host")
	port := viper.GetString("datasource.port")
	database := viper.GetString("datasource.database")
	username := viper.GetString("datasource.username")
	password := viper.GetString("datasource.password")
	charset := viper.GetString("datasource.charset")
	args := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&parseTime=true",
		username,
		password,
		host,
		port,
		database,
		charset)
	db, err := gorm.Open(driverName, args)
	if err != nil {
		fmt.Println("fail err mysql", err.Error())
	}
	// gorm 自动创建表,需要放入model层中的模型，比如 User{}
	db.AutoMigrate(User{})
	// 进行赋值 否则会空指针
	DB = db
	return db
}
// 获取DB的示例
func GetDB() *gorm.DB {
	return DB
}

package main
import (
	"github.com/gin-gonic/gin"
	_ "github.com/go-sql-driver/mysql"
	"github.com/spf13/viper"
	"go-gin-vue/common"
	"os"
)
func main() {
	InitConfig()
	// 尝试连接数据库
	db := common.InitDB()
	// 延迟关闭
	defer db.Close()
	r := gin.Default()
	// 加载路由
	r = CollectRouter(r)
	// 监听的端口
	// 可以在yml中定义server端口并且使用yml文件中定义的端口
	// port := viper.GetString("server.port") 
	if port != "" {
		panic(r.Run(":" + port))
	}
	r.Run(":9000") // 测试端口 9000 上启动服务
}
// 初始化config文件
func InitConfig() {
	workDir, _ := os.Getwd()
	// 读取的文件名
	viper.SetConfigName("application")
	// 读取的文件类型
	viper.SetConfigType("yml")
	// 读取的路径
	viper.AddConfigPath(workDir + "/config")
	err := viper.ReadInConfig()
	if err != nil {
		panic("")
	}
}