使用GORM进行增删改查操作 | 豆包MarsCode AI刷题

用户109970126782
47 阅读8分钟

使用 GORM 操作数据库增删改查问题解惑及示例

在使用 GORM 进行数据库操作的过程中，常常会遇到各种各样的问题。本文将针对在使用 GORM 操作数据库进行增删改查操作时遇到的一些常见问题进行解惑，并详细展示相应的操作示例代码及逻辑。

一、编译器下载依赖问题

在下载 gorm.io/driver/mysql v1.5.7 和 gorm.io/gorm v1.25.12 这两个依赖时，出现了超时报错的情况，导致 go.mod 文件爆红。类比 Java 中的 Maven，推测可能是由于没有设置镜像网站，致使下载超时，依赖未能成功下载。

最终，通过在 PowerShell 中设置七牛云的 goproxy 代理解决了该问题，设置命令如下：

$env:GOPROXY = "https://goproxy.cn"

完成上述设置后，依赖下载成功。

二、连接数据库

若要对数据库进行操作，如同在 Java 中需要加载数据库的 JDBC 驱动一样，在 Go 语言中使用 gorm 包操作数据库时，也需要加载相应的数据库驱动并建立连接。以下是使用 gorm 包连接数据库的示例代码：

const dsn = "username:password@tcp(127.0.0.1:3306)/gogo?charset=utf8mb4&parseTime=True&loc=Local"
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})

为了便于确认数据库是否成功连接，可添加如下打印功能：

if err!= nil {
    fmt.Println("连接数据库失败:", err)
    return
}
fmt.Println("数据库连接成功")

通过上述代码，在连接出现错误时，能够及时打印错误信息并终止程序执行；若连接成功，则会输出相应的提示信息，方便后续进行数据库相关操作。

三、在连接成功后进行增删改查操作

1.新建一个结构体

ID uint `gorm:"primaryKey"` 
Name string `gorm:""` 
Age int `gorm:""` 
Email string `gorm:""` 
Password string `gorm:""` 
}

2.并在你的数据库中新建一张users表，和结构体的字段一一对应

SET NAMES utf8mb4; SET FOREIGN_KEY_CHECKS = 0; -- ---------------------------- -- Table structure for users -- ---------------------------- DROP TABLE IF EXISTS `users`; CREATE TABLE `users` ( `Name` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL, `Age` int NULL DEFAULT NULL, `Email` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL, `Password` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL, `Id` int NOT NULL AUTO_INCREMENT, PRIMARY KEY (`Id`) USING BTREE ) ENGINE = InnoDB AUTO_INCREMENT = 6 CHARACTER SET = utf8mb4 COLLATE = utf8mb4_0900_ai_ci ROW_FORMAT = Dynamic; SET FOREIGN_KEY_CHECKS = 1;

PS：为什么新建表时表名要使用users，因为gorm进行操作时自动把你操作的结构体的复数当成表名

3. 新增操作

在 GORM 中执行新增（插入）操作较为便捷。以下是向 user 表插入一条新用户记录的示例代码及相关解释：

func createUser(db *gorm.DB) {
    user := User{
        Name:     "John Doe",
        Age:      25,
        Email:    "johndoe@example.com",
        Password: "password123",
    }

    result := db.Create(&user)
    if result.Error!= nil {
        fmt.Println("插入数据失败:", result.Error)
        return
    }
    fmt.Println("插入数据成功，新记录的 ID 为:", user.ID)
}

代码执行逻辑如下：

  • 创建实例并赋值：首先创建了一个 User 结构体的实例 user，并为其各个字段赋予相应的值，这些值构成了要插入到数据库表中的新记录内容。
  • 执行插入操作：接着调用 db.Create(&user) 方法，此时 GORM 会依据 User 结构体的定义以及相关配置，将该实例对应的记录插入到数据库中对应的 user 表内。此处遵循默认的表名映射规则，即结构体 User 对应的表名默认为 user 表（当然，也可通过特定方式显式指定表名）。
  • 判断操作结果：最后通过检查 result.Error 是否为 nil 来判断插入操作是否成功。若插入操作成功，并且数据库表中主键 Id 设置为自增类型，那么还可以通过 user.ID 获取新插入记录的自增主键值，并进行相应的成功提示打印。

4. 查询操作

查询操作可细分为查询多条记录和查询单条记录两种常见情形，GORM 为此提供了便捷且灵活的方法，以满足不同的查询需求。

查询多条记录（获取所有用户信息）

以下是查询 user 表中所有用户信息的示例代码及详细解释：

func getAllUsers(db *gorm.DB) {
    var users []User
    result := db.Find(&users)
    if result.Error!= nil {
        fmt.Println("查询所有用户失败:", result.Error)
        return
    }
    for _, u := range users {
        fmt.Printf("ID: %d, Name: %s, Age: %d, Email: %s\n", u.ID, u.Name, u.Age, u.Email)
    }
}

