go 使用 MySQL 数据库

一 MySQL 驱动

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

• https://github.com/go-sql-driver/mysql 支持database/sql，全部采用 go 写。

• https://github.com/ziutek/mymysql 支持database/sql，也支持自定义的接口，全部采用 go 写。

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