博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Go连接MYSQL
阅读量:2213 次
发布时间:2019-05-06

本文共 6266 字,大约阅读时间需要 20 分钟。

Go原生提供了连接数据库操作的支持,在用 Golang进行开发的时候,如果需要在和数据库交互,则可以使用database/sql包。这是一个对关系型数据库的通用抽象,它提供了标准的、轻量的、面向行的接口。

在Go中访问数据库需要用到sql.DB接口:它可以创建语句(statement)和事务(transaction),执行查询,获取结果。

使用数据库时,除了database/sql包本身,还需要引入想使用的特定数据库驱动。官方不提供实现,先下载第三方的实现,查看各种各样的实现版本。

本文测试数据库为mysql,使用的驱动为:github.com/go-sql-driver/mysql,需要引入的包为:

"database/sql"_ "github.com/go-sql-driver/mysql"

解释一下导入包名前面的"_"作用:

import 下划线(如:import _ github/demo)的作用:当导入一个包时,该包下的文件里所有init()函数都会被执行,然而,有些时候我们并不需要把整个包都导入进来,仅仅是是希望它执行init()函数而已。这个时候就可以使用 import _ 引用该包。

上面的mysql驱动中引入的就是mysql包中各个init()方法,你无法通过包名来调用包中的其他函数。导入时,驱动的初始化函数会调用sql.Register将自己注册在database/sql包的全局变量sql.drivers中,以便以后通过sql.Open访问。

执行数据库操作之前我们准备一张表:

CREATE TABLE `user` (    `id` bigint(20) NOT NULL AUTO_INCREMENT,    `name` varchar(45) DEFAULT '',    `age` int(11) NOT NULL DEFAULT '0',    `sex` tinyint(3) NOT NULL DEFAULT '0',    `phone` varchar(45) NOT NULL DEFAULT '',    PRIMARY KEY (`id`)) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8;

1. 初始化数据库连接:

DB, _ := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test")//设置数据库最大连接数DB.SetConnMaxLifetime(100)//设置上数据库最大闲置连接数DB.SetMaxIdleConns(10)//验证连接if err := DB.Ping(); err != nil {
fmt.Println("open database fail") return}fmt.Println("connnect success")

sql.Open()中的数据库连接串格式为:"用户名:密码@tcp(IP:端口)/数据库?charset=utf8"

DB的类型为:*sql.DB,有了DB之后我们就可以执行CRUD操作。Go将数据库操作分为两类:QueryExec。两者的区别在于前者会返回结果,而后者不会。

  • Query表示查询,它会从数据库获取查询结果(一系列行,可能为空)。
  • Exec表示执行语句,它不会返回行。

此外还有两种常见的数据库操作模式:

  • QueryRow表示只返回一行的查询,作为Query的一个常见特例。
  • Prepare表示准备一个需要多次使用的语句,供后续执行用。

2. 查询操作

var user Userrows, e := DB.Query("select * from user where id in (1,2,3)")if e == nil {
errors.New("query incur error")}for rows.Next(){
e := rows.Scan(user.sex, user.phone, user.name, user.id, user.age) if e != nil{
fmt.Println(json.Marshal(user)) }}rows.Close()//单行查询操作DB.QueryRow("select * from user where id=1").Scan(user.age, user.id, user.name, user.phone, user.sex)

整体工作流程如下:

  1. 使用db.Query()来发送查询到数据库,获取结果集Rows,并检查错误。
  2. 使用rows.Next()作为循环条件,迭代读取结果集。
  3. 使用rows.Scan从结果集中获取一行结果。
  4. 使用rows.Err()在退出迭代后检查错误。
  5. 使用rows.Close()关闭结果集,释放连接。

3. 增删改和Exec

通常不会约束你查询必须用Query,只是Query会返回结果集,而Exec不会返回。所以如果你执行的是增删改操作一般用Exec会好一些。Exec返回的结果是ResultResult接口允许获取执行结果的元数据:

type Result interface {
// 用于返回自增ID,并不是所有的关系型数据库都有这个功能。 LastInsertId() (int64, error) // 返回受影响的行数。 RowsAffected() (int64, error)}

4. 准备查询

如果你现在想使用占位符的功能,where 的条件想以参数的形式传入,Go提供了db.Prepare语句来帮你绑定。准备查询的结果是一个准备好的语句(prepared statement),语句中可以包含执行时所需参数的占位符(即绑定值)。准备查询比拼字符串的方式好很多,它可以转义参数,避免SQL注入。同时,准备查询对于一些数据库也省去了解析和生成执行计划的开销,有利于性能。

占位符

PostgreSQL使用$N作为占位符,N是一个从1开始递增的整数,代表参数的位置,方便参数的重复使用。MySQL使用?作为占位符,SQLite两种占位符都可以,而Oracle则使用:param1的形式。

MySQL               PostgreSQL            Oracle=====               ==========            ======WHERE col = ?       WHERE col = $1        WHERE col = :colVALUES(?, ?, ?)     VALUES($1, $2, $3)    VALUES(:val1, :val2, :val3)
stmt, e := DB.Prepare("select * from user where id=?")query, e := stmt.Query(1)query.Scan()

5. 事务的使用

通过db.Begin()来开启一个事务,Begin方法会返回一个事务对象Tx。在结果变量Tx上调用Commit()或者Rollback()方法会提交或回滚变更,并关闭事务。在底层,Tx会从连接池中获得一个连接并在事务过程中保持对它的独占。事务对象Tx上的方法与数据库对象sql.DB的方法一一对应,例如Query,Exec等。事务对象也可以准备(prepare)查询,由事务创建的准备语句会显式绑定到创建它的事务。

//开启事务tx, err := DB.Begin()if err != nil {
fmt.Println("tx fail")}//准备sql语句stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?")if err != nil {
fmt.Println("Prepare fail") return false}//设置参数以及执行sql语句res, err := stmt.Exec(user.id)if err != nil {
fmt.Println("Exec fail") return false}//提交事务tx.Commit()

