[FIX]优化配置读取

models/models.go

@@ -18,38 +18,22 @@ type Model struct {
 	DeleteOn   int `json:"deleted_on"`
 }
 
-func init() {
-	var (
-		err                                                     error
-		dbType, dbName, user, password, host, port, tablePrefix string
-	)
+func Setup() {
+	var err error
 
-	sec, err := setting.Cfg.GetSection("database")
-	if err != nil {
-		log.Fatal(2, "Fail to get section 'database': %v", err)
-	}
-
-	dbType = sec.Key("Type").String()
-	dbName = sec.Key("Name").String()
-	user = sec.Key("User").String()
-	password = sec.Key("Password").String()
-	host = sec.Key("Host").String()
-	port = sec.Key("Port").String()
-	tablePrefix = sec.Key("TablePrefix").String()
-
-	db, err = gorm.Open(dbType, fmt.Sprintf("host=%s user=%s dbname=%s port=%s sslmode=disable password=%s",
-		host,
-		user,
-		dbName,
-		port,
-		password))
+	db, err = gorm.Open(setting.DatabaseSetting.Type, fmt.Sprintf("host=%s user=%s dbname=%s port=%s sslmode=disable password=%s",
+		setting.DatabaseSetting.Host,
+		setting.DatabaseSetting.User,
+		setting.DatabaseSetting.Name,
+		setting.DatabaseSetting.Port,
+		setting.DatabaseSetting.Password))
 
 	if err != nil {
-		log.Println(err)
+		log.Fatalf("models.Setup err: %v", err)
 	}
 
 	gorm.DefaultTableNameHandler = func(db *gorm.DB, defaultTableName string) string {
-		return tablePrefix + defaultTableName
+		return setting.DatabaseSetting.TablePrefix + defaultTableName
 	}
 
 	db.AutoMigrate(&Employee{})