使用 GORM（Go 的 ORM 库）连接数据库| 豆包MarsCode AI刷题

旧愛
60 阅读7分钟

《使用GORM在Go中实现数据库的增删改查操作》

一、引言

在Go语言开发中，当我们需要与数据库进行交互时，使用对象关系映射（ORM）库可以大大简化操作流程。GORM就是一款广受欢迎的Go语言的ORM库，它提供了便捷的方式来连接数据库并执行常见的数据库操作，如增删改查。本文将详细介绍如何使用GORM连接数据库，并实现这些基本的数据库操作。

二、安装GORM及相关驱动

首先，我们需要安装GORM库以及对应数据库的驱动。假设我们使用MySQL数据库，以下是安装步骤：

  1. 确保你的项目已经初始化了Go模块（如果没有，可以通过 go mod init [项目名] 来初始化）。

  2. 安装GORM库：在终端执行 go get -u gorm.io/gorm。

  3. 安装MySQL驱动：执行 go get -u gorm.io/driver/mysql。

不同的数据库需要安装相应的驱动，GORM支持多种数据库，如PostgreSQL、SQLite等，安装驱动的方式类似。

三、连接数据库

安装好GORM和对应的驱动后，我们就可以开始连接数据库了。以下是使用GORM连接MySQL数据库的示例代码： package main

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

