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