LBRYFoundation/lbcwallet
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/lbcwallet.git synced 2025-08-23 17:47:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

walletdb: ensure transactions are rolled back on panic

Browse source
This commit is contained in:
Olaoluwa Osuntokun 2020-01-15 04:52:34 -08:00 • committed by Conner Fromknecht
parent e7e46cd6f9
commit da2fe0e3b0
No known key found for this signature in database
GPG key ID: E7D737B67FA592C7
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
walletdb/interface.go
View file

@ -210,11 +210,20 @@ func View(db DB, f func(tx ReadTx) error) error {
if err != nil { if err != nil {
return err return err
} }
// Make sure the transaction rolls back in the event of a panic.
defer func() {
if tx != nil {
tx.Rollback()
}
}()
err = f(tx) err = f(tx)
rollbackErr := tx.Rollback() rollbackErr := tx.Rollback()
if err != nil { if err != nil {
return err return err
} }
if rollbackErr != nil { if rollbackErr != nil {
return rollbackErr return rollbackErr
} }
@ -232,6 +241,14 @@ func Update(db DB, f func(tx ReadWriteTx) error) error {
if err != nil { if err != nil {
return err return err
} }
// Make sure the transaction rolls back in the event of a panic.
defer func() {
if tx != nil {
tx.Rollback()
}
}()
err = f(tx) err = f(tx)
if err != nil { if err != nil {
// Want to return the original error, not a rollback error if // Want to return the original error, not a rollback error if
@ -239,6 +256,7 @@ func Update(db DB, f func(tx ReadWriteTx) error) error {
_ = tx.Rollback() _ = tx.Rollback()
return err return err
} }
return tx.Commit() return tx.Commit()
} }