## Description Closes: #11378 --- ### Author Checklist *All items are required. Please add a note to the item if the item is not applicable and please add links to any relevant follow up issues.* I have... - [ ] included the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title - [ ] added `!` to the type prefix if API or client breaking change - [ ] targeted the correct branch (see [PR Targeting](https://github.com/cosmos/cosmos-sdk/blob/master/CONTRIBUTING.md#pr-targeting)) - [ ] provided a link to the relevant issue or specification - [ ] followed the guidelines for [building modules](https://github.com/cosmos/cosmos-sdk/blob/master/docs/building-modules) - [ ] included the necessary unit and integration [tests](https://github.com/cosmos/cosmos-sdk/blob/master/CONTRIBUTING.md#testing) - [ ] added a changelog entry to `CHANGELOG.md` - [ ] included comments for [documenting Go code](https://blog.golang.org/godoc) - [ ] updated the relevant documentation or specification - [ ] reviewed "Files changed" and left comments if necessary - [ ] confirmed all CI checks have passed ### Reviewers Checklist *All items are required. Please add a note if the item is not applicable and please add your handle next to the items reviewed if you only reviewed selected items.* I have... - [ ] confirmed the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title - [ ] confirmed `!` in the type prefix if API or client breaking change - [ ] confirmed all author checklist items have been addressed - [ ] reviewed state machine logic - [ ] reviewed API design and naming - [ ] reviewed documentation is accurate - [ ] reviewed tests and test coverage - [ ] manually tested (if applicable)
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package multi
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
|
|
dbm "github.com/cosmos/cosmos-sdk/db"
|
|
)
|
|
|
|
type dbDeleteVersionFails struct{ dbm.DBConnection }
|
|
type dbRWCommitFails struct{ dbm.DBConnection }
|
|
type dbRWCrudFails struct{ dbm.DBConnection }
|
|
type dbSaveVersionFails struct{ dbm.DBConnection }
|
|
type dbRevertFails struct {
|
|
dbm.DBConnection
|
|
// order of calls to fail on (eg. [1, 0] => first call fails; second succeeds)
|
|
failOn []bool
|
|
}
|
|
type dbVersionsIs struct {
|
|
dbm.DBConnection
|
|
vset dbm.VersionSet
|
|
}
|
|
type dbVersionsFails struct{ dbm.DBConnection }
|
|
type rwCommitFails struct{ dbm.DBReadWriter }
|
|
type rwCrudFails struct {
|
|
dbm.DBReadWriter
|
|
onKey []byte
|
|
}
|
|
|
|
func (dbVersionsFails) Versions() (dbm.VersionSet, error) { return nil, errors.New("dbVersionsFails") }
|
|
func (db dbVersionsIs) Versions() (dbm.VersionSet, error) { return db.vset, nil }
|
|
func (db dbRWCrudFails) ReadWriter() dbm.DBReadWriter {
|
|
return rwCrudFails{db.DBConnection.ReadWriter(), nil}
|
|
}
|
|
func (dbSaveVersionFails) SaveVersion(uint64) error { return errors.New("dbSaveVersionFails") }
|
|
func (db dbRevertFails) Revert() error {
|
|
fail := false
|
|
if len(db.failOn) > 0 {
|
|
fail, db.failOn = db.failOn[0], db.failOn[1:]
|
|
}
|
|
if fail {
|
|
return errors.New("dbRevertFails")
|
|
}
|
|
return db.DBConnection.Revert()
|
|
}
|
|
func (dbDeleteVersionFails) DeleteVersion(uint64) error { return errors.New("dbDeleteVersionFails") }
|
|
func (tx rwCommitFails) Commit() error {
|
|
tx.Discard()
|
|
return errors.New("rwCommitFails")
|
|
}
|
|
func (db dbRWCommitFails) ReadWriter() dbm.DBReadWriter {
|
|
return rwCommitFails{db.DBConnection.ReadWriter()}
|
|
}
|
|
|
|
func (rw rwCrudFails) Get(k []byte) ([]byte, error) {
|
|
if rw.onKey == nil || bytes.Equal(rw.onKey, k) {
|
|
return nil, errors.New("rwCrudFails.Get")
|
|
}
|
|
return rw.DBReadWriter.Get(k)
|
|
}
|
|
func (rw rwCrudFails) Has(k []byte) (bool, error) {
|
|
if rw.onKey == nil || bytes.Equal(rw.onKey, k) {
|
|
return false, errors.New("rwCrudFails.Has")
|
|
}
|
|
return rw.DBReadWriter.Has(k)
|
|
}
|
|
func (rw rwCrudFails) Set(k []byte, v []byte) error {
|
|
if rw.onKey == nil || bytes.Equal(rw.onKey, k) {
|
|
return errors.New("rwCrudFails.Set")
|
|
}
|
|
return rw.DBReadWriter.Set(k, v)
|
|
}
|
|
func (rw rwCrudFails) Delete(k []byte) error {
|
|
if rw.onKey == nil || bytes.Equal(rw.onKey, k) {
|
|
return errors.New("rwCrudFails.Delete")
|
|
}
|
|
return rw.DBReadWriter.Delete(k)
|
|
}
|