GORM基础运用 | 豆包 MarsCode AI 刷题

SUMiRE233
28 阅读8分钟

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

在 Go 语言的生态系统中，GORM（Go Object Relational Mapping）是一个功能强大的数据库操作库，它提供了简洁、高效的方式来与数据库进行交互，实现对数据的增删改查操作。本文将详细介绍如何使用 GORM 连接数据库，并展示如何进行增删改查操作，同时强调 GORM 的使用要点。

一、GORM 简介

GORM 是一个 Go 语言的对象关系映射库，它允许开发者使用面向对象的方式来操作数据库。GORM 支持多种数据库，包括 MySQL、PostgreSQL、SQLite 等。它提供了丰富的功能，如自动迁移、关联关系、事务处理等，使得数据库操作变得更加简单和高效。

二、准备工作

  1. 安装 GORM
    使用 Go 的包管理工具go mod来安装 GORM。在你的项目目录下执行以下命令：

bash

Copy

go get -u gorm.io/gorm

  1. 选择数据库
    根据你的需求选择一个数据库，本文以 MySQL 为例。确保你已经安装了 MySQL 数据库，并创建了一个数据库和表。

  2. 配置数据库连接
    在你的项目中，创建一个文件来配置数据库连接。以下是一个配置文件的示例：

go

Copy

package main

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

var DB *gorm.DB

func initDB() {
    username := "your_username"
    password := "your_password"
    host := "localhost"
    port := "3306"
    database := "your_database"

    dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", username, password, host, port, database)
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err!= nil {
        panic("连接数据库失败：" + err.Error())
    }

    DB = db
}

在这个示例中，你需要将your_username、your_password、your_database替换为你的实际数据库用户名、密码和数据库名。

三、创建模型

在 GORM 中，模型是与数据库表对应的结构体。以下是一个简单的用户模型的示例：

go

Copy

package main

type User struct {
    ID   uint `gorm:"primaryKey"`
    Name string
    Age  int
}

在这个示例中，我们定义了一个User结构体，它包含了ID、Name和Age三个字段。其中，ID字段被标记为primaryKey，表示它是主键。

四、插入数据

  1. 插入单条数据
    使用 GORM 插入单条数据非常简单。以下是一个插入单条数据的示例：

go

Copy

func insertUser() {
    user := User{Name: "Alice", Age: 25}
    result := DB.Create(&user)
    if result.Error!= nil {
        panic("插入数据失败：" + result.Error.Error())
    }
    fmt.Println("插入数据成功，ID 为：", user.ID)
}

在这个示例中，我们创建了一个User对象，并使用Create方法将其插入到数据库中。如果插入成功，Create方法会返回一个*gorm.DB对象，我们可以通过这个对象来获取插入操作的结果。

  1. 插入多条数据
    GORM 也支持插入多条数据。以下是一个插入多条数据的示例：

go

Copy

func insertUsers() {
    users := []User{
        {Name: "Bob", Age: 30},
        {Name: "Charlie", Age: 35},
    }
    result := DB.Create(&users)
    if result.Error!= nil {
        panic("插入数据失败：" + result.Error.Error())
    }
    fmt.Println("插入数据成功，插入了", result.RowsAffected, "条数据")
}

在这个示例中，我们创建了一个包含多个User对象的切片，并使用Create方法将其插入到数据库中。

五、查询数据

  1. 查询单条数据
    使用 GORM 查询单条数据非常简单。以下是一个查询单条数据的示例：

go

Copy

func queryUser() {
    var user User
    result := DB.First(&user, 1)
    if result.Error!= nil {
        panic("查询数据失败：" + result.Error.Error())
    }
    fmt.Println("查询数据成功，用户名为：", user.Name)
}

在这个示例中，我们使用First方法查询了ID为 1 的用户。如果查询成功，First方法会将查询结果赋值给user变量。

  1. 查询多条数据
    GORM 也支持查询多条数据。以下是一个查询多条数据的示例：

go

Copy

func queryUsers() {
    var users []User
    result := DB.Find(&users)
    if result.Error!= nil {
        panic("查询数据失败：" + result.Error.Error())
    }
    for _, user := range users {
        fmt.Println("用户名为：", user.Name)
    }
}

在这个示例中，我们使用Find方法查询了所有的用户。如果查询成功，Find方法会将查询结果赋值给users切片。

  1. 条件查询
    GORM 支持使用条件查询来筛选数据。以下是一个条件查询的示例：

go

Copy

func queryUsersByAge() {
    var users []User
    result := DB.Where("age >?", 30).Find(&users)
    if result.Error!= nil {
        panic("查询数据失败：" + result.Error.Error())
    }
    for _, user := range users {
        fmt.Println("用户名为：", user.Name)
    }
}

在这个示例中，我们使用Where方法查询了年龄大于 30 的用户。

六、更新数据

  1. 更新单条数据
    使用 GORM 更新单条数据非常简单。以下是一个更新单条数据的示例：

go

Copy

func updateUser() {
    var user User
    result := DB.First(&user, 1)
    if result.Error!= nil {
        panic("查询数据失败：" + result.Error.Error())
    }
    user.Name = "Alice Updated"
    result = DB.Save(&user)
    if result.Error!= nil {
        panic("更新数据失败：" + result.Error.Error())
    }
    fmt.Println("更新数据成功")
}

在这个示例中，我们首先查询了ID为 1 的用户，然后修改了用户的名字，并使用Save方法将修改后的数据保存到数据库中。

  1. 更新多条数据
    GORM 也支持更新多条数据。以下是一个更新多条数据的示例：

go

Copy

func updateUsers() {
    result := DB.Model(&User{}).Where("age >?", 30).Update("age", 31)
    if result.Error!= nil {
        panic("更新数据失败：" + result.Error.Error())
    }
    fmt.Println("更新数据成功，更新了", result.RowsAffected, "条数据")
}

在这个示例中，我们使用Model方法指定要更新的模型，然后使用Where方法筛选要更新的数据，最后使用Update方法更新数据。

七、删除数据

  1. 删除单条数据
    使用 GORM 删除单条数据非常简单。以下是一个删除单条数据的示例：

go

Copy

func deleteUser() {
    var user User
    result := DB.First(&user, 1)
    if result.Error!= nil {
        panic("查询数据失败：" + result.Error.Error())
    }
    result = DB.Delete(&user)
    if result.Error!= nil {
        panic("删除数据失败：" + result.Error.Error())
    }
    fmt.Println("删除数据成功")
}

在这个示例中，我们首先查询了ID为 1 的用户，然后使用Delete方法将其从数据库中删除。

  1. 删除多条数据
    GORM 也支持删除多条数据。以下是一个删除多条数据的示例：

go

Copy

func deleteUsers() {
    result := DB.Where("age >?", 30).Delete(&User{})
    if result.Error!= nil {
        panic("删除数据失败：" + result.Error.Error())
    }
    fmt.Println("删除数据成功，删除了", result.RowsAffected, "条数据")
}

在这个示例中，我们使用Where方法筛选要删除的数据，然后使用Delete方法将其从数据库中删除。

八、GORM 的使用要点

  1. 自动迁移
    GORM 支持自动迁移，它可以根据你的模型自动创建或更新数据库表。在你的项目中，你可以在初始化数据库连接后调用AutoMigrate方法来进行自动迁移。以下是一个自动迁移的示例：

go

Copy

func initDB() {
    // 配置数据库连接
    //...

    // 自动迁移
    err := DB.AutoMigrate(&User{})
    if err!= nil {
        panic("自动迁移失败：" + err.Error())
    }
}

在这个示例中，我们在初始化数据库连接后调用了AutoMigrate方法，并传入了User模型。如果数据库中不存在User表，GORM 会自动创建这个表。如果表已经存在，GORM 会根据模型的定义自动更新表结构。

  1. 关联关系
    GORM 支持多种关联关系，如一对一、一对多、多对多等。在你的项目中，你可以使用HasOne、HasMany、BelongsTo、ManyToMany等方法来定义关联关系。以下是一个一对多关联关系的示例：

go

Copy

package main

import (
    "gorm.io/gorm"
)

type User struct {
    ID   uint `gorm:"primaryKey"`
    Name string
    Age  int
    Orders []Order
}

type Order struct {
    ID     uint `gorm:"primaryKey"`
    UserID uint
    Amount float64
}

在这个示例中，我们定义了一个User模型和一个Order模型，它们之间是一对多的关联关系。一个用户可以有多个订单。在User模型中，我们使用HasMany方法定义了与Order模型的关联关系。在Order模型中，我们使用BelongsTo方法定义了与User模型的关联关系。

  1. 事务处理
    GORM 支持事务处理，它可以保证多个数据库操作的原子性。在你的项目中，你可以使用Begin、Commit、Rollback等方法来进行事务处理。以下是一个事务处理的示例：

go

Copy

func transaction() {
    tx := DB.Begin()
    defer func() {
        if r := recover(); r!= nil {
            tx.Rollback()
        }
    }()

    user := User{Name: "Alice", Age: 25}
    result := tx.Create(&user)
    if result.Error!= nil {
        panic("插入数据失败：" + result.Error.Error())
    }

    order := Order{UserID: user.ID, Amount: 100.0}
    result = tx.Create(&order)
    if result.Error!= nil {
        panic("插入数据失败：" + result.Error.Error())
    }

    tx.Commit()
    fmt.Println("事务处理成功")
}

在这个示例中，我们使用Begin方法开启了一个事务，然后在事务中插入了一个用户和一个订单。如果插入成功，我们使用Commit方法提交事务。如果插入失败，我们使用Rollback方法回滚事务。

  1. 预加载
    GORM 支持预加载关联关系，它可以减少数据库查询的次数。在你的项目中，你可以使用Preload方法来预加载关联关系。以下是一个预加载关联关系的示例：

go

Copy

func preload() {
    var user User
    result := DB.Preload("Orders").First(&user, 1)
    if result.Error!= nil {
        panic("查询数据失败：" + result.Error.Error())
    }
    fmt.Println("用户名为：", user.Name)
    for _, order := range user.Orders {
        fmt.Println("订单金额为：", order.Amount)
    }
}

在这个示例中，我们使用Preload方法预加载了User模型的Orders关联关系。这样，在查询用户数据时，GORM 会自动查询用户的订单数据，并将其赋值给Orders字段。

九、总结

本文介绍了如何使用 GORM 连接数据库，并实现了对数据的增删改查操作。同时，我们还强调了 GORM 的使用要点，如自动迁移、关联关系、事务处理和预加载等。通过使用 GORM，我们可以更加高效地进行数据库操作，提高开发效率。希望本文对你有所帮助。

avatar
文章
阅读
粉丝