Go 使用MySQL數(shù)據(jù)庫

目前Internet上流行的網(wǎng)站構(gòu)架方式是LAMP，其中的M即MySQL, 作為數(shù)據(jù)庫，MySQL以免費(fèi)、開源、使用方便為優(yōu)勢(shì)成為了很多Web開發(fā)的后端數(shù)據(jù)庫存儲(chǔ)引擎。

MySQL驅(qū)動(dòng)

Go中支持MySQL的驅(qū)動(dòng)目前比較多，有如下幾種，有些是支持database/sql標(biāo)準(zhǔn)，而有些是采用了自己的實(shí)現(xiàn)接口,常用的有如下幾種:

• 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寫。

接下來的例子我主要以第一個(gè)驅(qū)動(dòng)為例(我目前項(xiàng)目中也是采用它來驅(qū)動(dòng))，也推薦大家采用它，主要理由：

• 這個(gè)驅(qū)動(dòng)比較新，維護(hù)的比較好
• 完全支持database/sql接口
• 支持keepalive，保持長(zhǎng)連接,雖然星星fork的mymysql也支持keepalive，但不是線程安全的，這個(gè)從底層就支持了keepalive。

示例代碼

接下來的幾個(gè)小節(jié)里面我們都將采用同一個(gè)數(shù)據(jù)庫表結(jié)構(gòu)：數(shù)據(jù)庫test，用戶表userinfo，關(guān)聯(lián)用戶信息表userdetail。

CREATE TABLE `userinfo` (
    `uid` INT(10) NOT NULL AUTO_INCREMENT,
    `username` VARCHAR(64) NULL DEFAULT NULL,
    `departname` 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`)
)

如下示例將示范如何使用database/sql接口對(duì)數(shù)據(jù)庫表進(jìn)行增刪改查操作

package main

import (
    _ "github.com/go-sql-driver/mysql"
    "database/sql"
    "fmt"
    //"time"
)

func main() {
    db, err := sql.Open("mysql", "astaxie:astaxie@/test?charset=utf8")
    checkErr(err)

    //插入數(shù)據(jù)
    stmt, err := db.Prepare("INSERT userinfo SET username=?,departname=?,created=?")
    checkErr(err)

    res, err := stmt.Exec("astaxie", "研發(fā)部門", "2012-12-09")
    checkErr(err)

    id, err := res.LastInsertId()
    checkErr(err)

    fmt.Println(id)
    //更新數(shù)據(jù)
    stmt, err = db.Prepare("update userinfo set username=? where uid=?")
    checkErr(err)

    res, err = stmt.Exec("astaxieupdate", id)
    checkErr(err)

    affect, err := res.RowsAffected()
    checkErr(err)

    fmt.Println(affect)

    //查詢數(shù)據(jù)
    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)
    }

    //刪除數(shù)據(jù)
    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)

    db.Close()

}

func checkErr(err error) {
    if err != nil {
        panic(err)
    }
}

通過上面的代碼我們可以看出，Go操作Mysql數(shù)據(jù)庫是很方便的。

關(guān)鍵的幾個(gè)函數(shù)我解釋一下：

sql.Open()函數(shù)用來打開一個(gè)注冊(cè)過的數(shù)據(jù)庫驅(qū)動(dòng)，go-sql-driver中注冊(cè)了mysql這個(gè)數(shù)據(jù)庫驅(qū)動(dòng)，第二個(gè)參數(shù)是DSN(Data Source Name)，它是go-sql-driver定義的一些數(shù)據(jù)庫鏈接和配置信息。它支持如下格式：

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()函數(shù)用來返回準(zhǔn)備要執(zhí)行的sql操作，然后返回準(zhǔn)備完畢的執(zhí)行狀態(tài)。

db.Query()函數(shù)用來直接執(zhí)行Sql返回Rows結(jié)果。

stmt.Exec()函數(shù)用來執(zhí)行stmt準(zhǔn)備好的SQL語句

我們可以看到我們傳入的參數(shù)都是=?對(duì)應(yīng)的數(shù)據(jù)，這樣做的方式可以一定程度上防止SQL注入。