Golang如何優(yōu)雅連接MySQL數(shù)據(jù)庫(kù)?
Go原生就支持連接數(shù)據(jù)庫(kù)，所以在使用 Golang 開(kāi)發(fā)時(shí)，當(dāng)需要數(shù)據(jù)庫(kù)交互時(shí)，即可使用database/sql包。
在Go中訪問(wèn)DB需用sql.DB接口：可創(chuàng)建語(yǔ)句(statement)和事務(wù)(transaction)，執(zhí)行查詢，獲取結(jié)果。
使用DB時(shí)，除database/sql包，還需引入想使用的特定DB驅(qū)動(dòng)。官方不提供實(shí)現(xiàn)，需要先下載三方實(shí)現(xiàn)，點(diǎn)擊這里查看各種各樣的實(shí)現(xiàn)版本。
通常DB選型MySQL，所以選型驅(qū)動(dòng)為:github.com/go-sql-driver/mysql,需引入包：
"database/sql"_ "github.com/go-sql-driver/mysql"
包名前的"_"
import 下劃線(如：import _ github/demo)的作用：當(dāng)導(dǎo)入一個(gè)包時(shí)，該包下的文件里所有init()函數(shù)都會(huì)被執(zhí)行，然而，有些時(shí)候我們并不需要把整個(gè)包都導(dǎo)入進(jìn)來(lái)，僅僅是是希望它執(zhí)行init()函數(shù)而已。這個(gè)時(shí)候就可以使用 import _ 引用該包。
上面的MySQL驅(qū)動(dòng)中引入的就是MySQL包中各個(gè)init()方法，你無(wú)法通過(guò)包名來(lái)調(diào)用包中的其他函數(shù)。導(dǎo)入時(shí)，驅(qū)動(dòng)的初始化函數(shù)會(huì)調(diào)用sql.Register將自己注冊(cè)在database/sql包的全局變量sql.drivers中，以便以后通過(guò)sql.Open訪問(wèn)。
案例用數(shù)據(jù)表
初始化數(shù)據(jù)庫(kù)連接
sql.Open()中的數(shù)據(jù)庫(kù)連接串格式為："用戶名:密碼@tcp(IP:端口)/數(shù)據(jù)庫(kù)?charset=utf8"。DB的類型為:*sql.DB，有DB后即可執(zhí)行CRUD。
Go將數(shù)據(jù)庫(kù)操作分為兩類：Query與Exec
- Query表示查詢，它會(huì)從數(shù)據(jù)庫(kù)獲取查詢結(jié)果(一系列行，可能為空)。
- Exec表示執(zhí)行語(yǔ)句，它不會(huì)返回行。
常見(jiàn)數(shù)據(jù)庫(kù)操作模式：
- QueryRow只返回一行的查詢，作為Query的一個(gè)常見(jiàn)特例。
- Prepare準(zhǔn)備一個(gè)需要多次使用的語(yǔ)句，供后續(xù)執(zhí)行用。
查詢操作
- 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)
執(zhí)行流程
- 使用db.Query()來(lái)發(fā)送查詢到數(shù)據(jù)庫(kù)，獲取結(jié)果集Rows，并檢查錯(cuò)誤。
- 使用rows.Next()作為循環(huán)條件，迭代讀取結(jié)果集。
- 使用rows.Scan從結(jié)果集中獲取一行結(jié)果。
- 使用rows.Err()在退出迭代后檢查錯(cuò)誤。
- 使用rows.Close()關(guān)閉結(jié)果集，釋放連接。
增刪改和Exec
通常不會(huì)約束你查詢必須用Query，只是Query會(huì)返回結(jié)果集，而Exec不會(huì)返回。所以如果你執(zhí)行的是增刪改操作一般用Exec會(huì)好一些。Exec返回的結(jié)果是Result，Result接口允許獲取執(zhí)行結(jié)果的元數(shù)據(jù):
- type Result interface {
- // 用于返回自增ID，并不是所有的關(guān)系型數(shù)據(jù)庫(kù)都有這個(gè)功能。
- LastInsertId() (int64, error)
- // 返回受影響的行數(shù)。
- RowsAffected() (int64, error)
- }
準(zhǔn)備查詢
如果你現(xiàn)在想使用占位符的功能，where 的條件想以參數(shù)的形式傳入，Go提供了db.Prepare語(yǔ)句來(lái)幫你綁定。準(zhǔn)備查詢的結(jié)果是一個(gè)準(zhǔn)備好的語(yǔ)句(prepared statement)，語(yǔ)句中可以包含執(zhí)行時(shí)所需參數(shù)的占位符(即綁定值)。準(zhǔn)備查詢比拼字符串的方式好很多，它可以轉(zhuǎn)義參數(shù)，避免SQL注入。同時(shí)，準(zhǔn)備查詢對(duì)于一些數(shù)據(jù)庫(kù)也省去了解析和生成執(zhí)行計(jì)劃的開(kāi)銷，有利于性能。
占位符
PostgreSQL使用$N作為占位符，N是一個(gè)從1開(kāi)始遞增的整數(shù)，代表參數(shù)的位置，方便參數(shù)的重復(fù)使用。MySQL使用?作為占位符，SQLite兩種占位符都可以，而Oracle則使用:param1的形式。
- MySQL PostgreSQL Oracle
- ===== ========== ======
- WHERE col = ? WHERE col = $1 WHERE col = :col
- VALUES(?, ?, ?) VALUES($1, $2, $3) VALUES(:val1, :val2, :val3)
- stmt, e := DB.Prepare("select * from user where id=?")
- query, e := stmt.Query(1)
- query.Scan()
事務(wù)的使用
通過(guò)db.Begin()來(lái)開(kāi)啟一個(gè)事務(wù)，Begin方法會(huì)返回一個(gè)事務(wù)對(duì)象Tx。在結(jié)果變量Tx上調(diào)用Commit()或者Rollback()方法會(huì)提交或回滾變更，并關(guān)閉事務(wù)。在底層，Tx會(huì)從連接池中獲得一個(gè)連接并在事務(wù)過(guò)程中保持對(duì)它的獨(dú)占。事務(wù)對(duì)象Tx上的方法與數(shù)據(jù)庫(kù)對(duì)象sql.DB的方法一一對(duì)應(yīng)，例如Query,Exec等。事務(wù)對(duì)象也可以準(zhǔn)備(prepare)查詢，由事務(wù)創(chuàng)建的準(zhǔn)備語(yǔ)句會(huì)顯式綁定到創(chuàng)建它的事務(wù)。
- //開(kāi)啟事務(wù)
- tx, err := DB.Begin()
- if err != nil {
- fmt.Println("tx fail")
- }
- //準(zhǔn)備sql語(yǔ)句
- stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?")
- if err != nil {
- fmt.Println("Prepare fail")
- return false
- }
- //設(shè)置參數(shù)以及執(zhí)行sql語(yǔ)句
- res, err := stmt.Exec(user.id)
- if err != nil {
- fmt.Println("Exec fail")
- return false
- }
- //提交事務(wù)
- tx.Commit()
- 我們來(lái)一個(gè)完整的sql操作：package main
- import (
- "database/sql"
- "encoding/json"
- "fmt"
- _ "github.com/go-sql-driver/mysql"
- "github.com/pkg/errors"
- "strings"
- )
- //數(shù)據(jù)庫(kù)配置
- const (
- userName = "root"
- password = "123456"
- ip = "127.0.0.1"
- port = "3306"
- dbName = "test"
- )
- //Db數(shù)據(jù)庫(kù)連接池
- var DB *sql.DB
- type User struct {
- id int64
- name string
- age int8
- sex int8
- phone string
- }
- //注意方法名大寫(xiě)，就是public
- func InitDB() {
- //構(gòu)建連接："用戶名:密碼@tcp(IP:端口)/數(shù)據(jù)庫(kù)?charset=utf8"
- path := strings.Join([]string{userName, ":", password, "@tcp(", ip, ":", port, ")/", dbName, "?charset=utf8"}, "")
- //打開(kāi)數(shù)據(jù)庫(kù),前者是驅(qū)動(dòng)名，所以要導(dǎo)入：_ "github.com/go-sql-driver/mysql"
- DB, _ = sql.Open("mysql", path)
- //設(shè)置數(shù)據(jù)庫(kù)最大連接數(shù)
- DB.SetConnMaxLifetime(100)
- //設(shè)置上數(shù)據(jù)庫(kù)最大閑置連接數(shù)
- DB.SetMaxIdleConns(10)
- //驗(yàn)證連接
- 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 {
- //開(kāi)啟事務(wù)
- tx, err := DB.Begin()
- if err != nil {
- fmt.Println("tx fail")
- }
- //準(zhǔn)備sql語(yǔ)句
- stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?")
- if err != nil {
- fmt.Println("Prepare fail")
- return false
- }
- //設(shè)置參數(shù)以及執(zhí)行sql語(yǔ)句
- res, err := stmt.Exec(user.id)
- if err != nil {
- fmt.Println("Exec fail")
- return false
- }
- //提交事務(wù)
- tx.Commit()
- //獲得上一個(gè)insert的id
- fmt.Println(res.LastInsertId())
- return true
- }
- func InsertUser(user User) bool {
- //開(kāi)啟事務(wù)
- tx, err := DB.Begin()
- if err != nil {
- fmt.Println("tx fail")
- return false
- }
- //準(zhǔn)備sql語(yǔ)句
- stmt, err := tx.Prepare("INSERT INTO user (`name`, `phone`) VALUES (?, ?)")
- if err != nil {
- fmt.Println("Prepare fail")
- return false
- }
- //將參數(shù)傳遞到sql語(yǔ)句中并且執(zhí)行
- res, err := stmt.Exec(user.name, user.phone)
- if err != nil {
- fmt.Println("Exec fail")
- return false
- }
- //將事務(wù)提交
- tx.Commit()
- //獲得上一個(gè)插入自增的id
- fmt.Println(res.LastInsertId())
- return true
- }
- func main() {
- InitDB()
- Query()
- defer DB.Close()
- }
參考
https://www.cnblogs.com/rickiyang/p/11074180.html
