Go DB2. 使用MySQL数据库

MySQL驱动

Go中支持MySQL的驱动目前比较多，有如下几种，有些是支持database/sql标准，而有些是采用了自己的实现接口,常用的有如下几种:

* https://github.com/go-sql-driver/mysql 支持database/sql，全部采用go写（推荐）。

* https://github.com/ziutek/mymysql 支持database/sql，也支持自定义的接口，全部采用go写。

* https://github.com/Philio/GoMySQL 不支持database/sql，自定义接口，全部采用go写。

推荐第一个驱动，主要理由：

1）这个驱动比较新，维护的比较好

2）完全支持database/sql接口

3）支持keepalive，保持长连接，虽然mymysql也支持keepalive，但不是线程安全的，mysql从底层就支持了keepalive。

示例代码

1、数据库表结构：数据库test，用户表userinfo，关联用户信息表userdetail

CREATE TABLE `userinfo` (

    `uid` INT(10) NOT NULL AUTO_INCREMENT,

    `username` VARCHAR(64) NULL DEFAULT NULL,

    `department` VARCHAR(64) NULL DEFAULT NULL,

    `created` DATE NULL DEFAULT NULL,

    PRIMARY KEY (`uid`)

);

CREATE TABLE `userdetail` (

    `uid` INT(10) NOT NULL DEFAULT '0',

    `intro` TEXT NULL,

    `profile` TEXT NULL,

    PRIMARY KEY (`uid`)

)

2、使用database/sql接口对数据库表进行增删改查操作：

package main

import (

    "database/sql"

    "fmt"

    _ "github.com/go-sql-driver/mysql"

)

func main() {

    // 连接数据库

    db, err := sql.Open("mysql", "root:123456@/test?charset=utf8")

    checkErr(err)

    defer db.Close()

    // 插入数据

    stmt, err := db.Prepare("INSERT userinfo SET username=?, department=?, create=?")

    checkErr(err)

    res, err := stmt.Exec("lai", "rm", "2018-01-01")

    checkErr(err)

    // 最新记录ID

    id, err := res.LastInsertId()

    checkErr(err)

    fmt.Println(id)

    // 更新数据

    stmt, err = db.Prepare("UPDATE userinfo SET username=? WHERE uid=?")

    checkErr(err)

    res, err = stmt.Exec("laiupdate", id)

    checkErr(err)

    affect, err := res.RowsAffected()

    checkErr(err)

    fmt.Println(affect)

    // 查询数据

    rows, err := db.Query("SELECT * FROM userinfo")

    checkErr(err)

    for rows.Next() {

        var uid int

        var username string

        var department string

        var created string

        err = rows.Scan(&uid, &username, &department, &created)

        checkErr(err)

        fmt.Println(uid)

        fmt.Println(username)

        fmt.Println(department)

        fmt.Println(created)

    }

    // 删除数据

    stmt, err = db.Prepare("DELETE FROM userinfo WHERE uid=?")

    checkErr(err)

    res, err = stmt.Exec(id)

    checkErr(err)

    affect, err = res.RowsAffected()

    checkErr(err)

    fmt.Println(affect)

}

func checkErr(err error) {

    if err != nil {

        panic(err)

    }

}

通过上面的代码我们可以看出，Go操作MySQL数据库是很方便的。

关键几个函数：

sql.Open() 函数用来打开一个注册过的数据库驱动，go-sql-driver中注册了mysql这个数据库驱动，第二个参数是DSN(Data Source Name)，它是go-sql-driver定义的一些数据库链接和配置信息。它支持如下格式：

user@unix(/path/to/socket)/dbname?charset=utf8

user:password@tcp(localhost:5555)/dbname?charset=utf8

user:password@/dbname（推荐）

user:password@tcp([de:ad:be:ef::ca:fe]:80)/dbname

db.Prepare() 函数用来返回准备要执行的sql操作，然后返回准备完毕的执行状态。

stmt.Exec() 函数用来执行stmt准备好的SQL语句。

db.Query() 函数用来直接执行Sql返回Rows结果。

我们可以看到传入的参数都是=?对应的数据，这样做的方式可以一定程度上防止SQL注入。

范例

Go database/sql tutorial 提供了惯用的范例及详细的说明。