47. 访问MySql数据库实现增删改查

作为服务端程序，对数据库的访问是很常见的操作。我们来熟悉一下
go 语言访问 MySql 数据库的基本操作（增删改查）。
数据库访问需要用到标准库 "database/sql" 和 mysql 的驱动 "github.com/go-sql-driver/mysql" 。这两个包都需要引用。mysql 的驱动因为只是需要它的 init() 初始化，所以需要采用下划线引用的方式。

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

在访问数据库前，我们先在 MySql 里建好表并预先插入一些数据以便测试程序。请在 MySql 里执行下面的 SQL 脚本。

-- ----------------------------
-- Table structure for announcement
-- ----------------------------
DROP TABLE IF EXISTS `announcement`;
CREATE TABLE `announcement` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `imgUrl` varchar(128) COLLATE utf8_unicode_ci DEFAULT NULL,
  `detailUrl` varchar(128) COLLATE utf8_unicode_ci DEFAULT NULL,
  `createDate` varchar(10) COLLATE utf8_unicode_ci DEFAULT NULL,
  `state` int(11) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci;

-- ----------------------------
-- Records of announcement
-- ----------------------------
INSERT INTO `announcement` VALUES ('1', '/visitshop/img/ann/ann1.jpg', null, '2016-07-20', '0');
INSERT INTO `announcement` VALUES ('2', '/visitshop//img/ann/ann1.jpg', null, '2016-07-20', '0');
INSERT INTO `announcement` VALUES ('3', '/visitshop//img/ann/ann1.jpg', null, '2016-07-20', '0');
INSERT INTO `announcement` VALUES ('4', '/visitshop//img/ann/ann1.jpg', null, '2016-07-20', '0');

我们将在主函数里调用增删改查函数。先预设这四种操作的函数名

    query()    //查询
    query2()  //查询
    insert()    //插入
    update()  //修改
    remove()  //删除

然后分别实现这几个函数。
首先是 query() 查询数据。
要想访问数据库，首先要打开数据库链接。这就需要用到datebase/sql Open函数。

db, err := sql.Open("mysql", "root:@/shopvisit")

我们的数据库链接是这个样子的。为了简化操作，我的数据库用户 root 是没有密码的。而正常的数据库链接应该是这个样子的

db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/shopvisit")

这个链接说明数据库用户 root 的密码是 123456，使用 tcp 协议，数据库 ip 地址是 127.0.0.1，使用 3306 端口作为通讯端口。当前使用的库名是 shopvisit 。
当然链接数据库的方式其实是有好几种的

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

有兴趣的话，可以都试一试。
既然有了数据库链接的语句，就要有错误检查。而错误检查会较为频繁的出现（go 语言的特色）
所以写一个函数来直接处理它。

func check(err error) {
    if err != nil{
        fmt.Println(err)
        panic(err)
    }
}

这样每当需要对错误进行检查的时候，就执行 check(err)
当 db 链接数据库正常之后，我们可以执行 sql 查询语句了。

    rows, err := db.Query("SELECT * FROM shopvisit.announcement")
    check(err)

然后利用 for 循环遍历返回的结果。

