欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

golang連接sqlx庫的操作使用指南

 更新時(shí)間:2022年04月14日 11:58:51   作者:Jeff的技術(shù)棧  
這篇文章主要為大家介紹了golang連接sqlx庫的操作使用指南,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步早日升職加薪

sqlx庫使用指南

在項(xiàng)目中我們通常可能會使用database/sql連接MySQL數(shù)據(jù)庫。本文借助使用sqlx實(shí)現(xiàn)批量插入數(shù)據(jù)的例子,介紹了sqlx中可能被你忽視了的sqlx.InDB.NamedExec方法。

sqlx介紹

在項(xiàng)目中我們通??赡軙褂?code>database/sql連接MySQL數(shù)據(jù)庫。sqlx可以認(rèn)為是Go語言內(nèi)置database/sql的超集,它在優(yōu)秀的內(nèi)置database/sql基礎(chǔ)上提供了一組擴(kuò)展。這些擴(kuò)展中除了大家常用來查詢的Get(dest interface{}, ...) error和Select(dest interface{}, ...) error外還有很多其他強(qiáng)大的功能。

安裝sqlx

go get github.com/jmoiron/sqlx

基本使用

連接數(shù)據(jù)庫

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方法用來綁定SQL語句與結(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ù)操作

對于事務(wù)操作,我們可以使用sqlx中提供的db.Beginx()tx.Exec()方法。示例代碼如下:

func transactionDemo2()(err error) {
	tx, err := db.Beginx() // 開啟事務(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.Insqlx提供的一個(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)體,字段通過tag與數(shù)據(jù)庫中user表的列一致。

type User struct {
	Name string `db:"name"`
	Age  int    `db:"age"`
}

bindvars(綁定變量)

查詢占位符?在內(nèi)部稱為bindvars(查詢占位符),它非常重要。你應(yīng)該始終使用它們向數(shù)據(jù)庫發(fā)送值,因?yàn)樗鼈兛梢苑乐筍QL注入攻擊。database/sql不嘗試對查詢文本進(jìn)行任何驗(yàn)證;它與編碼的參數(shù)一起按原樣發(fā)送到服務(wù)器。除非驅(qū)動程序?qū)崿F(xiàn)一個(gè)特殊的接口,否則在執(zhí)行之前,查詢是在服務(wù)器上準(zhǔn)備的。因此bindvars是特定于數(shù)據(jù)庫的:

  • MySQL中使用?
  • PostgreSQL使用枚舉的$1、$2等bindvar語法
  • SQLite中?$1的語法都支持
  • Oracle中使用:name的語法

bindvars的一個(gè)常見誤解是,它們用來在sql語句中插入值。它們其實(shí)僅用于參數(shù)化,不允許更改SQL語句的結(jié)構(gòu)。例如,使用bindvars嘗試參數(shù)化列或表名將不起作用:

// ?不能用來插入表名(做SQL語句中表名的占位符)
db.Query("SELECT * FROM ?", "mytable")
// ?也不能用來插入列名(做SQL語句中列名的占位符)
db.Query("SELECT ?, ? FROM people", "name", "location")

自己拼接語句實(shí)現(xiàn)批量插入

比較笨,但是很好理解。就是有多少個(gè)User就拼接多少個(gè)(?, ?)。

// BatchInsertUsers 自行構(gòu)造批量插入的語句
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ù)對應(yīng)
		valueStrings = append(valueStrings, "(?, ?)")
		valueArgs = append(valueArgs, u.Name)
		valueArgs = append(valueArgs, u.Age)
	}
	// 自行拼接要執(zhí)行的具體語句
	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幫我們拼接語句和參數(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 會通過調(diào)用 Value()來展開它
	)
	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)問題,sql語句最后不能有空格和;,詳見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
}

把上面三種方法綜合起來試一下:

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查詢語句中實(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){
	// 動態(tài)填充id
	query, args, err := sqlx.In("SELECT name, age FROM user WHERE id IN (?)", ids)
	if err != nil {
		return
	}
	// sqlx.In 返回帶 `?` bindvar的查詢語句, 我們使用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){
	// 動態(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的查詢語句, 我們使用Rebind()重新綁定它
	query = DB.Rebind(query)
	err = DB.Select(&users, query, args...)
	return
}