我们来一个完整的sql操作:

package mainimport (    "database/sql"    "encoding/json"    "fmt"    _ "github.com/go-sql-driver/mysql"    "github.com/pkg/errors"    "strings")//数据库配置const (    userName = "root"    password = "123456"    ip       = "127.0.0.1"    port     = "3306"    dbName   = "test")//Db数据库连接池var DB *sql.DBtype User struct {
id int64 name string age int8 sex int8 phone string}//注意方法名大写,就是publicfunc InitDB() {
//构建连接:"用户名:密码@tcp(IP:端口)/数据库?charset=utf8" path := strings.Join([]string{
userName, ":", password, "@tcp(", ip, ":", port, ")/", dbName, "?charset=utf8"}, "") //打开数据库,前者是驱动名,所以要导入: _ "github.com/go-sql-driver/mysql" DB, _ = sql.Open("mysql", path) //设置数据库最大连接数 DB.SetConnMaxLifetime(100) //设置上数据库最大闲置连接数 DB.SetMaxIdleConns(10) //验证连接 if err := DB.Ping(); err != nil {
fmt.Println("open database fail") return } fmt.Println("connnect success")}//查询操作func Query() {
var user User rows, e := DB.Query("select * from user where id in (1,2,3)") if e == nil {
errors.New("query incur error") } for rows.Next() {
e := rows.Scan(user.sex, user.phone, user.name, user.id, user.age) if e != nil {
fmt.Println(json.Marshal(user)) } } rows.Close() DB.QueryRow("select * from user where id=1").Scan(user.age, user.id, user.name, user.phone, user.sex) stmt, e := DB.Prepare("select * from user where id=?") query, e := stmt.Query(1) query.Scan()}func DeleteUser(user User) bool {
//开启事务 tx, err := DB.Begin() if err != nil {
fmt.Println("tx fail") } //准备sql语句 stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?") if err != nil {
fmt.Println("Prepare fail") return false } //设置参数以及执行sql语句 res, err := stmt.Exec(user.id) if err != nil {
fmt.Println("Exec fail") return false } //提交事务 tx.Commit() //获得上一个insert的id fmt.Println(res.LastInsertId()) return true}func InsertUser(user User) bool {
//开启事务 tx, err := DB.Begin() if err != nil {
fmt.Println("tx fail") return false } //准备sql语句 stmt, err := tx.Prepare("INSERT INTO user (`name`, `phone`) VALUES (?, ?)") if err != nil {
fmt.Println("Prepare fail") return false } //将参数传递到sql语句中并且执行 res, err := stmt.Exec(user.name, user.phone) if err != nil {
fmt.Println("Exec fail") return false } //将事务提交 tx.Commit() //获得上一个插入自增的id fmt.Println(res.LastInsertId()) return true}func main() {
InitDB() Query() defer DB.Close()}

转载于:https://www.cnblogs.com/rickiyang/p/11074180.html

你可能感兴趣的文章
作为一名软件测试工程师,需要具备哪些能力
查看>>
【Pyton】【小甲鱼】类和对象:一些相关的BIF(内置函数)
查看>>
【Pyton】【小甲鱼】魔法方法
查看>>
单元测试需要具备的技能和4大阶段的学习
查看>>
【Loadrunner】【浙江移动项目手写代码】代码备份
查看>>
Python几种并发实现方案的性能比较
查看>>
[Jmeter]jmeter之脚本录制与回放,优化(windows下的jmeter)
查看>>
Jmeter之正则
查看>>
【JMeter】1.9上考试jmeter测试调试
查看>>
【虫师】【selenium】参数化
查看>>
【Python练习】文件引用用户名密码登录系统
查看>>
学习网站汇总
查看>>
【Python】用Python打开csv和xml文件
查看>>
【Loadrunner】性能测试报告实战
查看>>
【自动化测试】自动化测试需要了解的的一些事情。
查看>>
【selenium】selenium ide的安装过程
查看>>
【手机自动化测试】monkey测试
查看>>
【英语】软件开发常用英语词汇
查看>>
Fiddler 抓包工具总结
查看>>
【雅思】雅思需要购买和准备的学习资料
查看>>