Go 語言操作 MySQL 之 CURD 操作

Meng小羽發表於2020-07-01
GoMySql

友情提示：此篇文章大約需要閱讀 10分鐘 50秒，不足之處請多指教，感謝你的閱讀。 訂閱本站

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)
}

參考（學習）文章：Go語言操作MySQL

本作品採用《CC 協議》，轉載必須註明作者和本文連結
Meng小羽

相關文章