Golang中的工作单元模式应用 管理数据库事务边界的实践方案

type UnitOfWork interface {
    Begin() error
    Commit() error
    Rollback() error
}

type DefaultUnitOfWork struct {
    db *sql.DB
    tx *sql.Tx
}

func NewUnitOfWork(db *sql.DB) *DefaultUnitOfWork {
    return &DefaultUnitOfWork{db: db}
}

func (uow *DefaultUnitOfWork) Begin() error {
    tx, err := uow.db.Begin()
    if err != nil {
        return err
    }
    uow.tx = tx
    return nil
}

func (uow *DefaultUnitOfWork) Commit() error {
    if uow.tx == nil {
        return errors.New("transaction not started")
    }
    err := uow.tx.Commit()
    if err != nil {
        return err
    }
    uow.tx = nil // 清空事务
    return nil
}

func (uow *DefaultUnitOfWork) Rollback() error {
    if uow.tx == nil {
        return errors.New("transaction not started")
    }
    err := uow.tx.Rollback()
    if err != nil {
        return err
    }
    uow.tx = nil // 清空事务
    return nil
}

func TransferFunds(uow UnitOfWork, fromAccountID, toAccountID int, amount float64) error {
    err := uow.Begin()
    if err != nil {
        return err
    }

    // 执行转账操作（这里只是示例，需要替换成实际的数据库操作）
    _, err = uow.(*DefaultUnitOfWork).tx.Exec("UPDATE accounts SET balance = balance - ? WHERE id = ?", amount, fromAccountID)
    if err != nil {
        uow.Rollback()
        return err
    }

    _, err = uow.(*DefaultUnitOfWork).tx.Exec("UPDATE accounts SET balance = balance + ? WHERE id = ?", amount, toAccountID)
    if err != nil {
        uow.Rollback()
        return err
    }

    err = uow.Commit()
    if err != nil {
        return err
    }

    return nil
}