package common
import (
	"fmt"
	// 看清楚引入包，包的差異存在方法差異
	"github.com/jinzhu/gorm" 
	"github.com/spf13/viper"
)
// 用大寫(xiě)聲明(可以全局訪問(wèn))
var DB *gorm.DB
// 創(chuàng)建數(shù)據(jù)庫(kù)連接池
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 自動(dòng)創(chuàng)建表,需要放入model層中的模型，比如 User{}
	db.AutoMigrate(User{})
	// 進(jìn)行賦值 否則會(huì)空指針
	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()
	// 嘗試連接數(shù)據(jù)庫(kù)
	db := common.InitDB()
	// 延遲關(guān)閉
	defer db.Close()
	r := gin.Default()
	// 加載路由
	r = CollectRouter(r)
	// 監(jiān)聽(tīng)的端口
	// 可以在yml中定義server端口并且使用yml文件中定義的端口
	// port := viper.GetString("server.port") 
	if port != "" {
		panic(r.Run(":" + port))
	}
	r.Run(":9000") // 測(cè)試端口 9000 上啟動(dòng)服務(wù)
}
// 初始化config文件
func InitConfig() {
	workDir, _ := os.Getwd()
	// 讀取的文件名
	viper.SetConfigName("application")
	// 讀取的文件類型
	viper.SetConfigType("yml")
	// 讀取的路徑
	viper.AddConfigPath(workDir + "/config")
	err := viper.ReadInConfig()
	if err != nil {
		panic("")
	}
}