Go标准库中没有数据库驱动。参见http://golang.org/s/sqldrivers获取第三方驱动。
这里选择了Go-MySQL-Driver这个实现。地址是:https://github.com/go-sql-driver/mysql/。
安装:go get github.com/go-sql-driver/mysql
官方database/sql包文档:https://studygolang.com/pkgdoc
有关于MySQL基本操作可以去看我的另一篇简书://www.greatytc.com/p/2e01f4c6b626
使用的数据库定义如下
一、打开数据库
函数原型
func Open(driverName, dataSourceName string) (*DB, error)
Open打开一个dirverName指定的数据库,dataSourceName指定数据源,一般包至少括数据库文件名和(可能的)连接信息。
//第⼀步:打开数据库,格式是 ⽤户名:密码@/数据库名称?编码⽅式
db, err := sql.Open("mysql", "root:123@/mydb?charset=utf8")
if err != nil {
fmt.Println(err)
}
//关闭数据库
defer db.Close()
二、查询数据
函数原型
func (db *DB) Query(query string, args ...interface{}) (*Rows, error)
Query执行一次查询,返回多行结果(即Rows),一般用于执行select命令。参数args表示query中的占位参数。
var id int
var name string
var price float64
//查询数据,指定字段名,返回sql.Rows结果集
rows, err := db.Query("select id,name,price from product")
if err != nil {
fmt.Println(err)
}
for rows.Next() {
rows.Scan(&id, &name, &price)
fmt.Println(id, name, price)
}
defer rows.Close()
• rows.Next() 迭代查询数据.
• rows.Scan() 读取每一行的值
• rows.Close() 关闭查询
三、查询一行数据
函数原型
func (db *DB) QueryRow(query string, args ...interface{}) *Row
QueryRow执行一次查询,并期望返回最多一行结果(即Row)。QueryRow总是返回非nil的值,直到返回值的Scan方法被调用时,才会返回被延迟的错误。
//查询⼀⾏数据可以使用and(与)限定多个条件,或者使用or(或)
rows3 := db.QueryRow("SELECT id,name FROM product WHERE id = ? or price like ?", 1,1.11)
// rows3 := db.QueryRow("select id,name from product where id = 1")
rows3.Scan(&id, &name)
fmt.Println(id, name)
四、插入一行数据
函数原型
func (s *Stmt) Exec(args ...interface{}) (Result, error)
Exec使用提供的参数执行准备好的命令状态,返回Result类型的该状态执行结果的总结。
//插⼊⼀⾏数据
ret, _ := db.Exec("insert into product(id,name,price) values(15,'dog',30)")
//LastInsertId返回一个数据库生成的回应命令的整数。
//返回插入的ID
insID, _ := ret.LastInsertId()
fmt.Println(insID)
• ret.LastInsertId()插入的ID
• ret.RowsAffected()受影响的行数
五、修改数据
//更新数据,修改id>1的行的name
ret2, _ := db.Exec("update product set name= '000' where id > ?", 1)
//获取影响⾏数
aff_nums, _ := ret2.RowsAffected()
fmt.Println(aff_nums)
六、删除数据
//删除数据,删除id=3的行
ret3, _ := db.Exec("delete from product where id = ?",3)
delNums, _ := ret3.RowsAffected()
fmt.Println(delNums)
七、预处理
函数原型
func (db *DB) Prepare(query string) (*Stmt, error)
Prepare创建一个准备好的状态用于之后的查询和命令。返回值可以同时执行多个查询和命令。
//预处理
stmt, _ := db.Prepare("select id,name from product where id=? or name=?")
//查找id为3或者name为000的人
row4, _ := stmt.Query(3, "000")
//注意这⾥需要Next()下,不然下⾯取不到值
for row4.Next() {
row4.Scan(&id, &name)
fmt.Println(id, name)
}
//预处理插入数据
stmt2, _ := db.Prepare("insert into product(name,id,price) values(?,?,null)")
//name为lisi,id为22
rows5, _ := stmt2.Exec("lisi", 22)
fmt.Println(rows5.RowsAffected())
八、事务处理
函数原型
func (db *DB) Begin() (*Tx, error)
Begin开始一个事务。隔离水平由数据库驱动决定。
//开启事务
tx, _ := db.Begin()
//id为1的price+1,id为2的price-1
ret4, _ := tx.Exec("update product set price = price + 1 where id = ?", 1)
ret5, _ := tx.Exec("update product set price = price - 1 where id = ?", 2)
//如果id不存在,受影响行数则为0
//接收影响行数,为0则失败
updNums1, _ := ret4.RowsAffected()
updNums2, _ := ret5.RowsAffected()
if updNums1 > 0 && updNums2 > 0 { //只有两条更新同时成功,那么才提交
tx.Commit() //提交事务
fmt.Println("Success")
} else { //否则回滚
tx.Rollback() //回退事务
fmt.Println("Fail")
}
• tx.Commit() 提交事务
• tx.Rollback() 回退事务
demo
package main
import (
"database/sql"
"fmt"
_ "github.com/go-sql-driver/mysql"
)
type dbWorker struct {
dsn string
db *sql.DB
userInfo usertb
}
type usertb struct {
id int
//NullString代表一个可为NULL的字符串。
name sql.NullString
//NullInt64代表一个可为NULL的Float64值。
price sql.NullFloat64
}
func main() {
var err error
//初始化结构体,保存数据库参数
dbw := dbWorker{
dsn: "root:123@/mydb?charset=utf8",
}
//打开数据库,并保存给结构体内db
dbw.db, err = sql.Open("mysql", dbw.dsn)
//如果打开失败,panic退出
if err != nil {
panic(err)
}
//关闭数据库
defer dbw.db.Close()
//插入数据
dbw.insertData()
//获取数据
dbw.querData()
}
//创建方法,插入数据
func (dbw *dbWorker) insertData() {
//预处理,插入数据
stmt, _ := dbw.db.Prepare(`INSERT INTO product(name,id,price) VALUES(?,null,?)`)
//函数执行完毕,关闭
defer stmt.Close()
//插入数据
ret, err := stmt.Exec("hz", 29)
if err != nil {
fmt.Println(err)
return
}
//获取id,执行成功,打印
if LastInsertId, err := ret.LastInsertId(); err == nil {
fmt.Println("LastInsertId:", LastInsertId)
}
//获取行号,执行成功,打印
if RowsAffected, err := ret.RowsAffected(); err == nil {
fmt.Println("RowsAffected:", RowsAffected)
}
}
//初始化userInfo
func (dbw *dbWorker) querDataPre() {
dbw.userInfo = usertb{}
}
func (dbw *dbWorker) querData() {
//预处理,查询数据
stmt, _ := dbw.db.Prepare(`SELECT * From product where price >= ? AND price < ?`)
defer stmt.Close()
dbw.querDataPre()
//取price20到30之间的数据
rows, err := stmt.Query(20, 30)
defer rows.Close()
if err != nil {
fmt.Println(err)
return
}
for rows.Next(){
//取出数据库数据
rows.Scan(&dbw.userInfo.id,&dbw.userInfo.name,&dbw.userInfo.price)
if err != nil {
fmt.Println(err.Error())
continue
}
//如果取出的数据为null,则赋一个0值
if !dbw.userInfo.name.Valid{
dbw.userInfo.name.String = ""
}
if !dbw.userInfo.price.Valid{
dbw.userInfo.price.Float64 = 0.00
}
fmt.Println("get data,id:",dbw.userInfo.id,"name:",dbw.userInfo.name.String,"price",float64(dbw.userInfo.price.Float64))
}
err = rows.Err()
if err != nil{
fmt.Printf(err.Error())
}
}