Go語言操作MySQL之CURD操作
MySQL 是目前開發中最常見的關系型數據庫,使用 Go 語言進行操控數據庫需要使用 Go 自帶database/sql和驅動go-sql-driver/mysql來實現,
創建好 Go 項目,需要引用驅動依賴:
- go get -u github.com/go-sql-driver/mysql
使用 MySQL 驅動:
- func Open(driverName, dataSourceName string) (*DB, error)
Open 打開一個 dirverName 指定的數據庫,dataSourceName 指定數據源,一般至少包括數據庫文件名和其它連接必要的信息。
初始化連接
- var db *sql.DB //聲明一個全局的 db 變量
- // 初始化 MySQL 函數
- func initMySQL() (err error) {
- dsn := "root:password@tcp(127.0.0.1:3306)/dbname"
- db, err = sql.Open("mysql", dsn)
- if err != nil {
- return
- }
- err = db.Ping()
- if err != nil {
- return
- }
- return
- }
- func main() {
- // 初始化 MySQL
- err := initMySQL()
- if err != nil {
- panic(err)
- }
- defer db.Close()
- }
初始化連接 MySQL 后需要借助 db.Ping 函數來判斷連接是否成功。
SetMaxOpenConns
- func (db *DB) SetMaxOpenConns(n int)
SetMaxOpenConns設置與數據庫建立連接的最大數目。
如果 n 大于 0 且小于最大閑置連接數,會將最大閑置連接數減小到匹配最大開啟連接數的限制。
如果 n <= 0,不會限制最大開啟連接數,默認為0(無限制)。
SetMaxIdleConns
- func (db *DB) SetMaxIdleConns(n int)
SetMaxIdleConns設置連接池中的最大閑置連接數。
如果 n 大于最大開啟連接數,則新的最大閑置連接數會減小到匹配最大開啟連接數的限制。
如果 n <= 0,不會保留閑置連接。
CURD
進行 CURD 操作,需要對數據庫建立連接,同時有供操作的數據(數據庫與數據表):
初始化數據
建立數據庫 sql_demo
- CREATE DATABASE sql_demo;
- USE sql_demo;
創建數據表 user
- 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;
查詢 SELECT
便于接收數據,定義一個 user 結構體接收數據:
- type user struct {
- id int
- age int
- name string
- }
查詢一行數據
db.QueryRow() 執行一次查詢,并期望返回最多一行結果(即 Row )。QueryRow 總是返回非 nil 的值,直到返回值的 Scan 方法被調用時,才會返回被延遲的錯誤。(如:未找到結果)
- func (db *DB) QueryRow(query string, args ...interface{}) *Row
實例代碼如下:
- // 查詢一行數據
- func queryRowDemo() (u1 *user, err error) {
- // 聲明查詢語句
- sqlStr := "SELECT id,name,age FROM user WHERE id = ?"
- // 聲明一個 user 類型的變量
- var u user
- // 執行查詢并且掃描至 u
- err = db.QueryRow(sqlStr, 1).Scan(&u.id, &u.age, &u.name)
- if err != nil {
- return nil, err
- }
- u1 = &u
- return
- }
- func main() {
- // 初始化 MySQL
- err := initMySQL()
- if err != nil {
- panic(err)
- }
- defer db.Close()
- u1, err := queryRowDemo()
- if err != nil {
- fmt.Printf("err:%s", err)
- }
- fmt.Printf("id:%d, age:%d, name:%s\n", u1.id, u1.age, u1.name)
- 」
結果如下:
- id:1, age:111, name:22
多行查詢
db.Query()執行一次查詢,返回多行結果(即 Rows ),一般用于執行 select 命令。參數 args 表示 query 中的占位參數。
- func (db *DB) Query(query string, args ...interface{}) (*Rows, error)
實例代碼如下:
- // 查詢多行數據
- func queryMultiRowDemo() {
- sqlStr := "SELECT id,name,age FROM user WHERE id > ?"
- rows, err := db.Query(sqlStr, 0)
- if err != nil {
- fmt.Printf("query data failed,err:%s\n", err)
- return
- }
- // 查詢完數據后需要進行關閉數據庫鏈接
- defer rows.Close()
- for rows.Next() {
- var u user
- err := rows.Scan(&u.id, &u.age, &u.name)
- if err != nil {
- fmt.Printf("scan data failed, err:%v\n", err)
- return
- }
- fmt.Printf("id:%d name:%s age:%d\n", u.id, u.name, u.age)
- }
- }
執行結果:
- id:1 name:111 age:22
- id:3 name:張三 age:22
使用 rows.Next() 循環讀取結果集中的數據。
增加數據 INSERT
增加、刪除、更新操作均使用 Exec 方法。
- func (db *DB) Exec(query string, args ...interface{}) (Result, error)
實例代碼如下:
- // 增加一行數據
- func insertRowDemo() {
- sqlStr := "INSERT INTO user(name, age) VALUES(?, ?)"
- result, err := db.Exec(sqlStr, "小羽", 22)
- if err != nil {
- fmt.Printf("insert data failed, err:%v\n", err)
- return
- }
- id, err := result.LastInsertId()
- if err != nil {
- fmt.Printf("get insert lastInsertId failed, err:%v\n", err)
- return
- }
- fmt.Printf("insert success, id:%d\n", id)
- }
執行結果:
- insert success, id:4
更新數據 UPDATE
- // 更新一組數據
- func updateRowDemo() {
- sqlStr := "UPDATE user SET age = ? WHERE id = ?"
- result, err := db.Exec(sqlStr, 22, 1)
- if err != nil {
- fmt.Printf("update data failed, err:%v\n", err)
- return
- }
- n, err := result.RowsAffected()
- if err != nil {
- fmt.Printf("get rowsaffected failed, err:%v\n", err)
- return
- }
- fmt.Printf("update success, affected rows:%d\n", n)
- }
刪除數據 DELETE
- // 刪除一行數據
- func deleteRowDemo() {
- sqlStr := "DELETE FROM user WHERE id = ?"
- result, err := db.Exec(sqlStr, 2)
- if err != nil {
- fmt.Printf("delete data failed, err:%d\n", err)
- return
- }
- n, err := result.RowsAffected()
- if err != nil {
- fmt.Printf("get affected failed, err:%v\n", err)
- return
- }
- fmt.Printf("delete success, affected rows:%d\n", n)
- }