詳解Go語言中的數據庫操作

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

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

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

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

原生SQL方式

這段代碼實現了一個 Go 程序，它使用了標準庫中的 “database/sql” 包來連接 MySQL 數據庫并查詢表。

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

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

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

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

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

注意：

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

ORM方式

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

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

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 的結構體，其中包含了三個字段：Code、Price 和 gorm.Model。gorm.Model 是 GORM 提供的一個結構體，包含了一些預定義字段，如 ID、CreatedAt、UpdatedAt 等。

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

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

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

注意：