func main() { // 数据库连接字符串，格式为：用户名:密码@(127.0.0.1:3306)/数据库名?charset=utf8mb4&parseTime=True&loc=Local dsn := "root:password@(127.0.0.1:3306)/test_db?charset=utf8mb4&parseTime=True&loc=Local" var err error // 连接数据库 db, err := gorm.io/gorm.Open(mysql.New(mysql.Config{ DSN: dsn, }), &gorm.Config{}) if err!= nil { fmt.Println("连接数据库失败：", err) return } fmt.Println("数据库连接成功") } 在上述代码中：

• 首先定义了数据库连接字符串 dsn，其中包含了用户名、密码、数据库服务器地址、端口、数据库名以及一些字符集和时间格式的设置。

• 然后通过 gorm.Open 函数，传入MySQL驱动的配置信息，尝试连接数据库。如果连接过程中出现错误，会打印错误信息并退出程序；如果连接成功，则打印出“数据库连接成功”的提示。

四、定义数据模型

在使用GORM进行数据库操作之前，我们需要先定义数据模型，也就是与数据库表对应的结构体。假设我们要操作一个名为 users 的表，表中有 id（自增主键）、name、age 和 email 几个字段，对应的结构体定义如下： type User struct { ID uint gorm:"primaryKey" Name string gorm:"size:255" Age int Email string gorm:"size:255" } 在这个结构体定义中：

• 每个字段对应数据库表中的一个列。

• gorm:"primaryKey" 这个标签用于指定 ID 字段为表的主键。

• gorm:"size:255" 这样的标签可以设置字段对应的列在数据库中的一些属性，这里是设置字符串类型字段的最大长度为255。

五、增删改查操作

  1. 增加数据（Create）

以下是使用GORM向 users 表中添加一条新用户记录的代码： func main() { // 省略数据库连接部分代码，假设已经成功连接数据库并得到db对象

// 创建一个新用户实例
newUser := User{
    Name:  "John",
    Age:   25,
    Email: "john@example.com",
}

// 使用Create方法添加用户
result := db.Create(&newUser)
if result.Error!= nil {
    fmt.Println("添加用户失败：", result.Error)
    return
}
fmt.Println("添加用户成功，新用户的ID为：", newUser.ID)

} 在上述代码中：

• 首先创建了一个 User 结构体的实例 newUser，并设置了相应的字段值。

• 然后通过 db.Create 方法将这个新用户实例添加到数据库的 users 表中。如果添加过程中出现错误，会打印错误信息并退出程序；如果添加成功，则会打印出添加成功的提示以及新用户的ID值（因为在添加成功后，GORM会自动将自增主键的值赋给结构体的 ID 字段）。

  1. 删除数据（Delete）

假设我们要删除 users 表中 ID 为 1 的用户记录，以下是删除操作的代码： func main() { // 省略数据库连接部分代码，假设已经成功连接数据库并得到db对象

// 使用Delete方法删除指定ID的用户
result := db.Delete(&User{}, 1)
if result.Error!= nil {
    fmt.Println("删除用户失败：", result.Error)
    return
}
fmt.Println("删除用户成功")

} 在上述代码中：

• 通过 db.Delete 方法，传入要删除的结构体实例（这里传入一个空的 User 结构体实例，因为我们只需要根据条件删除，不需要具体的实例内容）和要删除的记录的ID值（这里是 1），就可以删除 users 表中指定ID的用户记录。如果删除过程中出现错误，会打印错误信息并退出程序；如果删除成功，则会打印出删除成功的提示。

  1. 修改数据（Modify）

比如我们要修改 users 表中 ID 为 2 的用户的年龄为 30，以下是修改操作的代码： func main() { // 省略数据库连接部分代码，假设已经成功连接数据库并得到db对象

// 创建一个要修改的用户实例，只设置需要修改的字段值
userToModify := User{
    ID:   2,
    Age:  30,
}

// 使用Updates方法修改用户数据
result := db.Updates(&userToModify)
if result.Error!= nil {
    fmt.Println("修改用户失败：", result.Error)
    return
}
fmt.Println("修改用户成功")

} 在上述代码中：

• 首先创建了一个 User 结构体的实例 userToModify，只设置了要修改的字段值（这里是 ID 和 Age）。

• 然后通过 db.Updates 方法将这个实例的内容更新到数据库的 users 表中对应的记录上。如果修改过程中出现错误，会打印错误信息并退出程序；如果修改成功，则会打印出修改成功的提示。

  1. 查询数据（Query）

以下是几种常见的查询数据的方式：

（1）查询所有用户记录 func main() { // 省略数据库连接部分代码，假设已经成功连接数据库并得到db对象

// 创建一个用于存储查询结果的切片
var users []User

// 使用Find方法查询所有用户记录
result := db.Find(&users)
if result.Error!= nil {
    fmt.Println("查询用户失败：", result.Error)
    return
}
fmt.Println("查询到的用户记录：", users)

} 在上述代码中：

• 首先创建了一个 User 结构体的切片 users，用于存储查询到的结果。

• 然后通过 db.Find 方法查询数据库的 users 表中所有用户记录，并将结果存储到 users 切片中。如果查询过程中出现错误，会打印错误信息并退出程序；如果查询成功，则会打印出查询到的用户记录。

（2）查询特定条件下的用户记录

比如查询年龄大于 20 的用户记录： func main() { // 省略数据库连接部分代码，假设已经成功连接数据库并得到db对象

// 创建一个用于存储查询结果的切片
var users []User

// 使用Where方法设置查询条件，然后用Find方法查询
result := db.Where("age > 20").Find(&users)
if result.Error!= nil {
    fmt.Println("查询用户失败：", result.Error)
    return
}
fmt.Println("查询到的用户记录：", users)

} 在上述代码中：

• 首先创建了一个 User 结构体的切片 users，用于存储查询到的结果。

• 然后通过 db.Where 方法设置查询条件（这里是年龄大于 20），再通过 Find 方法查询满足条件的用户记录，并将结果存储到 users 切片中。如果查询过程中出现错误，会打印错误信息并退出程序；如果查询成功，则会打印出查询到的用户记录。

六、总结

通过以上步骤，我们成功地使用GORM库连接了数据库，并实现了增删改查等基本的数据库操作。在实际应用中，我们可以根据具体的业务需求，灵活运用这些操作来处理数据库相关的任务。同时，GORM还提供了许多其他高级功能，如关联查询、事务处理等，可以进一步深入学习和探索。

avatar
文章
阅读
粉丝