cerc-io/lotus
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

statestore: More generic keys

Browse Source
This commit is contained in:
Łukasz Magiera 2019-11-01 12:14:32 +01:00
parent 2343ebc5b8
commit 02c3be9099
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
lib/statestore/store.go
View File

@ -2,8 +2,8 @@ package statestore
import ( import (
"bytes" "bytes"
"fmt"
"github.com/filecoin-project/lotus/lib/cborrpc" "github.com/filecoin-project/lotus/lib/cborrpc"
"github.com/ipfs/go-cid"
"github.com/ipfs/go-datastore" "github.com/ipfs/go-datastore"
"github.com/ipfs/go-datastore/query" "github.com/ipfs/go-datastore/query"
"golang.org/x/xerrors" "golang.org/x/xerrors"
@ -18,8 +18,19 @@ func New(ds datastore.Datastore) *StateStore {
return &StateStore{ds: ds} return &StateStore{ds: ds}
} }
func (st *StateStore) Begin(i cid.Cid, state interface{}) error { func toKey(k interface{}) datastore.Key {
k := datastore.NewKey(i.String()) switch t := k.(type) {
case uint64:
return datastore.NewKey(fmt.Sprint(t))
case fmt.Stringer:
return datastore.NewKey(t.String())
default:
panic("unexpected key type")
}
}
func (st *StateStore) Begin(i interface{}, state interface{}) error {
k := toKey(i)
has, err := st.ds.Has(k) has, err := st.ds.Has(k)
if err != nil { if err != nil {
return err return err
@ -36,8 +47,8 @@ func (st *StateStore) Begin(i cid.Cid, state interface{}) error {
return st.ds.Put(k, b) return st.ds.Put(k, b)
} }
func (st *StateStore) End(i cid.Cid) error { func (st *StateStore) End(i interface{}) error {
k := datastore.NewKey(i.String()) k := toKey(i)
has, err := st.ds.Has(k) has, err := st.ds.Has(k)
if err != nil { if err != nil {
return err return err
@ -70,12 +81,12 @@ func cborMutator(mutator interface{}) func([]byte) ([]byte, error) {
} }
// mutator func(*T) error // mutator func(*T) error
func (st *StateStore) Mutate(i cid.Cid, mutator interface{}) error { func (st *StateStore) Mutate(i fmt.Stringer, mutator interface{}) error {
return st.mutate(i, cborMutator(mutator)) return st.mutate(i, cborMutator(mutator))
} }
func (st *StateStore) mutate(i cid.Cid, mutator func([]byte) ([]byte, error)) error { func (st *StateStore) mutate(i interface{}, mutator func([]byte) ([]byte, error)) error {
k := datastore.NewKey(i.String()) k := toKey(i)
has, err := st.ds.Has(k) has, err := st.ds.Has(k)
if err != nil { if err != nil {
return err return err