44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
|
package repository
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TransactionManager 事务管理器接口
|
||
|
|
type TransactionManager interface {
|
||
|
|
// RunInTransaction 在事务中执行函数
|
||
|
|
RunInTransaction(ctx context.Context, fn func(ctx context.Context) error) error
|
||
|
|
}
|
||
|
|
|
||
|
|
// gormTransactionManager GORM 事务管理器实现
|
||
|
|
type gormTransactionManager struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewTransactionManager 创建事务管理器
|
||
|
|
func NewTransactionManager(db *gorm.DB) TransactionManager {
|
||
|
|
return &gormTransactionManager{db: db}
|
||
|
|
}
|
||
|
|
|
||
|
|
// RunInTransaction 在事务中执行函数
|
||
|
|
func (m *gormTransactionManager) RunInTransaction(ctx context.Context, fn func(ctx context.Context) error) error {
|
||
|
|
return m.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||
|
|
// 将事务 TX 存入 context
|
||
|
|
ctxWithTx := context.WithValue(ctx, txKey{}, tx)
|
||
|
|
return fn(ctxWithTx)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// txKey 事务 context key
|
||
|
|
type txKey struct{}
|
||
|
|
|
||
|
|
// GetTxFromContext 从 context 获取事务 TX
|
||
|
|
func GetTxFromContext(ctx context.Context) *gorm.DB {
|
||
|
|
if tx, ok := ctx.Value(txKey{}).(*gorm.DB); ok {
|
||
|
|
return tx
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|