go项目中使用数据库的配置文件

首先在项目的根目录下创建一个config.json文件,内容如下：

{
"database": {
"Dialect": "mysql",
"Database": "***", /*数据库名*/
"User": "***",/*用户名*/
"Password": "***",/*密码*/
"Charset": "utf8", /*字符集 utf8 */
"Host": "127.0.0.1", /*数据库地址*/
"Port": 3306, /*数据库端口*/
"MaxIdleConns": 5, /*空闲时最大的连接数*/
"MaxOpenConns": 10 /*最大的连接数*/
}
}

然后在根目录下创建一个config文件夹，然后创建config.go文件，内容如下：

package config

import (
"github.com/goes/logger"
"github.com/goes/utils"
"encoding/json"
"fmt"
"io/ioutil"
"os"
"regexp"
)

var jsonData map[string]interface{}

func initJSON() {
bytes, err := ioutil.ReadFile("./config.json")
if err != nil {
logger.Error("ReadFile: ", err.Error())
os.Exit(-1)
}
// 去除注释
configStr := string(bytes[:])
reg := regexp.MustCompile(`/\*.*\*/`)

configStr = reg.ReplaceAllString(configStr, "")
bytes = []byte(configStr)
if err := json.Unmarshal(bytes, &jsonData); err != nil {
logger.Log("json parse fail", err.Error())
os.Exit(-1)
}
}

type dbConfig struct {
Dialect      string
Database     string
User         string
Password     string
Charset      string
Host         string
Port         int
URL          string
MaxIdleConns int
MaxOpenConns int
}

var DBConfig dbConfig

func initDB() {
utils.SetObjectByJSON(&DBConfig, jsonData["database"].(map[string]interface{}))
url := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local",
DBConfig.User,
DBConfig.Password,
DBConfig.Host,
DBConfig.Port,
DBConfig.Database,
DBConfig.Charset)
DBConfig.URL = url
}

func init() {
initJSON()
initDB()
}

项目中的使用方法如下:

/**
*	初始化启动数据库服务
*/
func init() {
fmt.Println("数据库持久化服务器启动:")
db, err := gorm.Open(config.DBConfig.Dialect, config.DBConfig.URL)
if err != nil {
fmt.Println("open db connect error.", err)
os.Exit(-1)
}
db.DB().SetMaxIdleConns(config.DBConfig.MaxIdleConns)
db.DB().SetMaxOpenConns(config.DBConfig.MaxOpenConns)
model.DB = db
}