使用 GORM 连接数据库并实现增删改查操作 | 豆包MarsCode AI刷题

在现代应用开发中，数据库的操作必不可少。为了简化 Go 语言项目中的数据库操作，GORM 作为一款流行的对象关系映射（ORM）工具，提供了强大的功能，帮助开发者使用更接近面向对象的方式操作数据库，无需手写复杂的 SQL 语句。GORM 支持 MySQL、PostgreSQL、SQLite 等多种数据库，并具备丰富的特性，适用于 CRUD（创建、读取、更新、删除）操作、事务、关联查询等。本文将详细讲解如何使用 GORM 连接数据库并实现常见的增删改查操作。

一、GORM 与数据库驱动的安装

在开始使用 GORM 前，我们需要在项目中安装 GORM 及其对应的数据库驱动包。这里我们选择使用 MySQL 数据库，GORM 为 MySQL 提供了专用的驱动程序。

使用以下命令安装 GORM 和 MySQL 驱动程序：

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

安装完成后，我们可以开始编写代码，完成数据库连接的基础配置。

二、连接数据库

在 GORM 中，数据库连接的建立非常方便，通常通过配置连接字符串（DSN）来完成。连接字符串包含了数据库地址、用户名、密码、数据库名等基本信息。我们将创建一个 connectDatabase 函数用于连接 MySQL 数据库。

在 connectDatabase 函数中，我们使用 GORM 提供的 gorm.Open() 方法来创建一个数据库连接实例。以下是具体实现：

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

func connectDatabase() (*gorm.DB, error) {
    dsn := "username:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        return nil, err
    }
    return db, nil
}

参数解析

• username 和 password 分别是数据库的用户名和密码。
• 127.0.0.1:3306 是数据库服务器的地址和端口，本例中使用的是本地地址。
• dbname 表示数据库的名称。
• charset=utf8mb4 设定字符编码为 utf8mb4，支持更多字符集。
• parseTime=True 解析时间格式。
• loc=Local 设置时区为本地。

该函数返回 *gorm.DB 类型的实例，这是数据库操作的核心对象。如果连接失败，则返回错误信息。通过这种封装的方式，我们可以更方便地在项目中复用数据库连接逻辑。

三、定义数据模型

在 ORM 模式中，通常将数据表和结构体（Struct）一一对应，以实现面向对象的数据库操作。在 GORM 中，每个数据表通常被定义为一个结构体，结构体的每个字段映射到数据库表的列。接下来，我们定义一个 User 模型，用于映射 MySQL 数据库中的 users 表。

type User struct {
    ID        uint   `gorm:"primaryKey"`
    Name      string `gorm:"type:varchar(100)"`
    Email     string `gorm:"type:varchar(100);unique"`
    Age       int
}

字段解释

• ID 字段被标记为主键，数据类型为 uint，即无符号整型。GORM 自动识别带有 primaryKey 标签的字段为表的主键。
• Name 字段代表用户的姓名，类型为 string，并使用了 gorm:"type:varchar(100)" 标签来限制其长度为 100。
• Email 字段代表电子邮件地址，添加了唯一约束，确保同一表中不出现重复的电子邮件。
• Age 字段记录用户的年龄，类型为 int。

GORM 会根据模型自动推导数据库表名，例如 User 结构体对应数据库中的 users 表。

四、自动迁移

在数据模型定义完成后，我们可以使用 GORM 的自动迁移功能 AutoMigrate 来创建数据库表结构。AutoMigrate 是一种自动化的表结构管理功能，GORM 会根据模型自动生成建表语句并在数据库中创建表结构。通过这种方式，我们可以确保代码和数据库表结构保持同步，避免手动管理表结构的复杂性。

func autoMigrate(db *gorm.DB) {
    db.AutoMigrate(&User{})
}

调用 AutoMigrate 后，GORM 会自动检查 User 模型中的字段和属性，然后生成建表语句或更新表结构。这在开发阶段非常有用，特别是当模型字段频繁变动时，可以减少手动更新数据库表结构的工作量。

五、实现 CRUD 操作

在完成数据库连接和模型定义后，我们可以开始实现 CRUD 操作。CRUD 是数据库操作的核心，包括创建（Create）、读取（Read）、更新（Update）和删除（Delete）四种基本操作。接下来我们会逐一讲解每个操作的具体实现。

创建（Create）

GORM 的 Create 方法用于向数据库中插入新记录。我们可以定义一个 createUser 函数来封装插入操作，确保逻辑清晰并便于复用。

func createUser(db *gorm.DB, user User) error {
    result := db.Create(&user)
    return result.Error
}

在这个 createUser 函数中，我们使用 db.Create() 方法将传入的 User 结构体插入数据库表。GORM 会自动将结构体字段和数据库列匹配，无需手动编写 SQL 语句。

读取（Read）

读取数据是数据库操作中最常见的一项需求。GORM 提供了多种读取方法，例如 First、Last 和 Find。First 方法用于查询符合条件的第一条记录，而 Find 用于查询多条记录。

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

func getAllUsers(db *gorm.DB) ([]User, error) {
    var users []User
    result := db.Find(&users)
    return users, result.Error
}

在 getUserByID 函数中，我们通过用户的 ID 查询一条记录，db.First() 方法会自动生成查询语句并返回结果。getAllUsers 函数则查询所有用户信息。

