如何在GORM中拦截rollback？需在事务回滚后执行操作

针对你需要在事务回滚完成后执行自定义逻辑的需求，这里提供几种实用的实现方式：

1. 注册全局回滚回调 #

GORM允许直接给Rollback操作注册自定义回调，这个回调会在所有回滚操作（包括主动调用和自动触发）完成后执行。

代码示例 #

import (
    "gorm.io/gorm"
    "log"
)

func init() {
    // 注册回滚后的处理回调，确保在默认回滚逻辑之后执行
    gorm.DefaultCallback.Rollback().After("gorm:rollback").Register("custom:after_rollback", func(db *gorm.DB) {
        // 仅在回滚由错误触发时执行逻辑（可选）
        if db.Error != nil {
            // 这里编写你的回滚后处理逻辑
            // 比如记录错误日志、清理临时文件、发送告警等
            log.Printf("事务已回滚，触发原因：%v", db.Error)
            
            // 额外判断是否是创建操作失败（可选）
            if db.Statement.Operation == "create" {
                log.Println("回滚由创建操作失败触发")
                // 针对创建失败的专属处理
            }
        }
    })
}

2. 封装事务函数实现精细化控制 #

如果只需要针对特定操作（比如创建）的回滚做处理，可以封装自己的事务执行函数，在手动触发回滚后直接调用处理逻辑。

代码示例 #

import "gorm.io/gorm"

// 封装事务执行逻辑，自动处理回滚后的操作
func ExecuteWithTx(db *gorm.DB, txFunc func(tx *gorm.DB) error) error {
    tx := db.Begin()
    if tx.Error != nil {
        return tx.Error
    }

    // 处理panic导致的回滚
    defer func() {
        if r := recover(); r != nil {
            tx.Rollback()
            handlePostRollback()
        }
    }()

    // 执行传入的事务逻辑
    if err := txFunc(tx); err != nil {
        // 触发回滚
        if rollbackErr := tx.Rollback().Error; rollbackErr != nil {
            return rollbackErr
        }
        // 回滚完成后执行自定义逻辑
        handlePostRollback()
        return err
    }

    return tx.Commit().Error
}

// 回滚后的自定义处理函数
func handlePostRollback() {
    // 这里编写你的具体处理逻辑
    // 比如清理缓存、更新状态通知等
}

使用方式 #

把创建操作放到封装的事务函数中：

err := ExecuteWithTx(db, func(tx *gorm.DB) error {
    // 执行创建操作
    user := User{Name: "example"}
    if err := tx.Create(&user).Error; err != nil {
        return err // 触发回滚及后续处理
    }
    return nil
})

注意事项 #

• 如果你使用的是GORM v1版本，回调注册的语法略有差异，需要调整为db.Callback().Rollback().After("gorm:rollback").Register(...)的形式。
• 全局回调会作用于所有回滚操作，若需要区分不同操作类型，可以通过db.Statement.Operation字段判断（比如create、update等）。

内容的提问来源于stack exchange，提问作者noob fish