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

use namespaced datastore

Browse Source
This commit is contained in:
whyrusleeping 2019-08-12 12:05:57 -07:00
parent ae34757e2a
commit f8000066be
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
paych/store.go
View File

@ -2,11 +2,11 @@ package paych
import ( import (
"fmt" "fmt"
"strings"
"github.com/filecoin-project/go-lotus/chain/address" "github.com/filecoin-project/go-lotus/chain/address"
"github.com/filecoin-project/go-lotus/chain/types" "github.com/filecoin-project/go-lotus/chain/types"
"github.com/ipfs/go-datastore" "github.com/ipfs/go-datastore"
"github.com/ipfs/go-datastore/namespace"
dsq "github.com/ipfs/go-datastore/query" dsq "github.com/ipfs/go-datastore/query"
cbor "github.com/ipfs/go-ipld-cbor" cbor "github.com/ipfs/go-ipld-cbor"
@ -21,6 +21,7 @@ type Store struct {
} }
func NewStore(ds datastore.Batching) *Store { func NewStore(ds datastore.Batching) *Store {
ds = namespace.Wrap(ds, datastore.NewKey("/paych/"))
return &Store{ return &Store{
ds: ds, ds: ds,
} }
@ -39,7 +40,7 @@ type ChannelInfo struct {
} }
func dskeyForChannel(addr address.Address) datastore.Key { func dskeyForChannel(addr address.Address) datastore.Key {
return datastore.NewKey("/paych/" + addr.String()) return datastore.NewKey(addr.String())
} }
func (ps *Store) putChannelInfo(ci *ChannelInfo) error { func (ps *Store) putChannelInfo(ci *ChannelInfo) error {
@ -82,7 +83,7 @@ func (ps *Store) TrackChannel(ch *ChannelInfo) error {
} }
func (ps *Store) ListChannels() ([]address.Address, error) { func (ps *Store) ListChannels() ([]address.Address, error) {
res, err := ps.ds.Query(dsq.Query{Prefix: "/paych/", KeysOnly: true}) res, err := ps.ds.Query(dsq.Query{KeysOnly: true})
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -94,7 +95,7 @@ func (ps *Store) ListChannels() ([]address.Address, error) {
break break
} }
addr, err := address.NewFromString(strings.TrimPrefix(res.Key, "/paych/")) addr, err := address.NewFromString(res.Key)
if err != nil { if err != nil {
return nil, err return nil, err
} }