使用 GORM 连接数据库并实现增删改查操作

GORM 是 Go 语言中一个强大的 ORM（Object-Relational Mapping）库，它提供了简单的方式来连接数据库，并且能够轻松地进行数据库操作。在本文中，如何使用 GORM 来连接数据库并实现常见的增删改查操作。

步骤 1：安装 GORM

首先，需要在的 Go 项目中安装 GORM。可以使用下面的命令来安装：

   go get -u gorm.io/gorm go get -u gorm.io/driver/sqlite

上面的命令会安装 GORM 以及一个 SQLite 驱动，你也可以根据需要选择其他数据库驱动。

步骤 2：创建数据库模型

在使用 GORM 进行数据库操作之前，我们需要定义数据库模型。模型是一个 Go 结构体，代表了数据库中的表格。

package main 
import ( 
"gorm.io/gorm" 
"gorm.io/driver/sqlite" 
) 
// 定义模型结构 
type User struct {
    gorm.Model
    Username string 
    Email string
}

在上面的代码中，我们创建了一个 User 结构体，它继承了 gorm.Model，这个结构体会自动为我们添加一些常见的字段，如 ID、CreatedAt、UpdatedAt 和 DeletedAt。

步骤 3：连接数据库

现在，我们需要在 Go 代码中连接到数据库。

package main

import (
    "gorm.io/gorm"
    "gorm.io/driver/sqlite"
)

func main() {
    // 连接到 SQLite 数据库
    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    if err != nil {
        panic("无法连接数据库")
    }
    defer db.Close()

    // 自动迁移模型
    db.AutoMigrate(&User{})
}

在上述代码中，我们使用 gorm.Open 函数连接到 SQLite 数据库，并且通过 db.AutoMigrate(&User{}) 自动创建 User 表格。我们使用 GORM 的 gorm.Open() 函数来连接到 SQLite 数据库。我们还使用 AutoMigrate() 函数来自动创建表格（在这个示例中，我们将在后面的部分定义一个 Product 结构来表示表格）。

定义模型

在 GORM 中，模型是数据库表格的映射。我们需要创建一个 Go 结构来表示表格的每一行记录。以下是一个示例 Product 结构：

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

在上述结构中，我们使用了 gorm.Model 嵌入类型，它提供了一些默认的字段，如 ID、CreatedAt 和 UpdatedAt，用于跟踪记录的创建和更新时间。我们还定义了 Code 和 Price 字段来表示产品的编码和价格。

步骤 4：实现增删改查操作

现在，我们可以开始实现对数据库的操作了。

1. 创建用户（增加操作）

要向数据库中添加新记录，我们可以使用 Create() 函数。以下是一个示例代码，演示如何向数据库中添加一个新的产品记录：

func createUser(db *gorm.DB, username string, email string) {
    // 创建一个新的产品记录 
    newProduct := Product{Code: "Laptop", Price: 1000}
    db.Create(&newProduct)
}

在上述代码中，我们首先创建一个新的 Product 结构，然后使用 db.Create() 函数将它添加到数据库中。

2. 查询用户（查询操作）

GORM 提供了丰富的查询功能，使我们能够轻松地检索数据。以下是一些示例查询操作：有很多的查询方式，查询单个记录，查询多个记录，条件查询，按照排序查询。

下面示例一个查询方式：

func getUserByID(db *gorm.DB, id uint) User {
    var user User
    db.First(&user, id)
    return user
}

3. 更新用户（更新操作）

要更新数据库中的记录，我们可以使用 Save() 函数或 Update() 函数。以下是一些示例代码：

func updateUserEmail(db *gorm.DB, id uint, newEmail string) {
    // 更新产品价格 
    db.Model(&product).Update("Price", 1200) // 使用结构更新多个字段 
    
    db.Model(&product).Updates(Product{Price: 1200, Code: "New Laptop"})
}

在上述代码中，我们使用 Update() 函数来更新产品的价格，并使用 Updates() 函数同时更新多个字段。

4. 删除用户（删除操作）

要从数据库中删除记录，我们可以使用 Delete() 函数。以下是一个示例代码，演示如何删除特定的产品记录：


func delete(db *gorm.DB, id uint) {
    // 删除特定的产品记录 
    db.Delete(&product, 1)
}

在上述代码中，我们使用 Delete() 函数来删除 ID 为 1 的产品记录。使用 GORM 连接数据库，并执行基本的增、删、改、查操作。GORM 提供了丰富的功能，可以简化数据库操作，同时还提供了良好的模型定义和查询语法。在 Go 语言中需要与数据库交互时，GORM 是一个强大的工具，可以提高开发效率并减少错误。

注意事项

gorm.Open() : 这个函数用于连接到数据库。注意以下几点：

在第一个参数中指定数据库驱动（这里是 SQLite），并提供数据库连接字符串（在这个例子中是 "test.db"）。
第二个参数是 GORM 配置，通常为空白的 &gorm.Config{}。
始终检查连接错误 (err)，以确保成功连接到数据库。

模型结构体: 在 GORM 中，需要创建一个 Go 结构体来表示数据库表的映射。在模型结构体中，通常使用字段来表示表格中的列。注意以下几点：

模型结构体可以包含额外的字段，但通过嵌入 gorm.Model 可以自动添加默认的数据库字段（如 ID、CreatedAt 和 UpdatedAt）。
模型字段的名称和类型应与数据库表格的列对应。

Create() : 这个方法用于向数据库中添加新记录。注意以下几点：

创建一个新的结构体实例，填充所需的字段。
使用 & 操作符将结构体传递给 Create() 方法。
要确保字段的类型和值与模型结构体定义一致。

First() : 这个方法用于查询单个记录。注意以下几点：

第一个参数是要存储查询结果的结构体变量。
第二个参数是查询条件，通常是记录的唯一标识符。

Find() : 这个方法用于查询多个记录。注意以下几点：

第一个参数是一个切片，用于存储查询结果。
如果没有提供查询条件，将检索所有记录。
Model() : 这个方法用于指定要更新的模型。
Update() : 这个方法用于更新模型的字段。注意以下几点：
- 第一个参数是要更新的字段名称。
- 第二个参数是新的字段值。

Delete() : 这个方法用于删除记录。注意以下几点：

第一个参数是要删除的记录的结构体实例。
第二个参数是要删除的记录的唯一标识符。

总的注意事项：

错误处理: 始终检查数据库操作的错误。数据库连接可能会失败，查询可能不会找到记录，更新和删除可能会引发错误。良好的错误处理有助于诊断问题并增强应用程序的稳定性。
数据验证: 在执行数据库操作之前，进行数据验证是重要的。确保传递给数据库的数据类型和值是有效的，以防止数据库操作失败或引发异常。
事务: 在需要一组相关的数据库操作时，使用事务可以确保操作的一致性。GORM 提供了事务支持，允许在一个事务中执行多个操作，如果其中任何一个失败，可以回滚整个事务。
性能考虑: 考虑到性能，避免在大数据集上执行没有索引的查询，因为这可能导致较慢的查询性能。确保在数据库中使用适当的索引来提高查询速度。
模型设计: 数据库模型的设计非常重要。在创建模型时，考虑表之间的关系，使用外键来定义关联，以及如何组织数据以便于查询。

使用 GORM（Go 的 ORM 库）连接数据库，并实现增删改查操作，把实现过程整理成文章｜ 青训营