Golang 学习笔记（10）—— mysql操作

Golang

go-sql-driver/mysql

go操作mysql的驱动包很多，这里讲解当下比较流行的go-sql-driver/mysql

安装

执行下面两个命令：

go get github.com/go-sql-driver/mysql  //下载
go install github.com/go-sql-driver/mysql //安装

安装完成以后的文件截图

安装完成

使用

package

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

数据库

在mysql中建一张测试的表，sql如下：

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`)
)

连接

db, err := sql.Open("mysql", "用户名:密码@tcp(IP:端口)/数据库?charset=utf8")

insert

有2种方法。

直接使用Exec函数添加

result, err := db.Exec("INSERT INTO userinfo (username, departname, created) VALUES (?, ?, ?)","chain","dev","2018-01-04")

首先使用Prepare获得stmt，然后调用Exec添加

stmt, err := db.Prepare("INSERT userinfo SET username=?,departname=?,created=?")
res, err := stmt.Exec("iris", "test", "2018-01-04")

另一个经常用到的功能，获得刚刚添加数据的自增ID

id, err := res.LastInsertId()

示例

package main

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

var (
    dbhost = "xxx.xxx.xxx.xxx:3306"
    dbusername = "xxxx"
    dbpassword = "xxxx"
    dbname = "xxx"
)


func main(){
    Insert("chain", "dev", "1")
    Insert("chain", "dev", "2")
    Insert("iris", "test", "1")
    Insert("iris", "test", "2")
}
/*
  获取sql.DB对象
*/
func GetDB() *sql.DB{
    db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8", dbusername, dbpassword, dbhost, dbname))
    CheckErr(err)
    return db
}

/*
  插入数据
*/
func Insert(username, departname, method string)bool{
    db := GetDB()
    defer db.Close()

    if method == "1"{
        _, err := db.Exec("insert into userinfo(username,departname,created) values(?,?,?)",username,departname,time.Now())
        if err != nil{
            fmt.Println("insert err: ", err.Error())
            return false
        }
        fmt.Println("insert success!")
        return true
    }else if method == "2"{
        stmt, err := db.Prepare("INSERT userinfo SET username=?,departname=?,created=?")
        if err != nil{
            fmt.Println("insert prepare error: ", err.Error())
            return false
        }
        _, err = stmt.Exec(username, departname, time.Now())
        if err != nil{
            fmt.Println("insert exec error: ", err.Error())
            return false
        }
        fmt.Println("insert success!")
        return true
    }
    return false
}

运行结果

数据库结果

delete

与insert所用的方法一致，只是将sql语句改为对应的功能就行。

func main(){
    Delete(15)
}
/*
  根据id删除数据
*/
func Delete(id int) bool {
    db := GetDB()
    defer db.Close()

    stmt, err := db.Prepare("delete from userinfo where uid=?")
    if err != nil{
        fmt.Println("delete prepare error: ", err.Error())
        return false
    }
    _, err = stmt.Exec(id)
    if err != nil{
        fmt.Println("delete exec error: ", err.Error())
        return false
    }
    fmt.Println("delete success!")
    return true
}

运行结果

数据库结果

update

与insert所用的方法一致，只是将sql语句改为对应的功能就行。

func main(){
    UpdateName(13,"chairis")
}
/*
  根据id，修改名称
*/
func UpdateName(id int, name string)bool{
    db := GetDB()
    defer db.Close()

    stmt, err := db.Prepare("update userinfo set username=? where uid=?")
    if err != nil{
        fmt.Println("update name prepare error: ", err.Error())
        return false
    }
    _, err = stmt.Exec(name, id)
    if err != nil{
        fmt.Println("update name exec error: ", err.Error())
        return false
    }
    fmt.Println("update name success!")
    return true
}

运行结果

数据库结果

select

查询单条数据，QueryRow 函数

func main(){
    GetOne(13)
}
func GetOne(id int){
    db := GetDB()
    defer db.Close()
    var username, departname, created string
    err := db.QueryRow("select username, departname, created from userinfo where uid=?",id).Scan(&username, &departname, &created)
    if err != nil{
        fmt.Println("get one error: ", err.Error())
        return
    }
    fmt.Println("username: ", username, "departname: ", departname, "created: ", created)
}

运行结果

查询多条数据，并遍历
Query 获取数据，for xxx.Next() 遍历数据

func main(){
    GetAll()
}
func GetAll(){
    db := GetDB()
    defer db.Close()
    rows, err := db.Query("select username, departname, created from userinfo")
    if err != nil{
        fmt.Println("get all error: ", err.Error())
        return
    }
    for rows.Next(){
        var username, departname, created string
        if err := rows.Scan(&username, &departname, &created); err == nil{
            fmt.Println("username: ", username, "departname: ", departname, "created: ", created)
        }
    }
}

运行结果

事务

在操作数据库之前执行，db.Begin()
例：tx, err := db.Begin()
保存到数据库：err := tx.Commit()
回滚：err := tx.Rollback()
注意设置事务以后操作数据库就不是db了，而是tx
请看以下示例：

func main(){
    Trans()
}

func Trans(){
    db := GetDB()
    defer db.Close();

    tx, err := db.Begin()
    if err != nil{
        fmt.Println("db.Begin error: ", err.Error())
        return
    }
    isCommit := true
    defer func(){
        if isCommit{
            tx.Commit()
            fmt.Println("commit")
        }else{
            tx.Rollback()
            fmt.Println("Rollback")
        }
    }()
    _, err = tx.Exec("insert into userinfo(username,departname,created) values(?,?,?)","username","departname",time.Now())
    if err != nil{
        isCommit = false
    }
    _, err = tx.Exec("insert into userinfo(username,departname,created) values(?,?,?)","username","departname",time.Now())
    if err != nil{
        isCommit = false
    }
    _, err = tx.Exec("insert into userinfo(username,departname,created) values(?,?,?)","username","departname",time.Now())
    if err != nil{
        isCommit = false
    }
}

运行结果

全部代码

package main

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

var (
    dbhost = "xxx.xxx.xxx.xxx:3306"
    dbusername = "xxxxxx"
    dbpassword = "xxxxxx"
    dbname = "xxxxxx"
)


func main(){
    Trans()
}

func Trans(){
    db := GetDB()
    defer db.Close();

    tx, err := db.Begin()
    if err != nil{
        fmt.Println("db.Begin error: ", err.Error())
        return
    }
    isCommit := true
    defer func(){
        if isCommit{
            tx.Commit()
            fmt.Println("commit")
        }else{
            tx.Rollback()
            fmt.Println("Rollback")
        }
    }()
    _, err = tx.Exec("insert into userinfo(username,departname,created) values(?,?,?)","username","departname",time.Now())
    if err != nil{
        isCommit = false
    }
    _, err = tx.Exec("insert into userinfo(username,departname,created) values(?,?,?)","username","departname",time.Now())
    if err != nil{
        isCommit = false
    }
    _, err = tx.Exec("insert into userinfo(username,departname,created) values(?,?,?)","username","departname",time.Now())
    if err != nil{
        isCommit = false
    }
}

/*
  获取sql.DB对象
*/
func GetDB() *sql.DB{
    db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8", dbusername, dbpassword, dbhost, dbname))
    CheckErr(err)
    return db
}

/*
  插入数据
*/
func Insert(username, departname, method string)bool{
    db := GetDB()
    defer db.Close()

    if method == "1"{
        _, err := db.Exec("insert into userinfo(username,departname,created) values(?,?,?)",username,departname,time.Now())
        if err != nil{
            fmt.Println("insert err: ", err.Error())
            return false
        }
        fmt.Println("insert success!")
        return true
    }else if method == "2"{
        stmt, err := db.Prepare("INSERT userinfo SET username=?,departname=?,created=?")
        if err != nil{
            fmt.Println("insert prepare error: ", err.Error())
            return false
        }
        _, err = stmt.Exec(username, departname, time.Now())
        if err != nil{
            fmt.Println("insert exec error: ", err.Error())
            return false
        }
        fmt.Println("insert success!")
        return true
    }
    return false
}

/*
  根据id，修改名称
*/
func UpdateName(id int, name string)bool{
    db := GetDB()
    defer db.Close()

    stmt, err := db.Prepare("update userinfo set username=? where uid=?")
    if err != nil{
        fmt.Println("update name prepare error: ", err.Error())
        return false
    }
    _, err = stmt.Exec(name, id)
    if err != nil{
        fmt.Println("update name exec error: ", err.Error())
        return false
    }
    fmt.Println("update name success!")
    return true
}

/*
  根据id删除数据
*/
func Delete(id int) bool {
    db := GetDB()
    defer db.Close()

    stmt, err := db.Prepare("delete from userinfo where uid=?")
    if err != nil{
        fmt.Println("delete prepare error: ", err.Error())
        return false
    }
    _, err = stmt.Exec(id)
    if err != nil{
        fmt.Println("delete exec error: ", err.Error())
        return false
    }
    fmt.Println("delete success!")
    return true
}

func GetOne(id int){
    db := GetDB()
    defer db.Close()
    var username, departname, created string
    err := db.QueryRow("select username, departname, created from userinfo where uid=?",id).Scan(&username, &departname, &created)
    if err != nil{
        fmt.Println("get one error: ", err.Error())
        return
    }
    fmt.Println("username: ", username, "departname: ", departname, "created: ", created)
}

func GetAll(){
    db := GetDB()
    defer db.Close()
    
    rows, err := db.Query("select username, departname, created from userinfo")
    if err != nil{
        fmt.Println("get all error: ", err.Error())
        return
    }
    for rows.Next(){
        var username, departname, created string
        if err := rows.Scan(&username, &departname, &created); err == nil{
            fmt.Println("username: ", username, "departname: ", departname, "created: ", created)
        }
    }
    
}

func CheckErr(err error){
    if err != nil{
        fmt.Println("err: ", err.Error())
        panic(err)
    }
}

源码

github 源码地址

完

转载请注明出处：
Golang 学习笔记（10）—— mysql操作

目录
 上一节：Golang 学习笔记（09）—— json和xml解析
 下一节：Golang 学习笔记（11）—— 反射

最后编辑于：2018.01.14 12:41:41

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 159,117评论 4赞 362
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 67,328评论 1赞 293
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 108,839评论 0赞 243
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 44,007评论 0赞 206
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 52,384评论 3赞 287
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 40,629评论 1赞 219
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 31,880评论 2赞 313
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 30,593评论 0赞 198
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 34,313评论 1赞 243
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 30,575评论 2赞 246
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 32,066评论 1赞 260
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 28,392评论 2赞 253
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 33,052评论 3赞 236
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 26,082评论 0赞 8
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 26,844评论 0赞 195
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 35,662评论 2赞 274
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 35,575评论 2赞 270

Golang 学习笔记（10）—— mysql操作

go-sql-driver/mysql

安装

使用

package

数据库

连接

insert

delete

update

select

事务

全部代码

源码

完

推荐阅读更多精彩内容