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

无语98
103 阅读7分钟

GORM 简介

GORM 是 Go 语言中一个功能强大的 ORM（对象关系映射）库，它不仅支持数据库的 CRUD 操作，还提供了许多高级功能，如关联、预加载、事务管理等。使用 GORM，开发者可以将数据库表映射为 Go 的结构体，通过操作这些结构体来实现数据库的交互，极大地提高了开发效率。然而，ORM 库的抽象也带来了一些性能上的开销，因此在性能敏感型的应用中，需要权衡使用 ORM 的利弊。

GORM 的强大之处在于它的链式调用和灵活的配置，这使得代码更加简洁和易于维护。但是，这也意味着需要对 GORM 的内部机制有深入的理解，才能在遇到问题时快速定位和解决。例如，GORM 的预加载功能可以减少数据库查询的次数，但如果不正确使用，可能会导致查询效率降低。

环境准备与数据库连接

在开始之前，确保已经安装了 Go 语言环境和 MySQL 数据库。通过 go get 命令安装 GORM 库和 MySQL 驱动，这是连接数据库的第一步。连接字符串（DSN）是连接数据库的关键，它包含了数据库的地址、端口、用户名、密码等信息。在实际应用中，这些敏感信息应该从配置文件或环境变量中读取，而不是硬编码在代码中。

安装 GORM 和 MySQL 驱动

在终端中运行以下命令来安装 GORM 库和 MySQL 驱动：

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

创建数据库和表

在 MySQL 数据库中创建一个数据库（例如 test_db），并在该数据库中创建一个表（例如 users）：

CREATE DATABASE test_db;
USE test_db;

CREATE TABLE users (
  id INT AUTO_INCREMENT PRIMARY KEY,
  name VARCHAR(100) NOT NULL,
  age INT,
  email VARCHAR(100)
);

Snipaste_2024-11-15_10-31-20.png

代码与分析

在 main.go 文件中，我们首先导入了必要的包，并定义了 DSN 变量。使用 gorm.Open 函数连接数据库，并设置了日志输出，以便在连接失败时能够快速定位问题。

package main

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

var dsn = "username:password@tcp(127.0.0.1:3306)/test_db?charset=utf8mb4&parseTime=True&loc=Local"

func main() {
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatal("failed to connect database")
    }

    log.Println("Connected to database successfully")
}

请将 username 和 password 替换为你的 MySQL 用户名和密码。

实现 CRUD 操作

创建（Create）

在创建操作中，我们定义了一个 User 结构体，它与数据库中的 users 表相对应。GORM 通过结构体的字段和标签自动映射数据库的列。在实际开发中，这种映射关系需要根据数据库表的设计仔细调整，以确保数据的一致性和完整性。

type User struct {
    gorm.Model
    Name  string
    Age   int
    Email string
}

func createUser(db *gorm.DB, user User) {
    if err := db.Create(&user).Error; err != nil {
        log.Fatalf("Failed to create user: %v", err)
    }
    log.Printf("User created: %#v", user)
}

读取（Read）

读取操作包括单条查询和多条查询。GORM 提供了 First 和 Find 方法来实现这些操作。在实际应用中，我们经常需要根据复杂的条件进行查询，GORM 的 Where 方法提供了灵活的条件构造器。然而，过度使用复杂的查询条件可能会导致 SQL 语句的性能问题，因此在设计数据库和编写查询逻辑时，需要考虑到查询效率。

func findUser(db *gorm.DB, id uint) {
    var user User
    if err := db.First(&user, id).Error; err != nil {
        log.Fatalf("Failed to find user: %v", err)
    }
    log.Printf("Found user: %#v", user)
}

func findAllUsers(db *gorm.DB) {
    var users []User
    if err := db.Find(&users).Error; err != nil {
        log.Fatalf("Failed to find users: %v", err)
    }
    log.Printf("Found users: %#v", users)
}

更新（Update）

更新操作中，GORM 的 Save 方法提供了便捷的更新机制。它会根据结构体的主键自动更新对应的数据库记录。在实际开发中，我们需要注意更新操作的原子性和事务性，确保数据的一致性不被破坏。

func updateUser(db *gorm.DB, id uint) {
    var user User
    if err := db.First(&user, id).Error; err != nil {
        log.Fatalf("Failed to find user: %v", err)
    }

    user.Age = 31
    if err := db.Save(&user).Error; err != nil {
        log.Fatalf("Failed to update user: %v", err)
    }
    log.Printf("User updated: %#v", user)
}

删除（Delete）

删除操作中，GORM 提供了 Delete 方法来删除数据库记录。在实际应用中，删除操作需要谨慎处理，尤其是在涉及到外键约束的表中，不当的删除操作可能会导致数据完整性问题。

