go 使用 MySQL 数据库

ゞ浴缸里的玫瑰 2021-07-24 13:24 345阅读 0赞

# 一 MySQL 驱动 #

Go 中支持 MySQL 的驱动目前比较多，有如下几种，有些是支持 database/sql 标准，而有些是采用了自己的实现接口,常用的有如下几种。

*  [https://github.com/go-sql-driver/mysql][https_github.com_go-sql-driver_mysql] 支持database/sql，全部采用 go 写。
 *  [https://github.com/ziutek/mymysql][https_github.com_ziutek_mymysql] 支持database/sql，也支持自定义的接口，全部采用 go 写。
 *  [https://github.com/Philio/GoMySQL][https_github.com_Philio_GoMySQL] 不支持database/sql，自定义接口，全部采用 go 写。

本篇使用的是第1个驱动，它的好处如下：

这个驱动比较新，维护的比较好。

完全支持 database/sql 接口。

支持 keepalive，保持长连接。

# 二 实战 #

## 1 数据库脚本 ##

CREATE TABLE `userinfo` (
        `uid` INT(10) NOT NULL AUTO_INCREMENT,
        `username` VARCHAR(64) NULL DEFAULT NULL,
        `department` 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`)
    )

## 2 对数据库进行增删查改操作 ##

package main
    
    import (
       "database/sql"
       "fmt"
       _ "github.com/mysql-go"
    )
    
    func main() {
       db, err := sql.Open("mysql", "root:Mima123456@/test?charset=utf8")
       checkErr(err)
    
       // 插入数据
       stmt, err := db.Prepare("INSERT INTO userinfo SET username=?,department=?,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)
       }
    }

## 3 测试结果 ##

1
    1
    1
    astaxieupdate
    研发部门
    2012-12-09
    1

## 4 说明 ##

关键函数说明：

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注入。

[https_github.com_go-sql-driver_mysql]: https://github.com/go-sql-driver/mysql
[https_github.com_ziutek_mymysql]: https://github.com/ziutek/mymysql
[https_github.com_Philio_GoMySQL]: https://github.com/Philio/GoMySQL