GORM实践 | 豆包MarsCode AI刷题GORM 是一个非常流行的 Go 语言的 ORM（对象关系映射）库，它简

GORM 是一个非常流行的 Go 语言的 ORM（对象关系映射）库，它简化了与数据库的交互，使得开发者可以通过面向对象的方式来操作数据库。GORM 支持多种数据库，如 MySQL、PostgreSQL、SQLite 和 SQL Server。

主要特点

简单易用：GORM 提供了简洁的 API，使得数据库操作变得非常直观和简单。
自动迁移：GORM 支持自动迁移功能，可以根据结构体定义自动创建或更新数据库表结构。
链式 API：GORM 提供了链式 API，可以方便地进行复杂的查询和操作。
事务支持：GORM 支持事务操作，可以确保数据库操作的原子性。
预加载（Eager Loading） ：GORM 支持预加载关联数据，减少数据库查询次数。
钩子（Hooks） ：GORM 提供了多种钩子，可以在数据库操作前后执行自定义逻辑。
日志记录：GORM 内置了日志记录功能，可以方便地记录 SQL 执行情况。

1. 设置项目环境并安装依赖

首先，在你的工作目录下创建一个新Go项目，并初始化go module（假设项目名为gorm_crud_demo）：

mkdir gorm_crud_demo
cd gorm_crud_demo
go mod init gorm_crud_demo

接下来，安装GORM和MySQL驱动：

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

2. 使用GORM连接数据库

创建一个新的文件main.go，并在其中编写代码，配置数据库连接。

假设我们有一个名为testdb的MySQL数据库，用户名为root，密码为password。以下代码展示了如何使用GORM连接数据库：

package main

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

var DB *gorm.DB

func ConnectDatabase() {
    dsn := "root:password@tcp(127.0.0.1:3306)/testdb?charset=utf8mb4&parseTime=True&loc=Local"
    database, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatal("Failed to connect to database:", err)
    }

    DB = database
    log.Println("Database connection successful.")
}

func main() {
    ConnectDatabase()
    // 后续可以在这里调用CRUD操作
}

在上述代码中，我们定义了一个ConnectDatabase函数用于连接数据库，并且将数据库连接实例赋值给全局变量DB以便后续使用。gorm.Open 是 GORM 库中用于打开数据库连接的核心函数。它负责初始化数据库连接，并返回一个 *gorm.DB 实例，该实例可以用于后续的数据库操作。

func Open(dialector gorm.Dialector, config *gorm.Config) (db *gorm.DB, err error)

dialector: 这是一个 gorm.Dialector 接口的实现，用于指定数据库类型和连接信息。GORM 提供了多种数据库的 Dialector，例如 mysql.Open、postgres.Open、sqlite.Open 等。
config: 这是一个 gorm.Config 结构体的指针，用于配置 GORM 的行为，例如日志记录、事务管理等。

3. 定义模型

接下来，我们定义一个简单的模型User，包含ID、Name和Email字段。

package main

import (
    "gorm.io/gorm"
)

type User struct {
    ID    uint   `gorm:"primaryKey"`
    Name  string `gorm:"size:100"`
    Email string `gorm:"size:100;unique"`
}

在main函数中调用DB.AutoMigrate来创建对应的表结构：

func main() {
    ConnectDatabase()

    // 自动迁移创建表
    DB.AutoMigrate(&User{})
}

AutoMigrate 是 GORM 提供的一个非常方便的方法，用于自动迁移数据库表结构。它可以根据你定义的 Go 结构体自动创建或更新数据库表，确保数据库表结构与结构体的定义保持一致。

func (db *gorm.DB) AutoMigrate(dst ...interface{}) error

db: 一个 *gorm.DB 实例，表示数据库连接。
dst: 一个或多个 Go 结构体的指针，表示需要迁移的模型。

AutoMigrate 方法的主要功能包括：

创建表：如果数据库中不存在与结构体对应的表，AutoMigrate 会自动创建该表。
更新表：如果数据库中已存在与结构体对应的表，AutoMigrate 会根据结构体的定义自动更新表结构，例如添加新字段、修改字段类型等。
保留数据：AutoMigrate 在更新表结构时会尽量保留现有数据，不会删除数据或表。

4. 实现增删改查操作

接下来，我们将实现一些基本的CRUD操作，包括创建用户、查询用户、更新用户和删除用户。

4.1 创建用户

创建一个CreateUser函数，通过接收name和email作为参数来创建新用户：

func CreateUser(name string, email string) (*User, error) {
    user := User{Name: name, Email: email}
    result := DB.Create(&user)
    return &user, result.Error
}

可以在main函数中测试：

func main() {
    ConnectDatabase()
    DB.AutoMigrate(&User{})

    user, err := CreateUser("Alice", "alice@example.com")
    if err != nil {
        log.Fatal("Failed to create user:", err)
    }
    log.Println("User created:", user)
}

4.2 查询用户

编写GetUserByID和GetAllUsers函数，分别用于通过ID查询用户和获取所有用户：

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

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

在 GORM 中，First 和 Find 是两个常用的查询方法，用于从数据库中检索记录。它们的主要区别在于查询结果的数量和查询条件的使用方式。First 方法用于从数据库中检索满足条件的第一条记录。如果找不到记录，会返回一个错误。Find 方法用于从数据库中检索满足条件的所有记录。如果找不到记录，不会返回错误。

在main函数中调用这些查询函数以测试其效果：

func main() {
    ConnectDatabase()
    DB.AutoMigrate(&User{})

    // 查询用户
    user, err := GetUserByID(1)
    if err != nil {
        log.Println("User not found:", err)
    } else {
        log.Println("User found:", user)
    }

    users, err := GetAllUsers()
    if err != nil {
        log.Println("Failed to fetch users:", err)
    } else {
        log.Println("All users:", users)
    }
}

4.3 更新用户

编写UpdateUserEmail函数，根据用户ID更新用户的Email地址：

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

在main函数中测试：

func main() {
    ConnectDatabase()
    DB.AutoMigrate(&User{})

    err := UpdateUserEmail(1, "alice_updated@example.com")
    if err != nil {
        log.Println("Failed to update user email:", err)
    } else {
        log.Println("User email updated successfully.")
    }
}

4.4 删除用户

编写DeleteUser函数，根据用户ID删除用户：

func DeleteUser(id uint) error {
    result := DB.Delete(&User{}, id)
    return result.Error
}

在main函数中测试删除功能：

func main() {
    ConnectDatabase()
    DB.AutoMigrate(&User{})

    err := DeleteUser(1)
    if err != nil {
        log.Println("Failed to delete user:", err)
    } else {
        log.Println("User deleted successfully.")
    }
}

总结

本文介绍了如何在Go语言中使用GORM连接MySQL数据库，并实现基础的增删改查操作。通过GORM，我们可以很方便地进行数据库的操作，从而提升开发效率。