func deleteUser(db *gorm.DB, id uint) {
    if err := db.Delete(&User{}, id).Error; err != nil {
        log.Fatalf("Failed to delete user: %v", err)
    }
    log.Printf("User deleted with id: %v", id)
}

完整代码

在主函数中，我们将调用之前定义的数据库连接和 CRUD 操作的函数。下面是一个包含数据库连接和 CRUD 操作的完整主函数示例：

package main

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

// User 定义了一个用户模型，与数据库中的 users 表对应
type User struct {
    gorm.Model
    Name  string
    Age   int
    Email string
}

var dsn = "username:password@tcp(127.0.0.1:3306)/test_db?charset=utf8mb4&parseTime=True&loc=Local"

func main() {
    // 连接数据库
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatal("failed to connect database:", err)
    }

    // 打印连接成功的信息
    log.Println("Connected to database successfully")

    // 迁移模式，用于创建或修改数据库表结构
    if err := db.AutoMigrate(&User{}); err != nil {
        log.Fatalf("Failed to auto migrate: %v", err)
    }

    // 创建用户
    user := User{Name: "John Doe", Age: 30, Email: "johndoe@example.com"}
    createUser(db, user)

    // 查询用户
    findUser(db, 1)

    // 查询所有用户
    findAllUsers(db)

    // 更新用户
    updateUser(db, 1)

    // 删除用户
    deleteUser(db, 1)
}

// 创建用户函数
func createUser(db *gorm.DB, user User) {
    if err := db.Create(&user).Error; err != nil {
        log.Fatalf("Failed to create user: %v", err)
    }
    log.Printf("User created: %#v", user)
}

// 根据 ID 查询用户函数
func findUser(db *gorm.DB, id uint) {
    var user User
    if err := db.First(&user, id).Error; err != nil {
        log.Fatalf("Failed to find user: %v", err)
    }
    log.Printf("Found user: %#v", user)
}

// 查询所有用户函数
func findAllUsers(db *gorm.DB) {
    var users []User
    if err := db.Find(&users).Error; err != nil {
        log.Fatalf("Failed to find users: %v", err)
    }
    log.Printf("Found users: %#v", users)
}

// 更新用户函数
func updateUser(db *gorm.DB, id uint) {
    var user User
    if err := db.First(&user, id).Error; err != nil {
        log.Fatalf("Failed to find user: %v", err)
    }

    user.Age = 31
    if err := db.Save(&user).Error; err != nil {
        log.Fatalf("Failed to update user: %v", err)
    }
    log.Printf("User updated: %#v", user)
}

// 删除用户函数
func deleteUser(db *gorm.DB, id uint) {
    if err := db.Delete(&User{}, id).Error; err != nil {
        log.Fatalf("Failed to delete user: %v", err)
    }
    log.Printf("User deleted with id: %v", id)
}

在这个主函数中，我们首先连接到数据库，然后使用 AutoMigrate 方法来确保数据库中的表结构与我们的模型相匹配。接着，我们创建一个用户，查询用户，更新用户，最后删除用户。每个操作都封装在单独的函数中，这些函数在主函数中被调用。

请确保将 username 和 password 替换为你的 MySQL 数据库的实际用户名和密码。此外，这个示例假设你已经有一个名为 test_db 的数据库和一个名为 users 的表，表中至少包含 id、name、age 和 email 这些字段。如果表或字段不存在，AutoMigrate 方法将会自动创建它们。

结果：

Snipaste_2024-11-15_10-32-02.png

总结与个人见解

通过本文的介绍和实践，我们可以看到 GORM 作为一个 ORM 库，它极大地简化了数据库操作的复杂性，提高了开发效率。然而，ORM 库的抽象也带来了一定的性能开销和学习成本。在实际开发中，我们需要根据项目的具体需求和性能要求，合理使用 GORM 的各项功能。例如，在性能要求极高的场景下，可能需要直接使用 SQL 语句来替代 ORM 的某些操作。此外，GORM 的高级功能，如事务管理、预加载和关联操作，也需要根据实际业务逻辑合理应用，以避免不必要的性能损失。

总的来说，GORM 是一个强大的工具。作为开发者，我们需要深入理解 GORM 的工作原理和限制，结合实际业务需求，做出合理的技术选型和优化。在享受 GORM 带来的便利的同时，也要警惕它可能引入的问题，如过度抽象导致的性能问题，以及对 ORM 机制的过度依赖可能忽视 SQL 优化的重要性。正确使用 GORM，可以在提高开发效率的同时，保持应用的性能和可维护性。

avatar
文章
阅读
粉丝