Golang中使用Gorm Hooks与Postgres的实践指南

是的，GORM Hooks 完全可以与 PostgreSQL 数据库一起使用。GORM 的 Hooks 功能是数据库无关的，它通过回调机制在特定操作（如创建、更新、查询、删除）前后执行自定义逻辑，与底层使用的数据库驱动无关。

以下是一个在 PostgreSQL 中使用 GORM Hooks 的完整示例：

1. 定义模型并实现 Hook 方法

package main

import (
    "fmt"
    "time"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
    "gorm.io/gorm/logger"
)

type User struct {
    ID        uint      `gorm:"primaryKey"`
    Name      string    `gorm:"size:100"`
    Email     string    `gorm:"uniqueIndex;size:255"`
    Age       int
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt gorm.DeletedAt `gorm:"index"`
}

// BeforeCreate Hook - 在创建记录前执行
func (u *User) BeforeCreate(tx *gorm.DB) error {
    fmt.Println("BeforeCreate: 准备创建用户", u.Name)
    if u.Age < 0 {
        return fmt.Errorf("年龄不能为负数")
    }
    u.CreatedAt = time.Now()
    return nil
}

// AfterCreate Hook - 在创建记录后执行
func (u *User) AfterCreate(tx *gorm.DB) error {
    fmt.Println("AfterCreate: 用户创建成功，ID:", u.ID)
    // 这里可以触发其他业务逻辑，如发送通知
    return nil
}

// BeforeUpdate Hook - 在更新记录前执行
func (u *User) BeforeUpdate(tx *gorm.DB) error {
    fmt.Println("BeforeUpdate: 准备更新用户", u.ID)
    u.UpdatedAt = time.Now()
    return nil
}

// BeforeDelete Hook - 在删除记录前执行（软删除）
func (u *User) BeforeDelete(tx *gorm.DB) error {
    fmt.Println("BeforeDelete: 准备删除用户", u.ID)
    // 可以添加删除前的验证逻辑
    return nil
}

2. 初始化 PostgreSQL 连接并使用 Hooks

func main() {
    // PostgreSQL 连接配置
    dsn := "host=localhost user=postgres password=secret dbname=gorm_hooks port=5432 sslmode=disable"
    
    db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
        Logger: logger.Default.LogMode(logger.Info),
    })
    if err != nil {
        panic("连接数据库失败: " + err.Error())
    }

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

    // 示例1: 创建记录（触发 BeforeCreate 和 AfterCreate）
    user := User{
        Name:  "张三",
        Email: "zhangsan@example.com",
        Age:   25,
    }
    
    result := db.Create(&user)
    if result.Error != nil {
        fmt.Println("创建失败:", result.Error)
    }

    // 示例2: 更新记录（触发 BeforeUpdate）
    db.Model(&user).Update("Age", 26)

    // 示例3: 删除记录（触发 BeforeDelete）
    db.Delete(&user)

    // 示例4: 使用事务确保 Hook 的一致性
    err = db.Transaction(func(tx *gorm.DB) error {
        newUser := User{
            Name:  "李四",
            Email: "lisi@example.com",
            Age:   30,
        }
        
        if err := tx.Create(&newUser).Error; err != nil {
            return err // 回滚事务
        }
        
        // 其他数据库操作...
        return nil // 提交事务
    })
    
    if err != nil {
        fmt.Println("事务执行失败:", err)
    }
}

3. 高级 Hook 用法示例

// 全局 Hook - 对所有模型生效
func registerGlobalHooks(db *gorm.DB) {
    // 在每次查询后执行
    db.Callback().Query().After("gorm:query").Register("log_queries", func(db *gorm.DB) {
        fmt.Printf("查询执行完成，影响行数: %d\n", db.RowsAffected)
    })

    // 在每次更新后执行
    db.Callback().Update().After("gorm:update").Register("audit_updates", func(db *gorm.DB) {
        fmt.Println("数据已更新，可在此处添加审计日志")
    })
}

// 条件 Hook 示例
func (u *User) BeforeSave(tx *gorm.DB) error {
    // BeforeSave 在 Create 和 Update 前都会执行
    if u.Age > 150 {
        return fmt.Errorf("年龄数据异常")
    }
    
    // 数据加密示例
    if u.Email != "" {
        // 这里可以添加邮箱加密逻辑
        fmt.Println("BeforeSave: 处理邮箱数据")
    }
    return nil
}

// 使用 Hook 实现数据验证
func (u *User) Validate() error {
    if u.Name == "" {
        return fmt.Errorf("用户名不能为空")
    }
    if u.Age < 18 {
        return fmt.Errorf("用户必须年满18岁")
    }
    return nil
}

// 在 BeforeCreate 中调用验证
func (u *User) BeforeCreate(tx *gorm.DB) error {
    if err := u.Validate(); err != nil {
        return err
    }
    return nil
}

4. 运行输出示例

BeforeCreate: 准备创建用户 张三
BeforeSave: 处理邮箱数据
AfterCreate: 用户创建成功，ID: 1
BeforeUpdate: 准备更新用户 1
BeforeSave: 处理邮箱数据
BeforeDelete: 准备删除用户 1

关键点说明：

1. 数据库无关性：GORM Hooks 通过回调机制工作，与 PostgreSQL、MySQL 或其他数据库无关
2. 支持的 Hook 方法：
   • BeforeSave, BeforeCreate, AfterCreate, BeforeUpdate, AfterUpdate, BeforeDelete, AfterDelete, AfterFind
3. 事务支持：Hooks 在事务中正常工作，如果 Hook 返回错误，事务会回滚
4. 性能考虑：复杂的 Hook 逻辑可能影响性能，建议将耗时操作异步化
5. PostgreSQL 特定功能：可以结合 PostgreSQL 的触发器、JSONB 等特性在 Hook 中实现更复杂的业务逻辑

这个示例展示了在 PostgreSQL 中使用 GORM Hooks 的完整流程，包括基本用法、事务集成和高级技巧。