47. 访问MySql数据库实现增删改查 | 厚土Go学习笔记

作为服务端程序,对数据库的访问是很常见的操作。我们来熟悉一下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)
    }
}

原文发布于微信公众号 - Golang语言社区(Golangweb)

原文发表时间:2018-02-16

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏Java帮帮-微信公众号-技术文章全总结

Web-第六天 MySQL回顾学习

数据库就是存储数据的仓库,其本质是一个文件系统,数据按照特定的格式将数据存储起来,用户可以通过sql语句对数据库中的数据进行增加,修改,删除及查询操作

1182
来自专栏琯琯博客

Yii2 学习笔记之数据库篇

4517
来自专栏Kevin-ZhangCG

数据库索引

索引就是加快检索表中数据的方法。数据库的索引类似于书籍的索引。在书籍中,索引允许用户不必翻阅完整个书就能迅速地找到所需要的信息。在数据库中,索引也允许数据库...

1180
来自专栏云霄雨霁

数据库完整性总结

1914
来自专栏乐沙弥的世界

使用 Toad 实现 SQL 优化

       It is very easy for us to implement sql tuning by toad.  We need to do is...

932
来自专栏北京马哥教育

MongoDB权威指南学习笔记5---索引相关的知识点

1 查看查询计划 db.user.find({"username":"xxx"}) .explain() db.doc.find({"es_y":"2014"}...

2705
来自专栏ChaMd5安全团队

SQLI-LABS 更新帖(一)

docker搭建环境 请自己安装好docker,然后使用ubuntu:14.04的镜像 docker pull ubuntu:14.04 以下是pcat提供的...

3778
来自专栏逸鹏说道

SQL Server 深入解析索引存储(中)

概述 本篇文章是关于堆的存储结构。堆是不含聚集索引的表(所以只有非聚集索引的表也是堆)。堆的 sys.partitions 中具有一行,对于堆使用的每...

3648
来自专栏各种机器学习基础算法

常用但容易忘记的sql语句(sql server为主)

1.说明:随机取出10条数据 a.Sql Server: select top 10 * from tablename order by newid() b....

3019
来自专栏杨建荣的学习笔记

关于db_files和maxdatafiles的问题(r4笔记第31天)

昨天在做生产监控的时候发现有个库的表空间不够了,就发邮件给客户的dba去处理,但是得到的反馈是尝试添加的时候发现已经超过了数据文件的最大数限制。这个错误毫无疑问...

3596

扫码关注云+社区

领取腾讯云代金券