go 操作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寫。

接下來的例子我主要以第一個驅動爲例(我目前項目中也是採用它來驅動)，也推薦大家採用它，主要理由：

• 這個驅動比較新，維護的比較好
• 完全支持database/sql接口
• 支持keepalive，保持長連接,雖然星星fork的mymysql也支持keepalive，但不是線程安全的，這個從底層就支持了keepalive。

示例代碼

接下來的幾個小節裏面我們都將採用同一個數據庫表結構：數據庫test，用戶表userinfo，關聯用戶信息表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接口對數據庫表進行增刪改查操作

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)

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

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

    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("astaxieupdate", 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)

    db.Close()

}

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操作，然後返回準備完畢的執行狀態。

db.Query()函數用來直接執行Sql返回Rows結果。

stmt.Exec()函數用來執行stmt準備好的SQL語句

我們可以看到我們傳入的參數都是=?對應的數據，這樣做的方式可以一定程度上防止SQL注入。