简介
Go官方并不提供数据库驱动,而只是为开发数据库驱动定义了一些标准接口。我们操作MySQL数据库得需要使用第三方维护的MySQL驱动,推荐使用github.com/go-sql-driver/mysql
,主要理由如下: 维护的比较好 完全支持database/sql接口 * 支持keepalive,保持长连接
有了驱动,其实Go使用MySQL和其他语言差不多,只是具体的方法调用和API组织结构不太一样。
准备工作
use golang;
CREATE TABLE `user` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`username` varchar(100) NOT NULL DEFAULT '' COMMENT '用户名',
`email` varchar(50) NOT NULL DEFAULT '' COMMENT '用户邮箱',
`ctime` int(11) NOT NULL COMMENT '创建时间',
`utime` int(11) NOT NULL COMMENT '更新时间',
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;
先通过ssh登录服务器(因为安全考虑,不开放MySQL的root用户的远程登录权限),用root用户本地登录mysql新建一个用于测试的可远程登录的用户:
mysql> GRANT ALL PRIVILEGES ON golang.* to 'golang'@'%' IDENTIFIED BY '123456';
mysql> GRANT ALL PRIVILEGES ON golang.* to 'golang'@'localhost' IDENTIFIED BY '123456';
mysql> FLUSH PRIVILEGES;
接下来就可以通过golang用户登录新建的数据库进行操作了。
操作MySQL
先get数据库驱动,命令行运行go get github.com/go-sql-driver/mysql
会自动将驱动下载到$GOPATH/src中。
package main
import (
"database/sql"
"fmt"
"time"
_ "github.com/go-sql-driver/mysql"
)
func checkError(e error) {
if e != nil {
panic(e)
}
}
func main() {
// 打开MySQL连接
db, err := sql.Open("mysql", "golang:123456@tcp(127.0.0.1:3306)/golang?charset=utf8")
checkError(err)
fmt.Println("connected")
// ?为占位符,防止SQL注入
stmt, err := db.Prepare("insert into user(username, gender, email, ctime, utime) values(?, ?, ?, ?, ?)")
checkError(err)
timestamp := time.Now().Unix()
// 向占位符传参,Exec是一个不定参数函数,传入的参数与Prepare阶段设置的占位符相等
res, err := stmt.Exec("zhangsan", 1, "zhangsan@qq.com", timestamp, timestamp)
checkError(err)
// 获取生成的数据id
id, err := res.LastInsertId()
checkError(err)
fmt.Println(id)
stmt, err = db.Prepare("update user set email=? where id=?")
checkError(err)
res, err = stmt.Exec("zhangsan@163.com", id)
checkError(err)
aff, err := res.RowsAffected()
checkError(err)
fmt.Println("affected", aff)
rows, err := db.Query("select id, username, email from user")
for rows.Next() {
var id int
var username string
var email string
err = rows.Scan(&id, &username, &email)
checkError(err)
fmt.Println(id, username, email)
}
stmt, _ = db.Prepare("delete from user where id>?")
res, _ = stmt.Exec(1)
aff, _ = res.RowsAffected()
fmt.Println("affected", aff)
}
整个数据库操作的流程大概如下: sql.Open()函数用来打开一个注册过的数据库驱动; db.Prepare()函数用来返回准备要执行的sql操作,然后返回准备完毕的执行状态; stmt.Exec()函数用来执行stmt准备好的SQL语句; db.Query()函数用来直接执行Sql返回Rows结果。
数据库驱动的注册在数据库驱动包导入时执行,注册的逻辑写在包的init函数中。导入驱动包只需要import _ "driver/package"
,只需要执行其init(),而不需要调用包中导出的方法,调用的是Go语言规定的sql接口。