使用 GORM（Go 的 ORM 库）连接数据库，并实现增删改查操作 | 青训营

Muirhead
89 阅读5分钟

引言: 在现代应用程序开发中，数据库操作是不可或缺的一部分。为了更方便地与数据库交互，许多开发者选择使用对象关系映射（ORM）库。在 Go 语言中，GORM 是一个流行的 ORM 库，它提供了简单易用的方式来连接数据库并执行各种数据库操作。本文将向您展示如何使用 GORM 在 Go 中连接并操作 MySQL 数据库实现增删改查

GORM简介

GORM（Go Object Relational Mapping）是一个用于Go编程语言的对象关系映射（ORM）库。ORM是一种编程技术，用于在关系型数据库和面向对象编程语言之间建立映射，从而简化数据库操作和数据持久化的过程。GORM旨在简化Go语言开发者与数据库之间的交互，使其更加高效和易于维护。它提供了许多功能，包括：

  1. 模型定义: GORM允许你通过Go的结构体来定义数据库中的表结构。这使得你可以通过代码来定义数据库模型，而无需直接编写SQL表格定义语句。

  2. 数据查询和操作: GORM提供了强大的查询构建器，允许你使用链式方法来构建复杂的数据库查询。它支持诸如条件查询、排序、分页等功能，使得数据库操作变得更加简洁和灵活。

  3. 关联关系: GORM使得在不同模型之间建立关联关系变得容易，包括一对一、一对多、多对多等类型的关系。这使得你可以轻松地在代码中处理复杂的数据关联操作。

  4. 事务支持: GORM支持事务，可以确保数据库操作的原子性，以及在一系列操作中出现错误时进行回滚。

  5. 迁移工具: GORM提供了数据库迁移工具，允许你在代码变更时自动调整数据库模式，从而简化数据库结构的演化过程。

  6. 钩子函数: GORM允许你在模型生命周期的不同阶段注册钩子函数，以便在数据库操作之前或之后执行特定的逻辑。

  7. 数据库支持: GORM支持多种关系型数据库，包括但不限于MySQL、PostgreSQL、SQLite等。

  8. 插件扩展: GORM允许你通过插件来扩展其功能，从而满足特定项目的需求。

在Go社区中，GORM被广泛使用，因为它可以显著简化数据库操作，同时提供了很多便捷的功能。不过，像所有的技术一样，它也有一些限制和适用场景: 例如引入ORM库通常会引用一些性能开销，因为它们需要将对象和数据库表之间进行映射, 在某些高性能和低延迟的应用中，直接使用原始的SQL可能更加高效。以及对于复杂的查询需求，手写原始的SQL可能更容易编写和优化。因此，我们需要根据自己的项目需求来决定是否使用GORM以及如何使用。而本篇笔记，我们只需要实现最简单的CRUD操作，GORM能够极大地简化代码，提高开发效率。

实战

  • 创建一个数据库

这里我们使用MySQL创建一个名为nba_db的数据库，信息如下

image.png

  • 连接数据库

当连接MySQL数据库时，我们首先需要确保已经安装了GORM和MySQL驱动。代码如下:

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

然后创建一个"main.go"的文件，将以下代码写入到文件中

package main

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

func main() {
    // 定义数据库连接信息
    dsn := "user:password@tcp(127.0.0.1:3306)/database_name?charset=utf8mb4&parseTime=True&loc=Local"

    // 连接数据库
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
       fmt.Println("连接失败: ", err)
       return
    }

    sqlDB, err := db.DB()
    if err != nil {
       fmt.Println("连接失败: ", err)
       return
    }
    defer func(sqlDB *sql.DB) {
       err := sqlDB.Close()
       if err != nil {

       }
    }(sqlDB)

    fmt.Println("连接成功")
}

在上面的代码中找到注释 "定义数据库连接信息"，将以下的部分替换为我们自己的MySQL数据库连接信息:

  1. user: MySQL 用户名 (例如在本地，就可以是 "root")
  2. password: MySQL 密码
  3. database_name: 连接的数据库名 (例如在本篇笔记里就是 "nba_db")

以本篇笔记为例，修改后的信息为 dsn:="root:xxxx@tcp(127.0.0.1:3306)/nba_dbcharset=utf8mb4&parseTime=True&loc=Local"

然后运行代码检查能否连接成功，若成功则会看到以下输出（这里用的是Goland）:

image.png

  • 增加数据

使用GORM创建结构体来映射数据库表，并执行增加数据

type Player struct {
    ID       uint
    Name     string
    TeamID   uint
    Position string
}

然后我们在前面连接数据库代码的下面加入如下代码即可插入新数据、

// 创建新数据
newTeam := Teams{TeamId: 1, TeamName: "金州勇士", TeamCity: "洛杉矶"}
db.Create(&newTeam)

newPlayer := Player{PlayerId: 1, PlayerName: "Stephen Curry", PlayerTeamId: 1, PlayerPosition: "PG"}
db.Create(&newPlayer)

注意，这里提到一个一开始执行插入数据时的常见错误，我在一开始想要插入一个新的 "Player" 数据，这时由于表 Teams为空，会报错，因为在 Player中PlayerTeamId是foreign key，因此需要先在Teams中插入数据，并有对应的foreign key，才能在表Player中执行插入操作。

  • 查询数据

紧接着我们就可以在上面添加数据后的基础上查询数据，代码和输出结果如下

image.png

  • 更改数据

在这里我们直接使用 Update方法来更新球员的位置信息，而不是通过定义专门的结构体来实现。注意，在Update方法中，第一个参数是要更新的字段名，第二个参数是要更新的新值。代码及输出如下所示。

image.png

  • 删除数据

最后我们来看删除数据，跟上面的更改操作类似，只不过把Update方法换成Delete方法。

image.png

最后的最后，别忘了关闭数据库连接

image.png

总结

本文详细介绍了如何使用 GORM 在 Go 中连接到 MySQL 数据库并执行增删改查等操作。通过结构化的示例代码和步骤说明，我们了解了如何利用 GORM 来简化数据库操作，提高开发效率。相信无论是新手还是有经验的开发者，都可以通过本文快速上手并掌握在 Go 中进行数据库操作的基本技巧。

avatar
文章
阅读
粉丝