您好,登錄后才能下訂單哦!
這篇“golang怎么連接sqlx庫”文章的知識點大部分人都不太理解,所以小編給大家總結(jié)了以下內(nèi)容,內(nèi)容詳細(xì),步驟清晰,具有一定的借鑒價值,希望大家閱讀完這篇文章能有所收獲,下面我們一起來看看這篇“golang怎么連接sqlx庫”文章吧。
在項目中我們通??赡軙褂?code>database/sql連接MySQL數(shù)據(jù)庫。本文借助使用sqlx
實現(xiàn)批量插入數(shù)據(jù)的例子,介紹了sqlx
中可能被你忽視了的sqlx.In
和DB.NamedExec
方法。
在項目中我們通??赡軙褂?code>database/sql連接MySQL數(shù)據(jù)庫。sqlx
可以認(rèn)為是Go語言內(nèi)置database/sql
的超集,它在優(yōu)秀的內(nèi)置database/sql
基礎(chǔ)上提供了一組擴展。這些擴展中除了大家常用來查詢的Get(dest interface{}, ...) error
和Select(dest interface{}, ...) error
外還有很多其他強大的功能。
go get github.com/jmoiron/sqlx
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) }
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 }
與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ù)操作,我們可以使用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
提供的一個非常方便的函數(shù)。
為了方便演示插入數(shù)據(jù)操作,這里創(chuàng)建一個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;
定義一個user
結(jié)構(gòu)體,字段通過tag與數(shù)據(jù)庫中user表的列一致。
type User struct { Name string `db:"name"` Age int `db:"age"` }
查詢占位符?
在內(nèi)部稱為bindvars(查詢占位符),它非常重要。你應(yīng)該始終使用它們向數(shù)據(jù)庫發(fā)送值,因為它們可以防止SQL注入攻擊。database/sql
不嘗試對查詢文本進(jìn)行任何驗證;它與編碼的參數(shù)一起按原樣發(fā)送到服務(wù)器。除非驅(qū)動程序?qū)崿F(xiàn)一個特殊的接口,否則在執(zhí)行之前,查詢是在服務(wù)器上準(zhǔn)備的。因此bindvars
是特定于數(shù)據(jù)庫的:
MySQL中使用?
PostgreSQL使用枚舉的$1
、$2
等bindvar語法
SQLite中?
和$1
的語法都支持
Oracle中使用:name
的語法
bindvars
的一個常見誤解是,它們用來在sql語句中插入值。它們其實僅用于參數(shù)化,不允許更改SQL語句的結(jié)構(gòu)。例如,使用bindvars
嘗試參數(shù)化列或表名將不起作用:
// ?不能用來插入表名(做SQL語句中表名的占位符) db.Query("SELECT * FROM ?", "mytable") // ?也不能用來插入列名(做SQL語句中列名的占位符) db.Query("SELECT ?, ? FROM people", "name", "location")
比較笨,但是很好理解。就是有多少個User就拼接多少個(?, ?)
。
// 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 { // 此處占位符要與插入值的個數(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 }
前提是需要我們的結(jié)構(gòu)體實現(xiàn)driver.Valuer
接口:
func (u User) Value() (driver.Value, error) { return []interface{}{u.Name, u.Age}, nil }
使用sqlx.In
實現(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實現(xiàn)了 driver.Valuer, sqlx.In 會通過調(diào)用 Value()來展開它 ) fmt.Println(query) // 查看生成的querystring fmt.Println(args) // 查看生成的args _, err := DB.Exec(query, args...) return err }
注意 :該功能需1.3.1版本以上,并且1.3.1版本目前還有點問題,sql語句最后不能有空格和;
,詳見issues/690。
使用NamedExec
實現(xiàn)批量插入的代碼如下:
// BatchInsertUsers3 使用NamedExec實現(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) } }
關(guān)于sqlx.In
這里再補充一個用法,在sqlx
查詢語句中實現(xiàn)In查詢和FIND_IN_SET函數(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');
。
查詢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 }
查詢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)然,在這個例子里面你也可以先使用IN
查詢,然后通過代碼按給定的ids對查詢結(jié)果進(jìn)行排序。
以上就是關(guān)于“golang怎么連接sqlx庫”這篇文章的內(nèi)容,相信大家都有了一定的了解,希望小編分享的內(nèi)容對大家有幫助,若想了解更多相關(guān)的知識內(nèi)容,請關(guān)注億速云行業(yè)資訊頻道。
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進(jìn)行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。