MyMySQL 的 database/sql 接口使用

最近看到不少朋友对 Golang 中操作 MySQL 数据库有疑问，那么就此内容给大家分享一下吧。

MyMySQL 的原作者是来自波兰的 ziutek，他根据 MySQL 的协议标准使用 Golang 实现了 MyMySQL 包。根据他的介绍，这个包可以用在 MySQL 4.1 或更高版本上，并且在 5.0、5.1 版本上经过项目的实际验证。

只要用 MySQL 做过项目的朋友一定遇到过编码问题，set names 几乎成为了中文环境下使用 MySQL 的标配。那么 Golang 也不能例外。不过由于 database/sql 使用了随机的连接池，且未提供任何方法让所有连接都执行某个操作，于是这个麻烦事儿就落到了驱动的肩上。对于中文什么的，大老外一向不怎么在意。我提交了这个 issue 给 ziutek，不过对于他最终实现的方案不怎么满意。需要额外编写godrv.Register 来对编码进行设置。

同时当连接空闲时间超出服务器的 wait_timeout 时，会出现 broken pipe。对于这个问题，ziutek 的建议是使用 MyMySQL 的 autorc，但是 godrv 并没有基于 autorc 实现 database/sql。换句话说，要解决这个错误，只能使用 MyMySQL 原始接口。

鉴于这些问题，我只能 fork 了 ziutek 的项目，对 MyMySQL 打上相应的补丁。主要是针对编码设置进行了改进，并顺便实现了 keepalive 功能。

本文主要介绍 fork 的版本的使用。

在 fork 的版本中，还有一个改进就是统一了 go get 的安装接口，无需再逐一安装 MyMySQl 的子包，只需要

go get github.com/mikespook/mymysql

即可完成安装。

关于使用，还是用代码说话吧，完整代码看这里。为了通用，这里只介绍使用 database/sql 接口的使用。在实际项目中，我也建议大家尽量使用该接口，这样在最大可用性保障下保持兼容性。

import(
    "log"
    "database/sql"
    _ "github.com/mikespook/mymysql/godrv"
)

使用 MyMySQL 的 database/sql 接口要导入 godrv 包。在这个包的 init 函数中自动注册了 database/sql 的驱动，所以导入后无需再使用。

db, err := sql.Open("mymysql", "tcp://127.0.0.1:3306/test/root/xxiyy?charset=utf8&keepalive=1200")
if err != nil {
    log.Fatal(err)
}
defer db.Close()

使用 mymysql 驱动打开一个 sql.DB 连接。dsn 连接串是我对 ziutek 的 MyMySQL 的主要改进。支持多种连接方式：

• 使用 tcp 协议：[tcp://addr/]dbname/user/password[?params]
• 使用 unix sock：[unix://sockpath/]dbname/user/password[?params]

其中 tcp 协议的 addr 必须是含有主机名或 ip，且包含又分号分隔的端口号的字符串。如 localhost:3306、192.168.3.2:3307。unix sock 协议中的 sockpath 必须是 MySQL 的 sock 文件的绝对路径。

在上述 dsn 中，方括号内的是可选的内容，可省略。当协议信息（第一个方括号内）未指定时，使用 tcp://127.0.0.1:3306/ 作为默认值。

params 部分是用于当前数据库驱动的参数设置，当前来说，只有两个可设置的参数：

• charset：用于 ‘set names’ 设置连接编码。
• keepalive：每 keepalive 秒向服务器发送 PING。

需要特别强调的是，如果密码含有斜线（/），由于解析规则的缘故，需要用星号（*）代替。如果密码含有星号（*），则需要用两个星号（**）代替。例如：

• 原密码 [pass/wd]，在 dsn 中应写为 [pass*wd]。
• 原密码 [pass*wd]，在 dsn 中应写为 [pass**wd]。

stmt, err := db.Prepare("insert into `test` (`key`, `value`) values (?, ?)")
if err != nil {
    log.Fatal(err)
}
defer stmt.Close()
rslt, err := stmt.Exec("name", "foobar")
if err != nil {
    log.Fatal(err)
}
if a, err := rslt.RowsAffected(); err != nil {
    log.Print(err)
} else {
    log.Printf("[INS]Affected rows=%d", a)
}
if id, err := rslt.LastInsertId(); err != nil {
    log.Print(err)
} else {
    log.Printf("[INS]Last insert id=%d", id)
}

在 database/sql 接口里，有多种办法执行某个 SQL 语句。出于安全考虑，强烈建议大家不要使用字符串方式直接拼接 SQL 语句进行执行，如果可能，在大多数其他接口中都会出现的 Prepare 方法是首选。执行完 SQL 语句后，可以通过 sql.Result 得到这次执行所影响的行数，和最后插入的 Id 值。如果表中没有使用 autoincrement 作为主键，那么这个 Id 值永远为 0。

rows, err := db.Query("select * from `test`")
if err != nil {
    log.Fatal(err)
}
defer rows.Close()
for rows.Next() {
    var k, v string
    rows.Scan(&k, &v)
    log.Printf("[ROWS]key=%s, value=%s", k, v)
}

查询一个二维表，从中取得多行数据使用 Query 方法。需要注意的是 sql.Rows 的 Scan 方法接收的参数必须是指针。也就是说存放数据的地址要在 Scan 调用前准备好。Scan 内部使用了反射来识别参数类型并进行赋值，也可以传递一个 []interface{}。这时必须这样调用：rows.Scan(a…) 才能得到预期结果。

row := db.QueryRow("select * from `test` where `key` = ?", "name")
var k, v string
row.Scan(&k, &v)
log.Printf("[ROW]key=%s, value=%s", k, v)

对于有些如用主键进行的查询，仅仅会读出一行数据。那么可以用 QueryRow 进行更加简化的读取。

rslt, err = db.Exec("delete from `test`")
if a, err := rslt.RowsAffected(); err != nil {
    log.Print(err)
} else {
    log.Printf("[DEL]Affected rows=%d", a)
}
if id, err := rslt.LastInsertId(); err != nil {
    log.Print(err)
} else {
    log.Printf("[DEL]Last insert id=%d", id)
}

对于没有拼接 SQL 需要的操作，可以直接在 sql.DB 对象上调用 Exec 方法。返回的 sql.Result 与 sql.Stmt 的 Exec 方法一致。

关于基础的使用，大致就这些内容。database/sql 也支持事物，这部分信息大家可以参考其文档进行学习。