代码逻辑解析如下：

  • 定义存储切片：先定义了一个 User 结构体的切片 users，用于存放从数据库中查询到的所有用户记录。
  • 执行查询操作：接着使用 db.Find(&users) 方法，该方法会从数据库对应的 user 表中检索出所有记录，并将它们填充到 users 切片中。在此过程中，同样需要通过检查 result.Error 来判断查询过程是否出现错误。
  • 处理查询结果：若查询成功，则通过循环遍历 users 切片，将每条用户记录的详细信息打印输出，以便直观地查看查询结果。
查询单条记录（根据指定条件查询某一用户信息，比如根据 ID 查询）

以下是根据指定条件（如 ID）查询某一用户信息的示例代码及逻辑说明：

func getSingleUser(db *gorm.DB, id uint) {
    var user User
    result := db.Where("id =?", id).First(&user)
    if result.Error!= nil {
        fmt.Println("查询单个用户失败:", result.Error)
        return
    }
    fmt.Printf("ID: %d, Name: %s, Age: %d, Email: %s\n", user.ID, user.Name, user.Age, user.Email)
}

代码的具体执行逻辑为：

  • 定义存储变量：首先定义了一个 User 结构体变量 user，用于存储查询到的单条用户记录。
  • 构建查询条件并执行查询：通过 db.Where("id =?", id).First(&user) 语句来构建查询条件，即筛选出 id 字段值等于传入参数 id 的那条记录（此处假设 id 为主键，正常情况下只会有一条匹配记录），然后使用 First 方法获取符合该条件的第一条（也是唯一一条）记录，并将其赋值给 user 变量。
  • 判断查询结果并输出信息：最后根据 result.Error 判断查询是否成功，若成功则输出该用户记录的详细信息。

5. 更改操作

当需要更新数据库中已存在的记录时，例如修改某个用户的年龄或者邮箱等信息，可按照以下方式进行操作。以下示例展示的是更新 ID 为某个特定值（假设为 1 ）的用户的年龄信息：

func updateUserAge(db *gorm.DB, id uint, newAge int) {
    result := db.Model(&User{}).Where("id =?", id).Update("age", newAge)
    if result.Error!= nil {
        fmt.Println("更新用户年龄失败:", result.Error)
        return
    }
    fmt.Println("更新用户年龄成功")
}

代码执行步骤如下：

  • 指定操作模型：先通过 db.Model(&User{}) 明确指定要操作的数据库表对应的模型，也就是告知 GORM 此次更新操作是针对 User 结构体所映射的 user 表进行的。
  • 设置更新条件：然后使用 Where("id =?", id) 来设置更新的条件，这里限定了只对 id 字段值等于传入参数 id 的那条记录进行更新操作（示例中假设要更新 ID 为 1 的用户记录，实际使用时可传入不同的 id 值）。
  • 指定更新字段及值：最后通过 Update("age", newAge) 来指定要更新的字段为 age ，并将其值更新为传入的 newAge 参数值。同样根据 result.Error 判断更新操作是否成功，并给出相应的提示信息。

6. 删除操作

若要从数据库中删除某条记录，例如删除 ID 为某个特定值（假设为 1 ）的用户记录，可参考以下示例代码及操作步骤：

func deleteUser(db *gorm.DB, id uint) {
    result := db.Where("id =?", id).Delete(&User{})
    if result.Error!= nil {
        fmt.Println("删除用户失败:", result.Error)
        return
    }
    fmt.Println("删除用户成功")
}

具体操作步骤如下：

  • 构建删除条件：通过 db.Where("id =?", id) 来构建删除记录的筛选条件，即找出 id 字段值等于传入参数 id 的那条记录（这里同样假设 id 是主键，正常只会匹配到一条记录）。

  • 执行删除操作：接着使用 Delete(&User{}) 方法执行删除操作，该方法会将符合条件的记录从数据库对应的 user 表中删除。

  • 判断操作结果：最后根据 result.Error 判断删除操作是否成功，并输出对应的提示信息告知操作结果。

在实际的项目开发中，可在 main 函数或者其他合适的业务逻辑函数中对上述增删改查操作的函数进行调用，示例如下：

func main() {
    const dsn = "username:password@tcp(127.0.0.1:3306)/gogo?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err!= nil {
        fmt.Println("连接数据库失败:", err)
        return
    }
    fmt.Println("数据库连接成功")

    // 新增操作示例
    createUser(db)

    // 查询所有用户操作示例
    getAllUsers(db)

    // 查询单个用户操作示例（假设查询ID为1的用户）
    getSingleUser(db, 1)

    // 更新用户年龄操作示例（假设更新ID为1的用户年龄为30岁）
    updateUserAge(db, 1, 30)

    // 删除用户操作示例（假设删除ID为1的用户）
    deleteUser(db, 1)
}

通过上述方式，能够完整地利用 GORM 在 Go 语言项目中实现对数据库的增删改查操作，从而方便地与数据库进行交互，以满足各种业务场景下的数据处理需求。不过需要注意的是，在实际开发过程中，还需考虑更多诸如数据验证、错误处理的细节以及事务管理等更为复杂的情况，可根据具体项目进一步完善和扩展这些操作逻辑。

avatar
文章
阅读
粉丝