使用 GORM（Go 的 ORM 库）连接数据库的小技巧｜青训营要充分利用 GORM 的威力，除了基本的增删改查操作外

GORM（Go Object Relational Mapping）是 Go 语言中的一款强大的 ORM 库，它大大简化了数据库操作的流程，提供了许多便捷的方法和功能。然而，要充分利用 GORM 的威力，除了基本的增删改查操作外，还有一些小技巧可以让你的数据库操作更加高效和灵活。

1. 批量插入数据

当你需要插入大量数据时，单个单个地插入会造成性能问题。GORM 提供了批量插入的方法，可以一次性插入多条数据，提高插入效率：

var users = []User{
    {Name: "Alice", Age: 25},
    {Name: "Bob", Age: 30},
    // ...
}

db.Create(&users)

2. 预加载关联数据

在查询涉及到关联数据的时候，使用预加载可以避免 N+1 查询问题，提高查询效率。使用 Preload 方法可以一次性加载相关联的数据：

var users []User
db.Preload("Orders").Find(&users)

3. 事务嵌套

当你需要在一个事务内部开启另一个事务时，可以使用 GORM 的事务嵌套功能。这样可以确保多个事务要么一起成功，要么一起失败：

db.Transaction(func(tx *gorm.DB) error {
    // 第一个事务操作
    if err := tx.Create(&user1).Error; err != nil {
        return err
    }

    // 在第一个事务内部开启第二个事务
    return tx.Transaction(func(subTx *gorm.DB) error {
        // 第二个事务操作
        if err := subTx.Create(&user2).Error; err != nil {
            return err
        }
        return nil
    })
})

4. 使用 Raw SQL

尽管 GORM 提供了丰富的查询和操作方法，有时候你可能需要执行自定义的 SQL 查询。可以使用 Raw 方法执行原始的 SQL 查询：

var result ResultStruct
db.Raw("SELECT * FROM users WHERE age > ?", 25).Scan(&result)

5. 批量更新

要批量更新记录，可以使用 Updates 方法。这样可以避免循环单个更新每一条记录：

db.Model(&User{}).Where("age > ?", 25).Updates(map[string]interface{}{"status": "active"})

6. 软删除

GORM 支持软删除（软删除是将记录标记为删除状态而不是直接从数据库中删除）。要使用软删除，需要在模型中添加 gorm.Model 字段，然后使用 Delete 方法来软删除记录：

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

// 软删除用户
db.Delete(&user)

7. 自动迁移

GORM 提供了自动迁移工具，可以根据模型定义自动创建数据库表。但在生产环境中要谨慎使用，避免意外修改数据库结构。可以考虑使用手动迁移来更好地控制表结构变更。

8. 自定义回调

GORM 允许你在不同的操作之前或之后添加自定义回调函数。这可以帮助你在执行数据库操作时执行一些自定义的逻辑，如记录日志、计算字段等。

func (user *User) BeforeCreate(tx *gorm.DB) (err error) {
    // 在创建记录之前执行的操作
    return nil
}

在使用 GORM 时，掌握这些小技巧可以让你更好地应对各种场景。无论是提高性能、处理事务、执行原始 SQL 查询，还是进行软删除和自定义回调，这些技巧都能够让你的数据库操作更加灵活和高效。

使用 GORM（Go 的 ORM 库）连接数据库的小技巧｜ 青训营