feat: ADR-040: Implement KV Store with decoupled storage and SMT

db/adapter.go

@@ -0,0 +1,23 @@
+package db
+
+type readerRWAdapter struct{ DBReader }
+
+// ReaderAsReadWriter returns a ReadWriter that forwards to a reader and errors if writes are
+// attempted. Can be used to pass a Reader when a ReadWriter is expected
+// but no writes will actually occur.
+func ReaderAsReadWriter(r DBReader) DBReadWriter {
+	return readerRWAdapter{r}
+}
+
+func (readerRWAdapter) Set([]byte, []byte) error {
+	return ErrReadOnly
+}
+
+func (readerRWAdapter) Delete([]byte) error {
+	return ErrReadOnly
+}
+
+func (rw readerRWAdapter) Commit() error {
+	rw.Discard()
+	return nil
+}

db/dbtest/util.go

@@ -0,0 +1,54 @@
+package dbtest
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	dbm "github.com/cosmos/cosmos-sdk/db"
+)
+
+func AssertNext(t *testing.T, itr dbm.Iterator, expected bool) {
+	t.Helper()
+	require.Equal(t, expected, itr.Next())
+}
+
+func AssertDomain(t *testing.T, itr dbm.Iterator, start, end []byte) {
+	t.Helper()
+	ds, de := itr.Domain()
+	assert.Equal(t, start, ds, "checkDomain domain start incorrect")
+	assert.Equal(t, end, de, "checkDomain domain end incorrect")
+}
+
+func AssertItem(t *testing.T, itr dbm.Iterator, key []byte, value []byte) {
+	t.Helper()
+	v := itr.Value()
+	k := itr.Key()
+	assert.Exactly(t, key, k)
+	assert.Exactly(t, value, v)
+}
+
+func AssertInvalid(t *testing.T, itr dbm.Iterator) {
+	t.Helper()
+	AssertNext(t, itr, false)
+	AssertKeyPanics(t, itr)
+	AssertValuePanics(t, itr)
+}
+
+func AssertKeyPanics(t *testing.T, itr dbm.Iterator) {
+	t.Helper()
+	assert.Panics(t, func() { itr.Key() }, "checkKeyPanics expected panic but didn't")
+}
+
+func AssertValue(t *testing.T, db dbm.DBReader, key []byte, valueWanted []byte) {
+	t.Helper()
+	valueGot, err := db.Get(key)
+	assert.NoError(t, err)
+	assert.Equal(t, valueWanted, valueGot)
+}
+
+func AssertValuePanics(t *testing.T, itr dbm.Iterator) {
+	t.Helper()
+	assert.Panics(t, func() { itr.Value() })
+}

db/go.mod

@@ -1,7 +1,7 @@
-go 1.17
-
 module github.com/cosmos/cosmos-sdk/db
 