當(dāng)然,在這個(gè)例子里面你也可以先使用IN查詢,然后通過代碼按給定的ids對查詢結(jié)果進(jìn)行排序。

參考鏈接:

Illustrated guide to SQLX

以上就是golang連接sqlx庫的操作使用指南的詳細(xì)內(nèi)容,更多關(guān)于golang操作連接sqlx庫的資料請關(guān)注腳本之家其它相關(guān)文章!

相關(guān)文章

  • Golang棧結(jié)構(gòu)和后綴表達(dá)式實(shí)現(xiàn)計(jì)算器示例

    Golang棧結(jié)構(gòu)和后綴表達(dá)式實(shí)現(xiàn)計(jì)算器示例

    這篇文章主要為大家介紹了Golang棧結(jié)構(gòu)和后綴表達(dá)式實(shí)現(xiàn)計(jì)算器示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪
    2022-07-07
  • Go語言錯(cuò)誤處理異常捕獲+異常拋出

    Go語言錯(cuò)誤處理異常捕獲+異常拋出

    這篇文章主要介紹了Go語言錯(cuò)誤處理異常捕獲和異常拋出,Go語言的作者認(rèn)為java等語言的錯(cuò)誤處理底層實(shí)現(xiàn)較為復(fù)雜,就實(shí)現(xiàn)了函數(shù)可以返回錯(cuò)誤類型以及簡單的異常捕獲,雖然簡單但是也非常精妙,大大的提高了運(yùn)行效率,下文需要的朋友可以參考一下
    2022-02-02
  • 解決golang 關(guān)于全局變量的坑

    解決golang 關(guān)于全局變量的坑

    這篇文章主要介紹了解決golang 關(guān)于全局變量的坑,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2021-05-05
  • Go語言連接Oracle數(shù)據(jù)庫的方法

    Go語言連接Oracle數(shù)據(jù)庫的方法

    這篇文章主要介紹了Go語言連接Oracle數(shù)據(jù)庫的方法,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
    2021-02-02
  • Golang httptest包測試使用教程

    Golang httptest包測試使用教程

    這篇文章主要介紹了Golang httptest包測試使用,httptest包的理念是,非常容易模擬http服務(wù),也就是說模擬響應(yīng)寫(response writer),提供給http處理器(handle),讓我們測試http服務(wù)端和客戶端很容易
    2023-03-03
  • Go type關(guān)鍵字(類型定義與類型別名的使用差異)用法實(shí)例探究

    Go type關(guān)鍵字(類型定義與類型別名的使用差異)用法實(shí)例探究

    這篇文章主要為大家介紹了Go type關(guān)鍵字(類型定義與類型別名的使用差異)用法實(shí)例探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪
    2024-01-01
  • 一篇文章學(xué)會GO語言中的變量

    一篇文章學(xué)會GO語言中的變量

    Go語言是靜態(tài)類型語言,因此變量有明確類型的,編譯器也會檢查變量類型的正確性,下面這篇文章主要給大家介紹了關(guān)于GO語言中變量的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),需要的朋友可以參考下
    2022-07-07
  • Golang 文件操作:刪除指定的文件方式

    Golang 文件操作:刪除指定的文件方式

    這篇文章主要介紹了Golang 文件操作:刪除指定的文件方式,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2021-04-04
  • Go 1.13中errors包的新變化示例解析

    Go 1.13中errors包的新變化示例解析

    這篇文章主要為大家介紹了Go 1.13中errors包的新變化示例解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪
    2023-12-12
  • 多階段構(gòu)建優(yōu)化Go?程序Docker鏡像

    多階段構(gòu)建優(yōu)化Go?程序Docker鏡像

    這篇文章主要為大家介紹了多階段構(gòu)建優(yōu)化Go?程序Docker鏡像,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪
    2022-08-08

最新評論