【mysql】golang鏈接mysql操做示例增刪改查

golang自己沒有提供鏈接mysql的驅動，可是定義了標準接口供第三方開發驅動。這裏鏈接mysql能夠使用第三方庫，第三方庫推薦使用https://github.com/Go-SQL-Driver/MySQL這個驅動，更新維護都比較好。下面演示下具體的使用，完整代碼示例能夠參考最後。

下載驅動

sudo go get github.com/go-sql-driver/mysql

若是提示這樣的失敗信息：cannot download, $GOPATH not set. For more details see: go help gopath，能夠使用以下命令解決

sudo env GOPATH=/Users/chenjiebin/golang go get github.com/go-sql-driver/mysql

GOPATH的值根據自行環境進行替換。

建立測試表

在mysql test庫中建立測試表

CREATE TABLE IF NOT EXISTS `test`.`user` (
 `user_id` INT(11) UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '用戶編號',
 `user_name` VARCHAR(45) NOT NULL COMMENT '用戶名稱',
 `user_age` TINYINT(3) UNSIGNED NOT NULL DEFAULT 0 COMMENT '用戶年齡',
 `user_sex` TINYINT(3) UNSIGNED NOT NULL DEFAULT 0 COMMENT '用戶性別',
 PRIMARY KEY (`user_id`))
 ENGINE = InnoDB
 AUTO_INCREMENT = 1
 DEFAULT CHARACTER SET = utf8
 COLLATE = utf8_general_ci
 COMMENT = '用戶表'

數據庫鏈接

數據庫鏈接使用datebase/sql Open函數進行鏈接

db, err := sql.Open("mysql", "user:password@tcp(localhost:5555)/dbname?charset=utf8")

其中鏈接參數能夠有以下幾種形式：

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

一般咱們都用第二種。

插入操做

stmt, err := db.Prepare(`INSERT user (user_name,user_age,user_sex) values (?,?,?)`)
checkErr(err)

res, err := stmt.Exec("tony", 20, 1)
checkErr(err)

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

fmt.Println(id)

這裏使用結構化操做，不推薦使用直接拼接sql語句的方法。

查詢操做

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


for rows.Next() {

    var userId int

    var userName string

    var userAge int

    var userSex int

    rows.Columns()

    err = rows.Scan(&userId, &userName, &userAge, &userSex)

    checkErr(err)

    fmt.Println(userId)

    fmt.Println(userName)

    fmt.Println(userAge)

    fmt.Println(userSex)

}

這裏查詢的方式使用聲明4個獨立變量userId、userName、userAge、userSex來保存查詢出來的每一行的值。在實際開發中一般會封裝數據庫的操做，對這樣的查詢一般會考慮返回字典類型。

//構造scanArgs、values兩個數組，scanArgs的每一個值指向values相應值的地址
columns, _ := rows.Columns()
scanArgs := make([]interface{}, len(columns))
values := make([]interface{}, len(columns))

for i := range values {
    scanArgs[i] = &values[i]
}


for rows.Next() {
    //將行數據保存到record字典
    err = rows.Scan(scanArgs...)
    record := make(map[string]string)

    for i, col := range values {
        if col != nil {
            record[columns[i]] = string(col.([]byte))
        }
    }
    fmt.Println(record)
}

修改操做

stmt, err := db.Prepare(`UPDATE user SET user_age=?,user_sex=? WHERE user_id=?`)
checkErr(err)

res, err := stmt.Exec(21, 2, 1)
checkErr(err)

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

fmt.Println(num)

刪除操做

stmt, err := db.Prepare(`DELETE FROM user WHERE user_id=?`)
checkErr(err)

res, err := stmt.Exec(1)
checkErr(err)

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

fmt.Println(num)

修改和刪除操做都比較簡單，同插入數據相似，只是使用RowsAffected來獲取影響的數據行數。

完整代碼

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

func main() {
    insert()
}

//插入demo
func insert() {

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

    stmt, err := db.Prepare(`INSERT user (user_name,user_age,user_sex) values (?,?,?)`)
    checkErr(err)

    res, err := stmt.Exec("tony", 20, 1)
    checkErr(err)

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

    fmt.Println(id)

}


//查詢demo
func query() {

    db, err := sql.Open("mysql", "root:@/test?charset=utf8")
    checkErr(err)
    rows, err := db.Query("SELECT * FROM user")
    checkErr(err)

    //普通demo

    //for rows.Next() {

    //    var userId int
    //    var userName string
    //    var userAge int
    //    var userSex int

    //    rows.Columns()

    //    err = rows.Scan(&userId, &userName, &userAge, &userSex)
    //    checkErr(err)



    //    fmt.Println(userId)

    //    fmt.Println(userName)

    //    fmt.Println(userAge)

    //    fmt.Println(userSex)

    //}



    //字典類型

    //構造scanArgs、values兩個數組，scanArgs的每一個值指向values相應值的地址

    columns, _ := rows.Columns()

    scanArgs := make([]interface{}, len(columns))

    values := make([]interface{}, len(columns))

    for i := range values {
        scanArgs[i] = &values[i]
    }


    for rows.Next() {
        //將行數據保存到record字典

        err = rows.Scan(scanArgs...)
        record := make(map[string]string)

        for i, col := range values {
            if col != nil {
                record[columns[i]] = string(col.([]byte))
            }
        }
        fmt.Println(record)
    }
}

//更新數據
func update() {

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


    stmt, err := db.Prepare(`UPDATE user SET user_age=?,user_sex=? WHERE user_id=?`)
    checkErr(err)

    res, err := stmt.Exec(21, 2, 1)
    checkErr(err)

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

    fmt.Println(num)

}



//刪除數據
func remove() {
    db, err := sql.Open("mysql", "root:@/test?charset=utf8")
    checkErr(err)

    stmt, err := db.Prepare(`DELETE FROM user WHERE user_id=?`)
    checkErr(err)

    res, err := stmt.Exec(1)
    checkErr(err)

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

    fmt.Println(num)
}

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

小結

總體上來講都比較簡單，就是查詢那邊使用字典來存儲返回數據比較複雜一些。既然說到數據庫鏈接，一般應用中都會使用鏈接池來減小鏈接開銷，關於鏈接池下次整理一下再放上來。

參考資料：《go web編程》，go web編程中對數據庫鏈接作了比較詳細的解說，值得一看。