更新（Update）

更新数据库中的数据时，我们可以使用 Update、Updates 和 Save 方法，选择不同的更新方式。以下代码展示了更新用户电子邮件的示例：

func updateUserEmail(db *gorm.DB, id uint, newEmail string) error {
    result := db.Model(&User{}).Where("id = ?", id).Update("Email", newEmail)
    return result.Error
}

在 updateUserEmail 函数中，使用了 Model 方法指定更新的模型，结合 Where 方法筛选出需要更新的记录，最后使用 Update 修改指定字段。

删除（Delete）

删除数据是最后一个 CRUD 操作。GORM 的 Delete 方法可以删除指定记录。以下为删除用户的示例：

func deleteUser(db *gorm.DB, id uint) error {
    result := db.Delete(&User{}, id)
    return result.Error
}

在 deleteUser 函数中，我们通过传入用户的 ID 来删除特定的记录，GORM 会自动生成删除语句。

六、完整代码示例

以下是一个完整的 main 函数示例，演示如何连接数据库并执行 CRUD 操作：

func main() {
    db, err := connectDatabase()
    if err != nil {
        panic("failed to connect database")
    }
    autoMigrate(db)

    // 创建用户
    user := User{Name: "John", Email: "john@example.com", Age: 25}
    err = createUser(db, user)
    if err != nil {
        fmt.Println("Create error:", err)
    }

    // 获取用户
    retrievedUser, err := getUserByID(db, 1)
    if err == nil {
        fmt.Println("Retrieved user:", retrievedUser)
    } else {
        fmt.Println("Get user error:", err)
    }

    // 更新用户
    err = updateUserEmail(db, 1, "newemail@example.com")
    if err != nil {
        fmt.Println("Update error:", err)
    }

    // 删除用户
    err = deleteUser(db, 1)
    if err != nil {
        fmt.Println("Delete error:", err)
    }
}

1. 连接数据库

在 main 函数的第一步，我们调用了 connectDatabase() 函数，以建立与数据库的连接。connectDatabase() 函数负责创建数据库连接，并返回一个 *gorm.DB 实例。如果连接失败，则返回错误信息。使用 panic 是一种简单的错误处理方式，适用于示例代码，但在生产环境中，我们应采用更优雅的错误处理方式，比如记录日志或进行错误重试。

2. 自动迁移表结构

数据库连接成功后，调用 autoMigrate() 函数来检查并更新表结构。autoMigrate(db) 函数使用了 GORM 的自动迁移功能，可以根据模型（如 User 结构体）自动创建或更新数据库表结构。通过这种方式，当我们对模型字段进行增删改时，数据库表会自动更新，避免了手动编写 SQL 语句的繁琐。

3. 创建用户

在这一步中，我们创建一个 User 结构体实例并将其插入到数据库中。createUser() 函数调用 db.Create(&user) 来将用户插入数据库。这个函数会自动生成 SQL 插入语句，将结构体中的字段和数据库表的列进行映射。值得注意的是，user 结构体中的 ID 字段将会自动递增，因为它被定义为主键。GORM 会在插入成功后自动填充 ID 值到 user 实例中，这样可以让我们在插入完成后知道新记录的 ID 值。

4. 查询用户

创建用户后，我们尝试通过用户 ID 查询该用户的信息。getUserByID() 函数中调用了 db.First(&user, id)，用于查询 ID=1 的用户。First 方法会生成 SELECT 语句，并将查询结果映射到 User 结构体。若查询成功，函数返回查询的用户信息；若没有找到，err 会返回错误。

这种方式可以有效地从数据库中获取指定的记录。GORM 还提供了 Find 方法，用于查询多条记录，并支持链式方法，便于构建复杂的查询条件。

5. 更新用户

在数据库中插入和查询记录后，我们可以进一步对记录进行更新操作。updateUserEmail() 函数中使用 db.Model(&User{}).Where("id = ?", id).Update("Email", newEmail)，其中 Model 指定模型，Where 设置条件，而 Update 指定需要更新的字段。GORM 自动生成 UPDATE 语句，并将新邮箱地址更新到指定的用户记录中。

更新操作是数据库操作中常见的场景之一。GORM 提供了多种更新方法，例如 Updates 可以更新多个字段。我们可以根据需求灵活选择适合的更新方式。

6. 删除用户

在完成创建、查询、更新后，最后我们进行删除操作。deleteUser() 函数通过 db.Delete(&User{}, id) 来删除 ID=1 的用户记录。删除操作会生成 DELETE FROM users WHERE id = ? 的 SQL 语句，删除成功后该记录不再存在于数据库中。

七、总结

通过本文的讲解，展示了如何使用 GORM 在 Go 中完成数据库连接和增删改查操作。GORM 简化了数据库操作的流程，使得开发者可以通过定义模型和调用方法来操作数据库，而不必手写复杂的 SQL 语句。GORM 作为一个 Go 的 ORM 库，不仅支持基本的 CRUD 操作，还具备事务处理、关联查询等高级特性。在实际应用中，GORM 能够极大地简化数据库操作流程，尤其在复杂的业务场景下，让开发者专注于业务逻辑的实现，而非繁琐的 SQL 编写。