Support RollbackUnlessCommit

This commit is contained in:
Ming Deng
2021-03-21 23:52:55 +08:00
parent 88c522935f
commit c5bd3c3996
4 changed files with 86 additions and 23 deletions

View File

@@ -232,6 +232,14 @@ func (t *TxDB) Rollback() error {
return t.tx.Rollback()
}
func (t *TxDB) RollbackUnlessCommit() error {
err := t.tx.Rollback()
if err != sql.ErrTxDone {
return err
}
return nil
}
var _ dbQuerier = new(TxDB)
var _ txEnder = new(TxDB)