青训营GORM学习指南

记得多吃蔬菜
109 阅读3分钟

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

GORM 是 Go 语言中广泛使用的 ORM（对象关系映射）库，能够帮助开发者更加方便地操作数据库，避免了直接编写复杂的 SQL 语句。本文将带你学习如何使用 GORM 连接数据库，并进行增、删、改、查等常见数据库操作。

一、安装 GORM 和数据库驱动

在使用 GORM 之前，需要安装 GORM 及相应的数据库驱动。以 MySQL 为例，安装命令如下：

bash
複製程式碼
go get -u gorm.io/gorm
go get -u gorm.io/driver/mysql

二、连接数据库

创建一个 Go 项目，配置 GORM 连接 MySQL 数据库。以下是一个连接数据库的示例代码：

go
複製程式碼
package main

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

func main() {
    // 配置数据库连接信息
    dsn := "user: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 {
        fmt.Println("数据库连接失败:", err)
        return
    }
    fmt.Println("数据库连接成功！")
}

在 dsn 中，替换 user、password 和 dbname 为实际的数据库用户名、密码和数据库名。

三、定义数据模型

在 GORM 中，需要定义结构体来映射数据库表。以下是一个简单的 User 模型示例：

go
複製程式碼
type User struct {
    ID       uint   `gorm:"primaryKey"`
    Name     string `gorm:"size:100"`
    Email    string `gorm:"unique"`
    Password string
}

此结构体表示一个 User 表，包含 ID、Name、Email 和 Password 字段，其中 ID 是主键，Email 字段具有唯一约束。

四、自动迁移

GORM 提供了 AutoMigrate 方法，可以根据结构体自动创建或更新数据库表。以下代码演示如何执行自动迁移：

go
複製程式碼
err = db.AutoMigrate(&User{})
if err != nil {
    fmt.Println("自动迁移失败:", err)
} else {
    fmt.Println("自动迁移成功！")
}

AutoMigrate 会自动根据 User 结构体创建表，并应用结构体中的字段定义。

五、实现增删改查操作

  1. 创建记录

使用 Create 方法向数据库插入新记录：

go
複製程式碼
newUser := User{Name: "Alice", Email: "alice@example.com", Password: "password123"}
result := db.Create(&newUser)
if result.Error != nil {
    fmt.Println("创建用户失败:", result.Error)
} else {
    fmt.Println("用户创建成功，ID:", newUser.ID)
}

Create 方法将新用户插入数据库，并自动生成 ID。

  1. 查询记录

使用 Find 或 First 方法查询记录：

go
複製程式碼
var user User
// 查询 ID 为 1 的用户
result = db.First(&user, 1)
if result.Error != nil {
    fmt.Println("查询用户失败:", result.Error)
} else {
    fmt.Printf("查询到用户: %+v\n", user)
}

// 按条件查询
result = db.Where("email = ?", "alice@example.com").First(&user)
if result.Error != nil {
    fmt.Println("按条件查询失败:", result.Error)
} else {
    fmt.Printf("按条件查询结果: %+v\n", user)
}
  1. 更新记录

使用 Save 或 Updates 方法更新记录：

go
複製程式碼
// 更新用户的 Name 字段
user.Name = "Alice Smith"
result = db.Save(&user)
if result.Error != nil {
    fmt.Println("更新用户失败:", result.Error)
} else {
    fmt.Println("用户更新成功")
}

// 使用 Updates 方法批量更新
result = db.Model(&user).Updates(User{Name: "Alice Doe", Email: "alice.doe@example.com"})
if result.Error != nil {
    fmt.Println("批量更新失败:", result.Error)
} else {
    fmt.Println("批量更新成功")
}
  1. 删除记录

使用 Delete 方法删除记录：

go
複製程式碼
// 删除 ID 为 1 的用户
result = db.Delete(&User{}, 1)
if result.Error != nil {
    fmt.Println("删除用户失败:", result.Error)
} else {
    fmt.Println("用户删除成功")
}

六、GORM 常用特性解析

  1. 钩子函数

GORM 支持在执行数据库操作时触发钩子函数。这些钩子函数允许开发者在操作前后插入自定义逻辑。例如：

go
複製程式碼
func (u *User) BeforeCreate(tx *gorm.DB) (err error) {
    fmt.Println("正在创建用户:", u.Name)
    return
}

常见的钩子包括 BeforeCreate、AfterCreate、BeforeUpdate 和 AfterUpdate，可用于数据预处理或记录日志。

  1. 事务处理

GORM 支持事务处理，适用于需要确保多个操作一致性的场景：

go
複製程式碼
tx := db.Begin()
if err := tx.Error; err != nil {
    fmt.Println("事务开启失败:", err)
    return
}

// 插入操作
if err := tx.Create(&newUser).Error; err != nil {
    tx.Rollback() // 事务回滚
    fmt.Println("插入失败，事务回滚:", err)
    return
}

// 提交事务
if err := tx.Commit().Error; err != nil {
    fmt.Println("事务提交失败:", err)
}

通过事务，可以确保操作的原子性，即要么全部成功，要么全部回滚。

avatar
文章
阅读
粉丝