golang連接sqlx庫(kù)的操作使用指南
更新時(shí)間：2022年04月14日 11:58:51 作者：Jeff的技術(shù)棧
這篇文章主要為大家介紹了golang連接sqlx庫(kù)的操作使用指南，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步早日升職加薪
sqlx庫(kù)使用指南
在項(xiàng)目中我們通常可能會(huì)使用database/sql連接MySQL數(shù)據(jù)庫(kù)。本文借助使用sqlx實(shí)現(xiàn)批量插入數(shù)據(jù)的例子，介紹了sqlx中可能被你忽視了的sqlx.In和DB.NamedExec方法。
sqlx介紹
在項(xiàng)目中我們通常可能會(huì)使用database/sql連接MySQL數(shù)據(jù)庫(kù)。sqlx可以認(rèn)為是Go語(yǔ)言內(nèi)置database/sql的超集，它在優(yōu)秀的內(nèi)置database/sql基礎(chǔ)上提供了一組擴(kuò)展。這些擴(kuò)展中除了大家常用來(lái)查詢的Get(dest interface{}, ...) error和Select(dest interface{}, ...) error外還有很多其他強(qiáng)大的功能。
安裝sqlx
go get github.com/jmoiron/sqlx
基本使用
連接數(shù)據(jù)庫(kù)
var db *sqlx.DB
func initDB() (err error) {
dsn := "user:password@tcp(127.0.0.1:3306)/sql_test?charset=utf8mb4&parseTime=True"
// 也可以使用MustConnect連接不成功就panic
db, err = sqlx.Connect("mysql", dsn)
if err != nil {
fmt.Printf("connect DB failed, err:%v\n", err)
return
}
db.SetMaxOpenConns(20)
db.SetMaxIdleConns(10)
return
}
查詢
查詢單行數(shù)據(jù)示例代碼如下：
// 查詢單條數(shù)據(jù)示例
func queryRowDemo() {
sqlStr := "select id, name, age from user where id=?"
var u user
err := db.Get(&u, sqlStr, 1)
if err != nil {
fmt.Printf("get failed, err:%v\n", err)
return
}
fmt.Printf("id:%d name:%s age:%d\n", u.ID, u.Name, u.Age)
}
查詢多行數(shù)據(jù)示例代碼如下：
// 查詢多條數(shù)據(jù)示例
func queryMultiRowDemo() {
sqlStr := "select id, name, age from user where id > ?"
var users []user
err := db.Select(&users, sqlStr, 0)
if err != nil {
fmt.Printf("query failed, err:%v\n", err)
return
}
fmt.Printf("users:%#v\n", users)
}
插入、更新和刪除
sqlx中的exec方法與原生sql中的exec使用基本一致：
// 插入數(shù)據(jù)
func insertRowDemo() {
sqlStr := "insert into user(name, age) values (?,?)"
ret, err := db.Exec(sqlStr, "沙河小王子", 19)
if err != nil {
fmt.Printf("insert failed, err:%v\n", err)
return
}
theID, err := ret.LastInsertId() // 新插入數(shù)據(jù)的id
if err != nil {
fmt.Printf("get lastinsert ID failed, err:%v\n", err)
return
}
fmt.Printf("insert success, the id is %d.\n", theID)
}
// 更新數(shù)據(jù)
func updateRowDemo() {
sqlStr := "update user set age=? where id = ?"
ret, err := db.Exec(sqlStr, 39, 6)
if err != nil {
fmt.Printf("update failed, err:%v\n", err)
return
}
n, err := ret.RowsAffected() // 操作影響的行數(shù)
if err != nil {
fmt.Printf("get RowsAffected failed, err:%v\n", err)
return
}
fmt.Printf("update success, affected rows:%d\n", n)
}
// 刪除數(shù)據(jù)
func deleteRowDemo() {
sqlStr := "delete from user where id = ?"
ret, err := db.Exec(sqlStr, 6)
if err != nil {
fmt.Printf("delete failed, err:%v\n", err)
return
}
n, err := ret.RowsAffected() // 操作影響的行數(shù)
if err != nil {
fmt.Printf("get RowsAffected failed, err:%v\n", err)
return
}
fmt.Printf("delete success, affected rows:%d\n", n)
}
NamedExec
DB.NamedExec方法用來(lái)綁定SQL語(yǔ)句與結(jié)構(gòu)體或map中的同名字段。
func insertUserDemo()(err error){
sqlStr := "INSERT INTO user (name,age) VALUES (:name,:age)"
_, err = db.NamedExec(sqlStr,
map[string]interface{}{
"name": "七米",
"age": 28,
})
return
}
NamedQuery
與DB.NamedExec同理，這里是支持查詢。
func namedQuery(){
sqlStr := "SELECT * FROM user WHERE name=:name"
// 使用map做命名查詢
rows, err := db.NamedQuery(sqlStr, map[string]interface{}{"name": "七米"})
if err != nil {
fmt.Printf("db.NamedQuery failed, err:%v\n", err)
return
}
defer rows.Close()
for rows.Next(){
var u user
err := rows.StructScan(&u)
if err != nil {
fmt.Printf("scan failed, err:%v\n", err)
continue
}
fmt.Printf("user:%#v\n", u)
}
u := user{
Name: "七米",
}
// 使用結(jié)構(gòu)體命名查詢，根據(jù)結(jié)構(gòu)體字段的 db tag進(jìn)行映射
rows, err = db.NamedQuery(sqlStr, u)
if err != nil {
fmt.Printf("db.NamedQuery failed, err:%v\n", err)
return
}
defer rows.Close()
for rows.Next(){
var u user
err := rows.StructScan(&u)
if err != nil {
fmt.Printf("scan failed, err:%v\n", err)
continue
}
fmt.Printf("user:%#v\n", u)
}
}
事務(wù)操作
對(duì)于事務(wù)操作，我們可以使用sqlx中提供的db.Beginx()和tx.Exec()方法。示例代碼如下：
func transactionDemo2()(err error) {
tx, err := db.Beginx() // 開(kāi)啟事務(wù)
if err != nil {
fmt.Printf("begin trans failed, err:%v\n", err)
return err
}
defer func() {
if p := recover(); p != nil {
tx.Rollback()
panic(p) // re-throw panic after Rollback
} else if err != nil {
fmt.Println("rollback")
tx.Rollback() // err is non-nil; don't change it
} else {
err = tx.Commit() // err is nil; if Commit returns error update err
fmt.Println("commit")
}
}()
sqlStr1 := "Update user set age=20 where id=?"
rs, err := tx.Exec(sqlStr1, 1)
if err!= nil{
return err
}
n, err := rs.RowsAffected()
if err != nil {
return err
}
if n != 1 {
return errors.New("exec sqlStr1 failed")
}
sqlStr2 := "Update user set age=50 where i=?"
rs, err = tx.Exec(sqlStr2, 5)
if err!=nil{
return err
}
n, err = rs.RowsAffected()
if err != nil {
return err
}
if n != 1 {
return errors.New("exec sqlStr1 failed")
}
return err
}
sqlx.In
sqlx.In是sqlx提供的一個(gè)非常方便的函數(shù)。
sqlx.In的批量插入示例
表結(jié)構(gòu)
為了方便演示插入數(shù)據(jù)操作，這里創(chuàng)建一個(gè)user表，表結(jié)構(gòu)如下：
CREATE TABLE `user` (
`id` BIGINT(20) NOT NULL AUTO_INCREMENT,
`name` VARCHAR(20) DEFAULT '',
`age` INT(11) DEFAULT '0',
PRIMARY KEY(`id`)
)ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4;
結(jié)構(gòu)體
定義一個(gè)user結(jié)構(gòu)體，字段通過(guò)tag與數(shù)據(jù)庫(kù)中user表的列一致。
type User struct {
Name string `db:"name"`
Age int `db:"age"`
}
bindvars（綁定變量）
查詢占位符?在內(nèi)部稱為bindvars（查詢占位符）,它非常重要。你應(yīng)該始終使用它們向數(shù)據(jù)庫(kù)發(fā)送值，因?yàn)樗鼈兛梢苑乐筍QL注入攻擊。database/sql不嘗試對(duì)查詢文本進(jìn)行任何驗(yàn)證；它與編碼的參數(shù)一起按原樣發(fā)送到服務(wù)器。除非驅(qū)動(dòng)程序?qū)崿F(xiàn)一個(gè)特殊的接口，否則在執(zhí)行之前，查詢是在服務(wù)器上準(zhǔn)備的。因此bindvars是特定于數(shù)據(jù)庫(kù)的:
- MySQL中使用
? - PostgreSQL使用枚舉的
$1、$2等bindvar語(yǔ)法 - SQLite中
?和$1的語(yǔ)法都支持 - Oracle中使用
:name的語(yǔ)法
bindvars的一個(gè)常見(jiàn)誤解是，它們用來(lái)在sql語(yǔ)句中插入值。它們其實(shí)僅用于參數(shù)化，不允許更改SQL語(yǔ)句的結(jié)構(gòu)。例如，使用bindvars嘗試參數(shù)化列或表名將不起作用：
// ？不能用來(lái)插入表名（做SQL語(yǔ)句中表名的占位符）
db.Query("SELECT * FROM ?", "mytable")
// ？也不能用來(lái)插入列名（做SQL語(yǔ)句中列名的占位符）
db.Query("SELECT ?, ? FROM people", "name", "location")
自己拼接語(yǔ)句實(shí)現(xiàn)批量插入
比較笨，但是很好理解。就是有多少個(gè)User就拼接多少個(gè)(?, ?)。
// BatchInsertUsers 自行構(gòu)造批量插入的語(yǔ)句
func BatchInsertUsers(users []*User) error {
// 存放 (?, ?) 的slice
valueStrings := make([]string, 0, len(users))
// 存放values的slice
valueArgs := make([]interface{}, 0, len(users) * 2)
// 遍歷users準(zhǔn)備相關(guān)數(shù)據(jù)
for _, u := range users {
// 此處占位符要與插入值的個(gè)數(shù)對(duì)應(yīng)
valueStrings = append(valueStrings, "(?, ?)")
valueArgs = append(valueArgs, u.Name)
valueArgs = append(valueArgs, u.Age)
}
// 自行拼接要執(zhí)行的具體語(yǔ)句
stmt := fmt.Sprintf("INSERT INTO user (name, age) VALUES %s",
strings.Join(valueStrings, ","))
_, err := DB.Exec(stmt, valueArgs...)
return err
}
使用sqlx.In實(shí)現(xiàn)批量插入
前提是需要我們的結(jié)構(gòu)體實(shí)現(xiàn)driver.Valuer接口：
func (u User) Value() (driver.Value, error) {
return []interface{}{u.Name, u.Age}, nil
}
使用sqlx.In實(shí)現(xiàn)批量插入代碼如下：
// BatchInsertUsers2 使用sqlx.In幫我們拼接語(yǔ)句和參數(shù), 注意傳入的參數(shù)是[]interface{}
func BatchInsertUsers2(users []interface{}) error {
query, args, _ := sqlx.In(
"INSERT INTO user (name, age) VALUES (?), (?), (?)",
users..., // 如果arg實(shí)現(xiàn)了 driver.Valuer, sqlx.In 會(huì)通過(guò)調(diào)用 Value()來(lái)展開(kāi)它
)
fmt.Println(query) // 查看生成的querystring
fmt.Println(args) // 查看生成的args
_, err := DB.Exec(query, args...)
return err
}
使用NamedExec實(shí)現(xiàn)批量插入
注意 ：該功能需1.3.1版本以上，并且1.3.1版本目前還有點(diǎn)問(wèn)題，sql語(yǔ)句最后不能有空格和;，詳見(jiàn)issues/690。
使用NamedExec實(shí)現(xiàn)批量插入的代碼如下：
// BatchInsertUsers3 使用NamedExec實(shí)現(xiàn)批量插入
func BatchInsertUsers3(users []*User) error {
_, err := DB.NamedExec("INSERT INTO user (name, age) VALUES (:name, :age)", users)
return err
}
把上面三種方法綜合起來(lái)試一下：
func main() {
err := initDB()
if err != nil {
panic(err)
}
defer DB.Close()
u1 := User{Name: "七米", Age: 18}
u2 := User{Name: "q1mi", Age: 28}
u3 := User{Name: "小王子", Age: 38}
// 方法1
users := []*User{&u1, &u2, &u3}
err = BatchInsertUsers(users)
if err != nil {
fmt.Printf("BatchInsertUsers failed, err:%v\n", err)
}
// 方法2
users2 := []interface{}{u1, u2, u3}
err = BatchInsertUsers2(users2)
if err != nil {
fmt.Printf("BatchInsertUsers2 failed, err:%v\n", err)
}
// 方法3
users3 := []*User{&u1, &u2, &u3}
err = BatchInsertUsers3(users3)
if err != nil {
fmt.Printf("BatchInsertUsers3 failed, err:%v\n", err)
}
}
sqlx.In的查詢示例
關(guān)于sqlx.In這里再補(bǔ)充一個(gè)用法，在sqlx查詢語(yǔ)句中實(shí)現(xiàn)In查詢和FIND_IN_SET函數(shù)。
即實(shí)現(xiàn)SELECT * FROM user WHERE id in (3, 2, 1);
和SELECT * FROM user WHERE id in (3, 2, 1) ORDER BY FIND_IN_SET(id, '3,2,1');。
in查詢
查詢id在給定id集合中的數(shù)據(jù)。
// QueryByIDs 根據(jù)給定ID查詢
func QueryByIDs(ids []int)(users []User, err error){
// 動(dòng)態(tài)填充id
query, args, err := sqlx.In("SELECT name, age FROM user WHERE id IN (?)", ids)
if err != nil {
return
}
// sqlx.In 返回帶 `?` bindvar的查詢語(yǔ)句, 我們使用Rebind()重新綁定它
query = DB.Rebind(query)
err = DB.Select(&users, query, args...)
return
}
in查詢和FIND_IN_SET函數(shù)
查詢id在給定id集合的數(shù)據(jù)并維持給定id集合的順序。
// QueryAndOrderByIDs 按照指定id查詢并維護(hù)順序
func QueryAndOrderByIDs(ids []int)(users []User, err error){
// 動(dòng)態(tài)填充id
strIDs := make([]string, 0, len(ids))
for _, id := range ids {
strIDs = append(strIDs, fmt.Sprintf("%d", id))
}
query, args, err := sqlx.In("SELECT name, age FROM user WHERE id IN (?) ORDER BY FIND_IN_SET(id, ?)", ids, strings.Join(strIDs, ","))
if err != nil {
return
}
// sqlx.In 返回帶 `?` bindvar的查詢語(yǔ)句, 我們使用Rebind()重新綁定它
query = DB.Rebind(query)
err = DB.Select(&users, query, args...)
return
}
當(dāng)然，在這個(gè)例子里面你也可以先使用IN查詢，然后通過(guò)代碼按給定的ids對(duì)查詢結(jié)果進(jìn)行排序。
參考鏈接：
以上就是golang連接sqlx庫(kù)的操作使用指南的詳細(xì)內(nèi)容，更多關(guān)于golang操作連接sqlx庫(kù)的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
Go語(yǔ)言使用buffer讀取文件的實(shí)現(xiàn)示例
本文主要介紹了Go語(yǔ)言使用buffer讀取文件的實(shí)現(xiàn)示例，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-04-04
Go語(yǔ)言并發(fā)編程基礎(chǔ)上下文概念詳解
這篇文章主要為大家介紹了Go語(yǔ)言并發(fā)編程基礎(chǔ)上下文示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-08-08
這篇文章主要為大家介紹了Go java 算法之括號(hào)生成示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-08-08
Golang實(shí)現(xiàn)定時(shí)任務(wù)的幾種方法小結(jié)
在 Golang 開(kāi)發(fā)中,定時(shí)任務(wù)是常見(jiàn)的需求,本文將介紹幾種在 Golang 中實(shí)現(xiàn)定時(shí)任務(wù)的方法,包括 time 包的定時(shí)器、ticker,以及第三方庫(kù) cron,并通過(guò)示例代碼展示它們的使用方式,需要的朋友可以參考下2024-01-01
在Go語(yǔ)言編程中,`omitempty`標(biāo)記用于JSON編解碼過(guò)程中控制字段是否被包含,當(dāng)結(jié)構(gòu)體字段標(biāo)記為`omitempty`且字段值為空時(shí),該字段不會(huì)出現(xiàn)在生成的JSON中,有助于優(yōu)化JSON結(jié)構(gòu)和減小數(shù)據(jù)體積,通過(guò)具體示例解釋了`omitempty`的工作機(jī)制和實(shí)際效果2024-09-09
舉就是將數(shù)據(jù)值一一列出來(lái),枚舉可以用來(lái)表示一些固定的值,枚舉是常量組成的,下面這篇文章主要給大家介紹了關(guān)于Golang中如何實(shí)現(xiàn)枚舉的相關(guān)資料,需要的朋友可以參考下2022-07-07
詳解Go語(yǔ)言實(shí)現(xiàn)線性查找算法和二分查找算法
線性查找又稱順序查找，它是查找算法中最簡(jiǎn)單的一種。二分查找，也稱折半查找，相比于線性查找，它是一種效率較高的算法。本文將用Go語(yǔ)言實(shí)現(xiàn)這兩個(gè)查找算法，需要的可以了解一下2022-12-12
golang如何用http.NewRequest創(chuàng)建get和post請(qǐng)求
這篇文章主要介紹了golang如何用http.NewRequest創(chuàng)建get和post請(qǐng)求問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-03-03
這篇文章主要為大家介紹了GO語(yǔ)言類型判斷及類型斷言，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步早日升職加薪2022-04-04
這篇文章主要介紹了go逆向符號(hào)恢復(fù)的相關(guān)知識(shí),本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2023-08-08