fix(tfstated): rollack transaction in case of panic
This commit is contained in:
parent
f649f7bbbf
commit
b3980358cf
1 changed files with 13 additions and 8 deletions
|
@ -108,16 +108,21 @@ func (db *DB) SetVersionsHistoryLimit(n int) {
|
||||||
func (db *DB) WithTransaction(f func(tx *sql.Tx) error) error {
|
func (db *DB) WithTransaction(f func(tx *sql.Tx) error) error {
|
||||||
tx, err := db.writeDB.Begin()
|
tx, err := db.writeDB.Begin()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to begin transaction: %w", err)
|
||||||
}
|
|
||||||
err = f(tx)
|
|
||||||
if err == nil {
|
|
||||||
err = tx.Commit()
|
|
||||||
}
|
}
|
||||||
|
defer func() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err2 := tx.Rollback(); err2 != nil {
|
if err2 := tx.Rollback(); err2 != nil {
|
||||||
panic(fmt.Sprintf("failed to rollback transaction: %+v. Reason for rollback: %+v", err2, err))
|
panic(fmt.Sprintf("failed to rollback transaction: %+v. Reason for rollback: %+v", err2, err))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}()
|
||||||
|
if err = f(tx); err != nil {
|
||||||
|
return fmt.Errorf("failed to execute function inside transaction: %w", err)
|
||||||
|
} else {
|
||||||
|
if err = tx.Commit(); err != nil {
|
||||||
|
err = fmt.Errorf("failed to commit transaction: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue