Day5 gorm如何使用 | 青训营

啥都不会的小白
81 阅读5分钟

介绍：
GORM是一个强大的Go语言ORM（对象关系映射）库，它提供了简单易用的API来连接和操作MySQL数据库。本篇文章将详细介绍如何使用GORM来进行数据库操作，并分享一些性能优化的技巧，以提高数据库访问和查询的效率。

目录：

  1. 安装和引入GORM
  2. 创建数据库连接
  3. 定义模型结构
  4. 数据库迁移
  5. CRUD操作
    5.1 创建记录
    5.2 读取记录
    5.3 更新记录
    5.4 删除记录
  6. 查询数据
    6.1 查询单条记录
    6.2 查询多条记录
    6.3 排序和限制
    6.4 条件查询
    6.5 原生查询
  7. 关联查询
    7.1 一对一关联
    7.2 一对多关联
    7.3 多对多关联
  8. 事务操作
  9. 性能优化技巧
    9.1 批量操作
    9.2 关联查询预加载
    9.3 使用原生SQL
  10. 总结
  11. 安装和引入GORM
    首先，我们需要安装GORM库，可以通过以下命令在命令行中进行安装：
go get -u gorm.io/gorm

**

安装完成后，在代码中引入GORM库：

import "gorm.io/gorm"

**

  1. 创建数据库连接
    在开始使用GORM之前，我们需要创建一个数据库连接。使用GORM提供的Open函数来指定数据库类型、连接信息和其他参数：
db, err := gorm.Open(mysql.Open("user:password@tcp(localhost:3306)/database"), &gorm.Config{})
if err != nil {
    // 处理错误
}

**

  1. 定义模型结构
    在使用GORM操作数据库之前，我们需要定义模型结构，模型结构是与数据库表对应的Go结构体。例如，假设我们有一个名为User的表，可以定义如下结构体：
type User struct {
    gorm.Model
    Name  string
    Email string
}

**

此处使用了gorm.Model作为内嵌字段，它会自动为模型添加ID，CreatedAt，UpdatedAt和DeletedAt属性。

  1. 数据库迁移
    GORM提供了数据库迁移功能，用于创建和更新数据库表结构。可以通过以下命令进行迁移：
go run .\main.go migrate

**

在代码中添加迁移代码：

err := db.AutoMigrate(&User{})
if err != nil {
    // 处理错误
}

**

  1. CRUD操作
    5.1 创建记录：
user := User{Name: "Alice", Email: "alice@example.com"}
result := db.Create(&user)
if result.Error != nil {
    // 处理错误
}

**

5.2 读取记录：

var user User
result := db.First(&user, 1) // 读取ID为1的记录
if result.Error != nil {
    // 处理错误
}

**

5.3 更新记录：

result := db.Model(&user).Updates(User{Name: "Bob", Email: "bob@example.com"})
if result.Error != nil {
    // 处理错误
}

**

5.4 删除记录：

result := db.Delete(&user)
if result.Error != nil {
    // 处理错误
}

**

  1. 查询数据
    6.1 查询单条记录：
var user User
result := db.First(&user, "name = ?", "Alice")
if result.Error != nil {
    // 处理错误
}

**

6.2 查询多条记录：

var users []User
result := db.Find(&users, "age > ?", 18)
if result.Error != nil {
    // 处理错误
}

**

6.3 排序和限制：

var users []User
result := db.Order("name desc").Limit(10).Find(&users)
if result.Error != nil {
    // 处理错误
}

**

6.4 条件查询：

var users []User
result := db.Where("name LIKE ?", "%doe%").Find(&users)
if result.Error != nil {
    // 处理错误
}

**

6.5 原生查询：

var users []User
result := db.Raw("SELECT * FROM users WHERE age > ?", 18).Scan(&users)
if result.Error != nil {
    // 处理错误
}

**

  1. 关联查询
    7.1 一对一关联：
