詳解Go語言中的數(shù)據(jù)庫操作

更新時間：2023年02月07日 09:48:34 作者：孫琦Ray

數(shù)據(jù)庫是應(yīng)用開發(fā)中必須要掌握的技巧。這篇文章主要和大家介紹一下Go語言中相關(guān)的數(shù)據(jù)庫操作，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以了解一下

直接使用SQL語句，這種方式下，直接編寫SQL，簡單直觀，但是可維護(hù)性較差，同時對于數(shù)據(jù)庫依賴性比較強(qiáng)
使用ORM方式，將數(shù)據(jù)庫映射到對象上，方便操作，而且可以跨數(shù)據(jù)庫使用，也是開發(fā)框架中最常用的使用方式；同時，采用Migration管理數(shù)據(jù)庫Schema，形成版本控制

以下代碼及部分解釋由chatGPT生成

原生SQL方式

這段代碼實(shí)現(xiàn)了一個 Go 程序，它使用了標(biāo)準(zhǔn)庫中的 “database/sql” 包來連接 MySQL 數(shù)據(jù)庫并查詢表。

package main

import (
    "database/sql"
    "fmt"

    _ "github.com/go-sql-driver/mysql"
)

func main() {
    db, err := sql.Open("mysql", "username:password@tcp(host:port)/dbname")
    if err != nil {
        panic(err.Error())
    }
    defer db.Close()

    rows, err := db.Query("SELECT * FROM table_name")
    if err != nil {
        panic(err.Error())
    }
    defer rows.Close()

    columns, err := rows.Columns()
    if err != nil {
        panic(err.Error())
    }

    values := make([]sql.RawBytes, len(columns))
    scanArgs := make([]interface{}, len(values))
    for i := range values {
        scanArgs[i] = &values[i]
    }

    for rows.Next() {
        err = rows.Scan(scanArgs...)
        if err != nil {
            panic(err.Error())
        }

        var value string
        for i, col := range values {
            if col == nil {
                value = "NULL"
            } else {
                value = string(col)
            }
            fmt.Println(columns[i], ": ", value)
        }
        fmt.Println("-----------------------------------")
    }
    if err = rows.Err(); err != nil {
        panic(err.Error())
    }
}

首先，通過調(diào)用 sql.Open() 函數(shù)并傳入 “mysql” 作為驅(qū)動名稱和一個連接字符串來打開數(shù)據(jù)庫連接。如果打開連接失敗，程序會 panic。

接下來，使用 db.Query() 函數(shù)執(zhí)行 SQL 查詢語句，并將結(jié)果保存到變量 rows 中。如果查詢失敗，程序會 panic。

之后，調(diào)用 rows.Columns() 函數(shù)獲取表的列名，并在循環(huán)中調(diào)用 rows.Next() 函數(shù)來遍歷查詢結(jié)果。在每次循環(huán)中，調(diào)用 rows.Scan() 函數(shù)將當(dāng)前行的數(shù)據(jù)讀取到 values 數(shù)組中。然后，遍歷 values 數(shù)組，并將每列的值輸出到控制臺。

最后，使用 rows.Err() 函數(shù)檢查是否有任何錯誤，并在有錯誤時 panic。

注意：

需要先安裝 github.com/go-sql-driver/mysql
連接字符串需要替換成自己的連接信息
查詢語句需要替換成自己的表名

ORM方式

ORM (Object-Relational Mapping) 是一種將關(guān)系數(shù)據(jù)庫中的數(shù)據(jù)映射到對象上的技術(shù)。在 Go 語言中，可以使用第三方庫 “gorm” 來實(shí)現(xiàn) ORM 數(shù)據(jù)庫訪問，該庫也是Go語言使用較為廣泛的ORM庫。

下面是一個簡單的例子，它使用 gorm 庫來連接 MySQL 數(shù)據(jù)庫并查詢表：

package main

import (
    "fmt"

    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
)

type Product struct {
    gorm.Model
    Code  string
    Price uint
}

func main() {
    db, err := gorm.Open("mysql", "username:password@tcp(host:port)/dbname?charset=utf8&parseTime=True&loc=Local")
    if err != nil {
        panic("failed to connect database")
    }
    defer db.Close()

    // Migrate the schema
    db.AutoMigrate(&Product{})

    // Create
    db.Create(&Product{Code: "L1212", Price: 1000})

    // Read
    var product Product
    db.First(&product, 1)                   // find product with id 1
    db.First(&product, "code = ?", "L1212") // find product with code L1212

    // Update - update product's price to 2000
    db.Model(&product).Update("Price", 2000)

    // Delete - delete product
    db.Delete(&product)
}

該示例定義了一個名為 Product 的結(jié)構(gòu)體，其中包含了三個字段：Code、Price 和 gorm.Model。gorm.Model 是 GORM 提供的一個結(jié)構(gòu)體，包含了一些預(yù)定義字段，如 ID、CreatedAt、UpdatedAt 等。

程序首先連接到 MySQL 數(shù)據(jù)庫，然后使用 AutoMigrate 函數(shù)自動創(chuàng)建 Product 表，如果表已經(jīng)存在，則修改表結(jié)構(gòu)。

接下來使用 Create 函數(shù)向 Product 表中插入一條數(shù)據(jù)。使用 First 函數(shù)查詢表中的第一條記錄。

之后，使用 Model 函數(shù)更新該條記錄的價格，最后使用 Delete 函數(shù)刪除該條記錄。

注意：