32 lines
723 B
Go
32 lines
723 B
Go
|
package stack
|
||
|
|
|
||
|
|
import "github.com/tendermint/basecoin/state"
|
||
|
|
|
||
|
|
type prefixStore struct {
|
||
|
|
prefix []byte
|
||
|
|
store state.KVStore
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ state.KVStore = prefixStore{}
|
||
|
|
|
||
|
|
func (p prefixStore) Set(key, value []byte) {
|
||
|
|
key = append(key, p.prefix...)
|
||
|
|
p.store.Set(key, value)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p prefixStore) Get(key []byte) (value []byte) {
|
||
|
|
key = append(key, p.prefix...)
|
||
|
|
return p.store.Get(key)
|
||
|
|
}
|
||
|
|
|
||
|
|
// stateSpace will unwrap any prefixStore and then add the prefix
|
||
|
|
func stateSpace(store state.KVStore, app string) state.KVStore {
|
||
|
|
// unwrap one-level if wrapped
|
||
|
|
if pstore, ok := store.(prefixStore); ok {
|
||
|
|
store = pstore.store
|
||
|
|
}
|
||
|
|
// wrap it with the prefix
|
||
|
|
prefix := append([]byte(app), byte(0))
|
||
|
|
return prefixStore{prefix, store}
|
||
|
|
}
|