for rows.Next() {

在循环体内，我们先取得记录的列（字段），把列名参数的值和列地址关联。

        columns, _ := rows.Columns()

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

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

再把数据保存到 record 字典中

        //将数据保存到 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) 之后，一定要记得释放资源

rows.Close()

养成好习惯，麻烦会很少。
看一下 query() 的完整代码

func query() {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    rows, err := db.Query("SELECT * FROM shopvisit.announcement")
    check(err)

    for rows.Next() {
        columns, _ := rows.Columns()

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

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

        //将数据保存到 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)
    }
    rows.Close()

}

在 main 函数中，注释掉其他函数，只留下 query()，运行d看结果。多运行几次，比较每次运行结果。
第一次运行：

map[state:0 id:1 imgUrl:/visitshop/img/ann/ann1.jpg createDate:2016-07-20]
map[id:2 imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0]
map[id:3 imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0]
map[createDate:2016-07-20 state:0 id:4 imgUrl:/visitshop//img/ann/ann1.jpg]

第二次运行：

map[id:1 imgUrl:/visitshop/img/ann/ann1.jpg createDate:2016-07-20 state:0]
map[id:2 imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0]
map[id:3 imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0]
map[id:4 imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0]

第三次运行：

map[createDate:2016-07-20 state:0 id:1 imgUrl:/visitshop/img/ann/ann1.jpg]
map[id:2 imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0]
map[imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0 id:3]
map[id:4 imgUrl:/visitshop//img/ann/ann1.jpg createDate:2016-07-20 state:0]

为什么每次都不一样呢？这是因为我们使用了 map 字典来保存列。map 是无序的，所以每次都是随机的显示顺序。
这显然不符合我们一般的结果需要，那么，我们来编写 query2()。
仍然是链接数据库的语句作为开端，再跟着是查询语句。

    db, err := sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/shopvisit?charset=utf8")
    check(err)

    rows, err := db.Query("SELECT id,imgUrl,createDate,state FROM announcement")
    check(err)

既然已经查询无错了，那么直接 for 循环记录结果

    for rows.Next(){
        var id int
        var state int
        var imgUrl string
        var createDate string
        //注意这里的Scan括号中的参数顺序，和 SELECT 的字段顺序要保持一致。
        if err := rows.Scan(&id,&imgUrl,&createDate,&state); err != nil {
            log.Fatal(err)
        }
        fmt.Printf("%s id is %d on %s with state %d\n", imgUrl, id, createDate, state)
    }

没一次循环的时候，我们都可以按照我们想要的顺序，取得所有的字段值。唯一需要注意的是， rows.Scan 的参数顺序，需要和 select 语句的字段保持顺序一致。这里主要指的是数据类型。参数名可以不同。
这里做个比较

db.Query("SELECT id,imgUrl,createDate,state FROM announcement")
rows.Scan(&id,&imgUrl,&createDate,&state);

&符号是取变量的地址，注意观察变量的数据类型和 select 后面参数的数据类型必须是一致的。声明变量时的顺序无所谓，Scan调用变量时的顺序要注意 select 参数的顺序一致。

        var id int
        var state int
        var imgUrl string
        var createDate string

然后可以按照你想要的顺序打印输出

fmt.Printf("%s id is %d on %s with state %d\n", imgUrl, id, createDate, state)

完整 query2() 函数代码

func query() {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    rows, err := db.Query("SELECT * FROM shopvisit.announcement")
    check(err)

    for rows.Next() {
        columns, _ := rows.Columns()

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

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

        //将数据保存到 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)
    }
    rows.Close()

}
func query2()  {
    fmt.Println("Query2")
    db, err := sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/shopvisit?charset=utf8")
    check(err)

    rows, err := db.Query("SELECT id,imgUrl,createDate,state FROM announcement")
    check(err)

    for rows.Next(){
        var id int
        var state int
        var imgUrl string
        var createDate string
        //注意这里的Scan括号中的参数顺序，和 SELECT 的字段顺序要保持一致。
        if err := rows.Scan(&id,&imgUrl,&createDate,&state); err != nil {
            log.Fatal(err)
        }
        fmt.Printf("%s id is %d on %s with state %d\n", imgUrl, id, createDate, state)
    }

    if err := rows.Err(); err != nil {
        log.Fatal(err)
    }
    rows.Close()
}

修改 main 函数中的当前可执行函数为 query2()，运行结果如下

Query2
/visitshop/img/ann/ann1.jpg id is 1 on 2016-07-20 with state 0
/visitshop//img/ann/ann1.jpg id is 2 on 2016-07-20 with state 0
/visitshop//img/ann/ann1.jpg id is 3 on 2016-07-20 with state 0
/visitshop//img/ann/ann1.jpg id is 4 on 2016-07-20 with state 0

这回每次的运行结果就稳定了。
插入数据 insert() 函数，链接数据库还是一样的，而 db 调用的函数改成了 Prepare 。执行的 sql 语句需要这样写

    stmt, err := db.Prepare(`INSERT announcement (imgUrl, detailUrl, createDate, state) VALUES (?, ?, ?, ?)`)
    check(err)

插入的值，必须按照 sql 顺序来插入

    res, err := stmt.Exec("/visitshop/img/ann/cofox1.png",nil,"2017-09-06",0)
    check(err)

返回刚插入的这条记录的 id

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

完整的 insert() 函数代码

func insert()  {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    stmt, err := db.Prepare(`INSERT announcement (imgUrl, detailUrl, createDate, state) VALUES (?, ?, ?, ?)`)
    check(err)

    res, err := stmt.Exec("/visitshop/img/ann/cofox1.png",nil,"2017-09-06",0)
    check(err)

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

    fmt.Println(id)
    stmt.Close()

}

执行后会打印出当前插入记录的 id
修改函数和插入函数结构类似， sql 语句不同

    stmt, err := db.Prepare("UPDATE announcement set imgUrl=?, detailUrl=?, createDate=?, state=? WHERE id=?")
    check(err)

那么参数语句也要增加个 id 值（注意 id 参数是你要修改的那条记录的 id）

    res, err := stmt.Exec("/visitshop/img/ann/cofox2.png", nil, "2017-09-05", 1, 7)
    check(err)

修改的结果返回语句就也调用了不同的函数 res.RowsAffected

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

完整的修改函数代码

func update() {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    stmt, err := db.Prepare("UPDATE announcement set imgUrl=?, detailUrl=?, createDate=?, state=? WHERE id=?")
    check(err)

    res, err := stmt.Exec("/visitshop/img/ann/cofox2.png", nil, "2017-09-05", 1, 7)
    check(err)

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

    fmt.Println(num)
    stmt.Close()
}

执行后，打印修改了的记录条数。
而删除函数的代码与修改函数的代码比较起来就只有 sql 语句和参数的差别了。其他的完全一样。
完整的删除函数的代码

func remove() {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    stmt, err := db.Prepare("DELETE FROM announcement WHERE id=?")
    check(err)

    res, err := stmt.Exec(7)
    check(err)

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

    fmt.Println(num)
    stmt.Close()

}

执行删除函数，最后打印出删除的记录条数。
为方便查看，给出完整代码。如果你想测试运行，每次需要运行哪个功能，请相应的在 main 函数中注释掉其他不执行的函数。
完整代码示例

package main

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

func main() {
    query()
    //query2()
    //insert()
    //update()
    //remove()
}



//查询数据
func query() {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    rows, err := db.Query("SELECT * FROM shopvisit.announcement")
    check(err)

    for rows.Next() {
        columns, _ := rows.Columns()

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

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

        //将数据保存到 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)
    }
    rows.Close()

}
func query2()  {
    fmt.Println("Query2")
    db, err := sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/shopvisit?charset=utf8")
    check(err)

    rows, err := db.Query("SELECT id,imgUrl,createDate,state FROM announcement")
    check(err)

    for rows.Next(){
        var id int
        var state int
        var imgUrl string
        var createDate string
        //注意这里的Scan括号中的参数顺序，和 SELECT 的字段顺序要保持一致。
        if err := rows.Scan(&id,&imgUrl,&createDate,&state); err != nil {
            log.Fatal(err)
        }
        fmt.Printf("%s id is %d on %s with state %d\n", imgUrl, id, createDate, state)
    }

    if err := rows.Err(); err != nil {
        log.Fatal(err)
    }
    rows.Close()
}


//插入数据
func insert()  {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    stmt, err := db.Prepare(`INSERT announcement (imgUrl, detailUrl, createDate, state) VALUES (?, ?, ?, ?)`)
    check(err)

    res, err := stmt.Exec("/visitshop/img/ann/cofox1.png",nil,"2017-09-06",0)
    check(err)

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

    fmt.Println(id)
    stmt.Close()

}

//修改数据
func update() {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    stmt, err := db.Prepare("UPDATE announcement set imgUrl=?, detailUrl=?, createDate=?, state=? WHERE id=?")
    check(err)

    res, err := stmt.Exec("/visitshop/img/ann/cofox2.png", nil, "2017-09-05", 1, 7)
    check(err)

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

    fmt.Println(num)
    stmt.Close()
}

//删除数据
func remove() {
    db, err := sql.Open("mysql", "root:@/shopvisit")
    check(err)

    stmt, err := db.Prepare("DELETE FROM announcement WHERE id=?")
    check(err)

    res, err := stmt.Exec(7)
    check(err)

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

    fmt.Println(num)
    stmt.Close()

}

func check(err error) {
    if err != nil{
        fmt.Println(err)
        panic(err)
    }
}

最后编辑于：2017.12.27 23:25:12

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

47. 访问MySql数据库实现增删改查

推荐阅读更多精彩内容