type Profile struct {
    gorm.Model
    UserID   uint
    Phone    string
    User     User `gorm:"foreignKey:UserID"`
}

**

7.2 一对多关联：

type Post struct {
    gorm.Model
    UserID uint
    Title  string
    User   User `gorm:"foreignKey:UserID"`
}

**

7.3 多对多关联：

type Role struct {
    gorm.Model
    Name  string
    Users []User `gorm:"many2many:user_roles;"`
}

**

  1. 事务操作
    GORM支持事务操作，我们可以使用Begin和Commit来启动和提交事务：
tx := db.Begin()
// 执行一系列的操作
tx.Create(&user)
tx.Update("name", "John")
// ...
tx.Commit()

**

  1. 性能优化技巧
    9.1 批量操作：
    当需要处理大批量数据时，可以使用GORM的批量操作功能来优化性能。GORM提供了CreateInBatches和Update方法，可以一次性插入或更新多条记录，减少与数据库的交互次数，从而提高效率。例如：
// 批量插入记录
users := []User{{Name: "Alice", Email: "alice@example.com"}, {Name: "Bob", Email: "bob@example.com"}}
result := db.CreateInBatches(&users, 100) // 每次插入100条记录
if result.Error != nil {
    // 处理错误
}

// 批量更新记录
result := db.Table("users").Where("age > ?", 18).Update("is_adult", true)
if result.Error != nil {
    // 处理错误
}

**

9.2 关联查询预加载：
GORM的预加载功能可以提前将关联数据加载到内存中，减少查询次数，提高查询效率。可以使用Preload方法进行预加载。例如：

// 预加载一对一关联
var user User
result := db.Preload("Profile").First(&user)
if result.Error != nil {
    // 处理错误
}

// 预加载一对多关联
var user User
result := db.Preload("Posts").First(&user)
if result.Error != nil {
    // 处理错误
}

// 预加载多对多关联
var user User
result := db.Preload("Roles").First(&user)
if result.Error != nil {
    // 处理错误
}

**

在预加载的同时，可以使用Select方法来指定需要加载的字段，以减少内存消耗。例如：

var user User
result := db.Preload("Posts", func(db *gorm.DB) *gorm.DB {
    return db.Select("id, title")
}).First(&user)
if result.Error != nil {
    // 处理错误
}

**

通过合理使用预加载，可以避免N+1查询问题，提高查询性能。

9.3 使用原生SQL

在某些情况下，使用原生SQL可以提供更高的性能和灵活性。GORM提供了Raw方法来执行原生SQL查询，可以根据具体需求选择使用GORM的查询方法还是原生SQL。

// 执行原生SQL查询
var users []User
result := db.Raw("SELECT * FROM users WHERE age > ?", 18).Scan(&users)
if result.Error != nil {
    // 处理错误
}

// 执行原生SQL更新
result := db.Exec("UPDATE users SET name = ? WHERE age > ?", "John", 18)
if result.Error != nil {
    // 处理错误
}

// 执行原生SQL删除
result := db.Exec("DELETE FROM users WHERE age > ?", 18)
if result.Error != nil {
    // 处理错误
}

**

原生SQL语句可以灵活地编写复杂的查询和操作逻辑，并且可以充分利用数据库的特性和索引，提高查询和操作的性能。

不过需要注意的是，使用原生SQL可能会导致代码与特定数据库紧密耦合，不够跨数据库，同时也有一定的安全风险。在使用原生SQL时，应尽量遵循良好的维护和安全性实践。

  1. 总结
    本文详细介绍了GORM的使用方法，包括安装和引入GORM库，创建数据库连接，定义模型结构，数据库迁移，CRUD操作，查询数据，关联查询和事务操作等。同时，还分享了一些性能优化的技巧，以提高数据库访问和查询的效率。使用GORM，我们能够更加方便快捷地进行数据库操作，并且通过性能优化技巧，提升应用程序的性能和响应速度。
avatar
文章
阅读
粉丝