sqlx應用

sqlx

連接數據庫

聲明全局變量 + 定義結構體 + 連接數據庫

// 聲明全局變量db
var db *sqlx.DB

// 結構體存儲user表內輸出
type User struct {
	ID   int
	Age  int `db:"age"`
	Name string `db:"name"`
}

// 連接數據庫
func initDB() (err error) {

	dsn := "root:wangjian@tcp(127.0.0.1:3306)/db1?charset=utf8mb4&parseTime=True"
	// 連接數據庫 open + ping
	// db = sqlx.MustConnect("mysql", dsn) // 連接不成功直接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
}

查詢

? 是占位符 可以防止SQL注入攻擊

提供四種查詢的方法：

• 單條數據查詢 / db.Get()
• 多條數據查詢 / db.Select()
• 結構體與字段綁定查詢 / db.NamedQuery()
• 多參數查詢 / sqlx.In()

// 1. 查詢單條數據 / db.Get() / 相當於 query + scan
func queryRowDemo() {
	// sql語句 ——> 字符串
	sqlStr := "select id, name, age from user where id=?"
	var u User
	// 指定id為1 / 將數據存入結構體實例中 
	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)
}


// 2. 查詢多條數據 / db.Select()
func queryMultiRowDemo() {
	// sql語句 ——> 字符串
	sqlStr := "select id, name, age from user where id > ?"
	var users []User
	// 指定id>1 / 將數據存入結構體實例中
	err := db.Select(&users, sqlStr, 1)
	if err != nil {
		fmt.Printf("query failed, err:%v\n", err)
		return
	}
	fmt.Printf("users:%#v\n", users)
}


// 3. 綁定SQL語句與結構體或map中的同名字段，進行查詢操作 / db.NamedQuery() + rows.Next()
func namedQuery(){
	sqlStr := "SELECT * FROM user WHERE name=:name"

	// 1. 使用map做命名查詢
	rows, err := db.NamedQuery(sqlStr,
		map[string]interface{}{"name": "Tom"},
	)
	if err != nil {
		fmt.Printf("db.NamedQuery failed, err:%v\n", err)
		return
	}
	defer rows.Close() // 延時關閉

	// 循環存入user結構體實例，並輸出
	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)
	}
    
    
// 4. 批量查詢 根據批量ID查詢數據 / sqlx.In() + db.Rebind() + db.Select()
/*
users,_ := QueryByIDs([]int{4,5,6})
	for _, user := range users {
		fmt.Printf("user:%#v\n", user)
	}
*/
func QueryByIDs(ids []int)(users []User, err error){
	// 動態填充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
}

插入

語法：

插入單行：
insert into 表名(字段名1,字段名2,....) value (值1, 值2....)
插入多行：
insert into 表名(字段名1,字段名2,....) values(值1, 值2....),(值1, 值2....),...

實例

提供三種插入方法

• 插入單條（行）數據 / db.Exec()
• 結構體與字段綁定插入 / db.NamedExec()
• 批量插入數據 / sqlx.In()

// 1. 插入單條數據 / db.Exec()
func insertRowDemo() {
	sqlStr := "insert into user(name, age) values (?,?)"
	ret, err := db.Exec(sqlStr, "俠奢", 18)
	if err != nil {
		fmt.Printf("insert failed, err:%v\n", err)
		return
	}
	// 獲取新插入數據的id / 驗證是否插入成功
	theID, err := ret.LastInsertId()
	if err != nil {
		fmt.Printf("get lastinsert ID failed, err:%v\n", err)
		return
	}
	fmt.Printf("insert success, the id is %d.\n", theID)
}


// 2. 綁定SQL語句與結構體或map中的同名字段，進行插入操作 / db.NamedExec()
func insertUserDemo()(err error){
	sqlStr := "INSERT INTO user (name,age) VALUES (:name,:age)"
	_, err = db.NamedExec(sqlStr,
		map[string]interface{}{
			"name": "Tom",
			"age": 13,
		})
	return
}

// 3. 批量插入數據  sqlx.In / 注意傳入的參數是[]interface{}

// 3.1 結構體實現driver.Valuer接口
func (u User) Value() (driver.Value, error) {
	return []interface{}{u.Name, u.Age}, nil
}

// 3.2 sqlx.In拼接 SQL 語句和 插入參數 , 注意傳入的參數是[]interface{}
/*
	u1 := User{Age:  1, Name: "Daming"}
	u2 := User{Age:  2, Name: "Xiaoming"}
	u3 := User{Age:  3, Name: "Zhongming"}
	users := []interface{}{u1, u2, u3}
	BatchInsertUsers2(users)
*/
func BatchInsertUsers2(users []interface{}) error {
	query, args, _ := sqlx.In(
		"INSERT INTO user (name, age) VALUES (?), (?), (?)",
		users..., // 如果arg實現了 driver.Valuer, sqlx.In 會通過調用 Value()來展開它
	)
	fmt.Println(query) // 查看生成的querystring
	fmt.Println(args)  // 查看生成的args
	_, err := db.Exec(query, args...)
	return err
}

更新

語法

update 表名 set 字段名 = 新值, 字段名 = 新值 where 篩選條件

實例

// 更新數據 / db.Exec()
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()
	if err != nil {
		fmt.Printf("get RowsAffected failed, err:%v\n", err)
		return
	}
	fmt.Printf("update success, affected rows:%d\n", n)
}

刪除

語法

delete from 表名 where 篩選條件

實例

// 刪除數據 / db.Exec()
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()
	if err != nil {
		fmt.Printf("get RowsAffected failed, err:%v\n", err)
		return
	}
	fmt.Printf("delete success, affected rows:%d\n", n)
}

事務

// 事務操作 / db.Beginx()
func transactionDemo2()(err error) {
	// 開啟事務
	tx, err := db.Beginx()
	if err != nil {
		fmt.Printf("begin trans failed, err:%v\n", err)
		return err
	}

	defer func() {
		// 捕獲panic
		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")
		}
	}()

	// 更新語句1
	sqlStr1 := "Update user set age=18 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")
	}

	// 更新語句2
	sqlStr2 := "Update user set age=99 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
}