5.2 使用MySQL数据库

目前Internet上流行的网站构架方式是LAMP,其中的M即MySQL, 作为数据库,MySQL以免费、开源、使用方便为优势成为了很多Web开发的后端数据库存储引擎。

MySQL驱动

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

接下来的例子我主要以第一个驱动为例(我目前项目中也是采用它来驱动),也推荐大家采用它,主要理由:

  • 这个驱动比较新,维护的比较好
  • 完全支持database/sql接口
  • 支持keepalive,保持长连接,虽然星星fork的mymysql也支持keepalive,但不是线程安全的,这个从底层就支持了keepalive。

示例代码

接下来的几个小节里面我们都将采用同一个数据库表结构:数据库test,用户表userinfo,关联用户信息表userdetail。

  1. CREATE TABLE `userinfo` (
  2. `uid` INT(10) NOT NULL AUTO_INCREMENT,
  3. `username` VARCHAR(64) NULL DEFAULT NULL,
  4. `departname` VARCHAR(64) NULL DEFAULT NULL,
  5. `created` DATE NULL DEFAULT NULL,
  6. PRIMARY KEY (`uid`)
  7. );
  8. CREATE TABLE `userdetail` (
  9. `uid` INT(10) NOT NULL DEFAULT '0',
  10. `intro` TEXT NULL,
  11. `profile` TEXT NULL,
  12. PRIMARY KEY (`uid`)
  13. )

如下示例将示范如何使用database/sql接口对数据库表进行增删改查操作

  1. package main
  2. import (
  3. "database/sql"
  4. "fmt"
  5. //"time"
  6. _ "github.com/go-sql-driver/mysql"
  7. )
  8. func main() {
  9. db, err := sql.Open("mysql", "astaxie:astaxie@/test?charset=utf8")
  10. checkErr(err)
  11. //插入数据
  12. stmt, err := db.Prepare("INSERT userinfo SET username=?,departname=?,created=?")
  13. checkErr(err)
  14. res, err := stmt.Exec("astaxie", "研发部门", "2012-12-09")
  15. checkErr(err)
  16. id, err := res.LastInsertId()
  17. checkErr(err)
  18. fmt.Println(id)
  19. //更新数据
  20. stmt, err = db.Prepare("update userinfo set username=? where uid=?")
  21. checkErr(err)
  22. res, err = stmt.Exec("astaxieupdate", id)
  23. checkErr(err)
  24. affect, err := res.RowsAffected()
  25. checkErr(err)
  26. fmt.Println(affect)
  27. //查询数据
  28. rows, err := db.Query("SELECT * FROM userinfo")
  29. checkErr(err)
  30. for rows.Next() {
  31. var uid int
  32. var username string
  33. var department string
  34. var created string
  35. err = rows.Scan(&uid, &username, &department, &created)
  36. checkErr(err)
  37. fmt.Println(uid)
  38. fmt.Println(username)
  39. fmt.Println(department)
  40. fmt.Println(created)
  41. }
  42. //删除数据
  43. stmt, err = db.Prepare("delete from userinfo where uid=?")
  44. checkErr(err)
  45. res, err = stmt.Exec(id)
  46. checkErr(err)
  47. affect, err = res.RowsAffected()
  48. checkErr(err)
  49. fmt.Println(affect)
  50. db.Close()
  51. }
  52. func checkErr(err error) {
  53. if err != nil {
  54. panic(err)
  55. }
  56. }

通过上面的代码我们可以看出,Go操作Mysql数据库是很方便的。

关键的几个函数我解释一下:

sql.Open()函数用来打开一个注册过的数据库驱动,go-sql-driver中注册了mysql这个数据库驱动,第二个参数是DSN(Data Source Name),它是go-sql-driver定义的一些数据库链接和配置信息。它支持如下格式:

  1. user@unix(/path/to/socket)/dbname?charset=utf8
  2. user:password@tcp(localhost:5555)/dbname?charset=utf8
  3. user:password@/dbname
  4. user:password@tcp([de:ad:be:ef::ca:fe]:80)/dbname

db.Prepare()函数用来返回准备要执行的sql操作,然后返回准备完毕的执行状态。

db.Query()函数用来直接执行Sql返回Rows结果。

stmt.Exec()函数用来执行stmt准备好的SQL语句

我们可以看到我们传入的参数都是=?对应的数据,这样做的方式可以一定程度上防止SQL注入。