golang主从数据库自动负载均衡代理插件库mssqlx的使用

简介

mssqlx是一个可嵌入、高可用、高性能且轻量级的数据库客户端库，支持Go 1.9及以上版本。主要特性包括：

• 作为sqlx的扩展层
• 自动代理任何主从、主主数据库，兼容Wsrep、Galera集群等
• 自动轻量级轮询负载均衡查询
• 内置对Wsrep、Galera和一些数据库驱动程序的错误处理
• 自动健康检查

安装

go get -u github.com/linxGnu/mssqlx

连接数据库

mssqlx兼容所有database/sql支持的数据库类型。以下是MySQL使用示例：

import (
    _ "github.com/go-sql-driver/mysql"
    "github.com/linxGnu/mssqlx"
)

dsn := "root:123@(%s:3306)/test?charset=utf8&collation=utf8_general_ci&parseTime=true"
masterDSNs := []string{
    fmt.Sprintf(dsn, "172.31.25.233"), // 主节点1地址
    fmt.Sprintf(dsn, "172.31.24.233"), // 主节点2地址(如果有)
    fmt.Sprintf(dsn, "172.31.23.233"), // 主节点3地址(如果有)
}
slaveDSNs := []string{
    fmt.Sprintf(dsn, "172.31.25.234"), // 从节点1地址
    fmt.Sprintf(dsn, "172.31.25.235"), // 从节点2地址
    fmt.Sprintf(dsn, "172.31.25.236"), // 从节点3地址
}

db, _ := mssqlx.ConnectMasterSlaves("mysql", masterDSNs, slaveDSNs)

连接Galera集群

建议设置如下标志：

db, _ := mssqlx.ConnectMasterSlaves("mysql", masterDSNs, slaveDSNs, mssqlx.WithWsrep())

自定义读查询源

读查询将在主节点和从节点之间分配：

db, _ := mssqlx.ConnectMasterSlaves("mysql", masterDSNs, slaveDSNs, mssqlx.WithReadQuerySource(mssqlx.ReadQuerySourceAll))

配置

建议在查询前进行配置：

db.SetMaxIdleConns(20) // 设置所有节点的最大空闲连接数
// db.SetMasterMaxIdleConns(20) // 设置主节点的最大空闲连接数
// db.SetSlaveMaxIdleConns(20) // 设置从节点的最大空闲连接数

db.SetMaxOpenConns(50) // 设置所有节点的最大打开连接数
// db.SetMasterMaxOpenConns(50) 
// db.SetSlaveMaxOpenConns(50)
    
// 如果节点失败，定期检查健康状态以自动重连(毫秒)，默认为500
db.SetHealthCheckPeriod(1000) 
// db.SetMasterHealthCheckPeriod(1000)
// db.SetSlaveHealthCheckPeriod(1000)

查询示例

Select查询

type Person struct {
    FirstName string `db:"first_name"`
    LastName  string `db:"last_name"`
    Email     string
    Data      []byte
}

var people []Person
db.Select(&people, "SELECT * FROM person WHERE id > ? and id < ? ORDER BY first_name ASC", 1, 1000)

Get查询

var person Person
db.Get(&person, "SELECT * FROM person WHERE id = ?", 1)

Queryx查询

// 使用单个结构体循环遍历行
var person Person

rows, err := db.Queryx("SELECT * FROM person") // 或者使用db.QueryxOnMaster(...)
for rows.Next() {
    if err := rows.StructScan(&person); err != nil {
        log.Fatalln(err)
    } 
    fmt.Printf("%#v\n", person)
}

命名查询

// 使用单个结构体循环遍历行
var person Person

rows, err := db.NamedQuery(`SELECT * FROM person WHERE first_name = :fn`, map[string]interface{}{"fn": "Bin"}) // 或者使用db.NamedQueryOnMaster(...)
for rows.Next() {
    if err := rows.StructScan(&person); err != nil {
        log.Fatalln(err)
    } 
    fmt.Printf("%#v\n", person)
}

Exec操作(插入/更新/删除等)

result, err := db.Exec("DELETE FROM person WHERE id < ?", 100)

事务

// 推荐这样写事务
tx, e := db.Begin()
if e != nil {
	return e
}
    
shouldAutoRollBack := true
defer func() {
	if e := recover(); e != nil {
		err = fmt.Errorf("%v", e)
		tx.Rollback()
	} else if err != nil && shouldAutoRollBack {
		tx.Rollback()
	}
}()
			
if _, err = tx.Exec("INSERT INTO person(first_name, last_name, email, data) VALUES (?,?,?,?)", "Jon", "Dow", "jon@gmail", []byte{1, 2}); err != nil {
        return
}
    
if _, err = tx.Exec("INSERT INTO person(first_name, last_name, email, data) VALUES (?,?,?,?)", "Jon", "Snow", "snow@gmail", []byte{1}); err != nil {
    return
}
			
if err = tx.Commit(); err != nil {
    shouldAutoRollBack = false
}

注意事项

• API支持仅在主节点或从节点上执行查询。在主节点上查询的函数名有后缀OnMaster，在从节点上查询的函数名有后缀OnSlave
• 默认情况下，select/show查询会在从节点上负载均衡