+go 1.17
+
 require (
 	github.com/dgraph-io/badger/v3 v3.2103.1
 	github.com/google/btree v1.0.0

db/prefix/prefix.go

@@ -0,0 +1,159 @@
+package prefix
+
+import (
+	dbm "github.com/cosmos/cosmos-sdk/db"
+)
+
+// Prefix Reader/Writer lets you namespace multiple DBs within a single DB.
+type prefixR struct {
+	db     dbm.DBReader
+	prefix []byte
+}
+
+type prefixRW struct {
+	db     dbm.DBReadWriter
+	prefix []byte
+}
+
+var _ dbm.DBReader = (*prefixR)(nil)
+var _ dbm.DBReadWriter = (*prefixRW)(nil)
+
+func NewPrefixReader(db dbm.DBReader, prefix []byte) prefixR {
+	return prefixR{
+		prefix: prefix,
+		db:     db,
+	}
+}
+
+func NewPrefixReadWriter(db dbm.DBReadWriter, prefix []byte) prefixRW {
+	return prefixRW{
+		prefix: prefix,
+		db:     db,
+	}
+}
+
+func prefixed(prefix []byte, key []byte) []byte {
+	return append(prefix, key...)
+}
+
+// Get implements DBReader.
+func (pdb prefixR) Get(key []byte) ([]byte, error) {
+	if len(key) == 0 {
+		return nil, dbm.ErrKeyEmpty
+	}
+	return pdb.db.Get(prefixed(pdb.prefix, key))
+}
+
+// Has implements DBReader.
+func (pdb prefixR) Has(key []byte) (bool, error) {
+	if len(key) == 0 {
+		return false, dbm.ErrKeyEmpty
+	}
+	return pdb.db.Has(prefixed(pdb.prefix, key))
+}
+
+// Iterator implements DBReader.
+func (pdb prefixR) Iterator(start, end []byte) (dbm.Iterator, error) {
+	if (start != nil && len(start) == 0) || (end != nil && len(end) == 0) {
+		return nil, dbm.ErrKeyEmpty
+	}
+
+	var pend []byte
+	if end == nil {
+		pend = cpIncr(pdb.prefix)
+	} else {
+		pend = prefixed(pdb.prefix, end)
+	}
+	itr, err := pdb.db.Iterator(prefixed(pdb.prefix, start), pend)
+	if err != nil {
+		return nil, err
+	}
+	return newPrefixIterator(pdb.prefix, start, end, itr), nil
+}
+
+// ReverseIterator implements DBReader.
+func (pdb prefixR) ReverseIterator(start, end []byte) (dbm.Iterator, error) {
+	if (start != nil && len(start) == 0) || (end != nil && len(end) == 0) {
+		return nil, dbm.ErrKeyEmpty
+	}
+
+	var pend []byte
+	if end == nil {
+		pend = cpIncr(pdb.prefix)
+	} else {
+		pend = prefixed(pdb.prefix, end)
+	}
+	ritr, err := pdb.db.ReverseIterator(prefixed(pdb.prefix, start), pend)
+	if err != nil {
+		return nil, err
+	}
+	return newPrefixIterator(pdb.prefix, start, end, ritr), nil
+}
+
+// Discard implements DBReader.
+func (pdb prefixR) Discard() error { return pdb.db.Discard() }
+
+// Set implements DBReadWriter.
+func (pdb prefixRW) Set(key []byte, value []byte) error {
+	if len(key) == 0 {
+		return dbm.ErrKeyEmpty
+	}
+	return pdb.db.Set(prefixed(pdb.prefix, key), value)
+}
+
+// Delete implements DBReadWriter.
+func (pdb prefixRW) Delete(key []byte) error {
+	if len(key) == 0 {
+		return dbm.ErrKeyEmpty
+	}
+	return pdb.db.Delete(prefixed(pdb.prefix, key))
+}
+
+// Get implements DBReadWriter.
+func (pdb prefixRW) Get(key []byte) ([]byte, error) {
+	return NewPrefixReader(pdb.db, pdb.prefix).Get(key)
+}
+
+// Has implements DBReadWriter.
+func (pdb prefixRW) Has(key []byte) (bool, error) {
+	return NewPrefixReader(pdb.db, pdb.prefix).Has(key)
+}
+
+// Iterator implements DBReadWriter.
+func (pdb prefixRW) Iterator(start, end []byte) (dbm.Iterator, error) {
+	return NewPrefixReader(pdb.db, pdb.prefix).Iterator(start, end)
+}
+
+// ReverseIterator implements DBReadWriter.
+func (pdb prefixRW) ReverseIterator(start, end []byte) (dbm.Iterator, error) {
+	return NewPrefixReader(pdb.db, pdb.prefix).ReverseIterator(start, end)
+}
+
+// Close implements DBReadWriter.
+func (pdb prefixRW) Commit() error { return pdb.db.Commit() }
+
+// Discard implements DBReadWriter.
+func (pdb prefixRW) Discard() error { return pdb.db.Discard() }
+
+// Returns a slice of the same length (big endian), but incremented by one.
+// Returns nil on overflow (e.g. if bz bytes are all 0xFF)
+// CONTRACT: len(bz) > 0
+func cpIncr(bz []byte) (ret []byte) {
+	if len(bz) == 0 {
+		panic("cpIncr expects non-zero bz length")
+	}
+	ret = make([]byte, len(bz))
+	copy(ret, bz)
+	for i := len(bz) - 1; i >= 0; i-- {
+		if ret[i] < byte(0xFF) {
+			ret[i]++
+			return
+		}
+		ret[i] = byte(0x00)
+		if i == 0 {
+			// Overflow
+			return nil
+		}
+	}
+	return nil
+}

db/prefix/prefix_iterator.go

@@ -0,0 +1,114 @@
+package prefix
+
+import (
+	"bytes"
+	"fmt"
+
+	dbm "github.com/cosmos/cosmos-sdk/db"
+)
+
+// IteratePrefix is a convenience function for iterating over a key domain
+// restricted by prefix.
+func IteratePrefix(db dbm.DBReader, prefix []byte) (dbm.Iterator, error) {
+	var start, end []byte
+	if len(prefix) == 0 {
+		start = nil
+		end = nil
+	} else {
+		start = prefix
+		end = cpIncr(prefix)
+	}
+	itr, err := db.Iterator(start, end)
+	if err != nil {
+		return nil, err
+	}
+	return itr, nil
+}
+
+// Strips prefix while iterating from Iterator.
+type prefixDBIterator struct {
+	prefix []byte
+	start  []byte
+	end    []byte
+	source dbm.Iterator
+	err    error
+}
+
+var _ dbm.Iterator = (*prefixDBIterator)(nil)
+
+func newPrefixIterator(prefix, start, end []byte, source dbm.Iterator) *prefixDBIterator {
+	return &prefixDBIterator{
+		prefix: prefix,
+		start:  start,
+		end:    end,
+		source: source,
+	}
+}
+
+// Domain implements Iterator.
+func (itr *prefixDBIterator) Domain() (start []byte, end []byte) {
+	return itr.start, itr.end
+}
+
+func (itr *prefixDBIterator) valid() bool {
+	if itr.err != nil {
+		return false
+	}
+
+	key := itr.source.Key()
+	if len(key) < len(itr.prefix) || !bytes.Equal(key[:len(itr.prefix)], itr.prefix) {
+		itr.err = fmt.Errorf("received invalid key from backend: %x (expected prefix %x)",
+			key, itr.prefix)
+		return false
+	}
+
+	return true
+}
+
+// Next implements Iterator.
+func (itr *prefixDBIterator) Next() bool {
+	if !itr.source.Next() {
+		return false
+	}
+	if !bytes.HasPrefix(itr.source.Key(), itr.prefix) {
+		return false
+	}
+	// Empty keys are not allowed, so if a key exists in the database that exactly matches the
+	// prefix we need to skip it.
+	if bytes.Equal(itr.source.Key(), itr.prefix) {
+		return itr.Next()
+	}
+	return true
+}
+
+// Next implements Iterator.
+func (itr *prefixDBIterator) Key() []byte {
+	itr.assertIsValid()
+	key := itr.source.Key()
+	return key[len(itr.prefix):] // we have checked the key in Valid()
+}
+
+// Value implements Iterator.
+func (itr *prefixDBIterator) Value() []byte {
+	itr.assertIsValid()
+	return itr.source.Value()
+}
+
+// Error implements Iterator.
+func (itr *prefixDBIterator) Error() error {
+	if err := itr.source.Error(); err != nil {
+		return err
+	}
+	return itr.err
+}
+
+// Close implements Iterator.
+func (itr *prefixDBIterator) Close() error {
+	return itr.source.Close()
+}
+
+func (itr *prefixDBIterator) assertIsValid() {
+	if !itr.valid() {
+		panic("iterator is invalid")
+	}
+}