iris 数据库配置及连接方法封装｜Go主题月

上一节讲完项目结构 & 生成数据表映射之后，接下来就是项目连接数据库操作数据库的代码封装：

连接数据库配置

数据库配置的主要目的就是为了把 Golang 链接数据库链接

“[User]:[Pwd]@tcp([Host]:[Port])/[DbName]?charset=[DbCharset]”

中的各项参数，变成结构体可标准配置。比如：

type DbConf struct {
	Host string
	Port int
	User string
	Pwd string
	DbName string
	DbCharset string
}
复制代码

工业项目一般都涉及多人开发，开发维护的时候为了统一管理。开发组内都会有自己的代码标准，所以这也是项目需要这配置那配置的配置化原因

真正链接数据库的时候，使用fmt格式化输出函数，就可以轻易获得链接数据库链接

const DriverName  = "mysql"

var MasterDbConf DbConf = DbConf{
	Host : "127.0.0.1",
	Port : 3306,
	User : "root",
	Pwd : "",
    DbName : "superstar",
    DbCharset: "utf8",
}

conf := MasterDbConf
driverSource := fmt.Sprintf(
    "%s:%s@tcp(%s:%d)/%s?charset=%s",
    c.User, c.Pwd, c.Host, c.Port, c.DbName, c.DbCharset
)
fmt.Println(driverSource)

masterEngine, err := xorm.NewEngine(conf.DriverName, driverSource)
if err != nil {
    log.Fatal("dbhelper.DbInstanceMaster err:", err)
    return nil
}
复制代码

配置可以不只一份，比如主从配置，读写分离。数据同步交给数据库主从同步，读取数据xorm.NewEngine注意使用从库配置即可：

var SlaveDbConf DbConf = DbConf{
	Host : "127.0.0.1",
	Port : 3306,
	User : "root",
	Pwd : "",
	DbName : "superstar",
    DbCharset: "utf8",
}
复制代码

到了实操阶段，我们在项目根目录下的conf建db.go文件：

conf/db.go

package conf

const DriverName  = "mysql"

type DbConf struct {
	Host string
	Port int
	User string
	Pwd string
	DbName string
}

var MasterDbConf DbConf = DbConf{
	Host : "127.0.0.1",
	Port : 3306,
	User : "root",
	Pwd : "",
	DbName : "superstar",
}

var SlaveDbConf DbConf = DbConf{
	Host : "127.0.0.1",
	Port : 3306,
	User : "root",
	Pwd : "",
	DbName : "superstar",
}
复制代码

连接数据库方法封装

封装好数据库配置之后，为了使链接数据库方法变得一致，有利于日后统一管理。于是把上面提及的数据库链接方式封装成函数方便调用

datasource/dbhelper.go

package datasource

import (
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"log"
	"superstar/conf"
	"sync"
	"xorm.io/xorm"
)

// 数据库单例 dbhelper.go
var (
	masterEngine *xorm.Engine
	slaveEngine  *xorm.Engine
	lock         sync.Mutex
)

func InstanceMaster() *xorm.Engine {
	if masterEngine != nil {
		return masterEngine
	}

	lock.Lock()
	defer lock.Unlock()

	// 协程时，上面十个一起进来，每个都到 lock.Lock() 加锁挂起。这里避免 第一个加锁然后创建实例后，每个都创建实例
	if masterEngine != nil {
		return masterEngine
	}

	c := conf.MasterDbConf
	driverSource := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8",
		c.User, c.Pwd, c.Host, c.Port, c.DbName)
	fmt.Println(driverSource)

	masterEngine, err := xorm.NewEngine(conf.DriverName, driverSource)
	if err != nil {
		log.Fatal("dbhelper.DbInstanceMaster err:", err)
		return nil
	}

	// DEBUG 模式，打印全部sql, 帮助对比 orm 与 sql 执行对照关系
	masterEngine.ShowSQL(false)
	masterEngine.SetTZLocation(conf.SysTimeLocation)

	return masterEngine
}
复制代码