lotus/paych/store.go

115 lines
2.1 KiB
Go
Raw Normal View History

2019-08-09 21:42:56 +00:00
package paych
import (
"fmt"
"strings"
2019-08-09 21:42:56 +00:00
"github.com/filecoin-project/go-lotus/chain/address"
"github.com/filecoin-project/go-lotus/chain/types"
"github.com/ipfs/go-datastore"
dsq "github.com/ipfs/go-datastore/query"
cbor "github.com/ipfs/go-ipld-cbor"
2019-08-09 21:42:56 +00:00
)
func init() {
cbor.RegisterCborType(ChannelInfo{})
}
type Store struct {
2019-08-09 21:42:56 +00:00
ds datastore.Batching
}
func NewStore(ds datastore.Batching) *Store {
return &Store{
2019-08-09 21:42:56 +00:00
ds: ds,
}
}
const (
DirInbound = 1
DirOutbound = 2
)
type ChannelInfo struct {
Channel address.Address
ControlAddr address.Address
Direction int
Vouchers []*types.SignedVoucher
}
func dskeyForChannel(addr address.Address) datastore.Key {
return datastore.NewKey("/paych/" + addr.String())
}
func (ps *Store) putChannelInfo(ci *ChannelInfo) error {
k := dskeyForChannel(ci.Channel)
b, err := cbor.DumpObject(ci)
if err != nil {
return err
}
return ps.ds.Put(k, b)
}
func (ps *Store) getChannelInfo(addr address.Address) (*ChannelInfo, error) {
k := dskeyForChannel(addr)
b, err := ps.ds.Get(k)
if err != nil {
return nil, err
}
var ci ChannelInfo
if err := cbor.DecodeInto(b, &ci); err != nil {
return nil, err
}
return &ci, nil
}
func (ps *Store) TrackChannel(ch *ChannelInfo) error {
_, err := ps.getChannelInfo(ch.Channel)
switch err {
default:
return err
case nil:
return fmt.Errorf("already tracking channel: %s", ch.Channel)
case datastore.ErrNotFound:
return ps.putChannelInfo(ch)
}
}
func (ps *Store) ListChannels() ([]address.Address, error) {
res, err := ps.ds.Query(dsq.Query{Prefix: "/paych/", KeysOnly: true})
if err != nil {
return nil, err
}
var out []address.Address
for {
res, ok := res.NextSync()
if !ok {
break
}
addr, err := address.NewFromString(strings.TrimPrefix(res.Key, "/paych/"))
if err != nil {
return nil, err
}
out = append(out, addr)
}
return out, nil
2019-08-09 21:42:56 +00:00
}
func (ps *Store) AddVoucher(ch address.Address, sv *types.SignedVoucher) error {
2019-08-09 21:42:56 +00:00
panic("nyi")
}
func (ps *Store) VouchersForPaych(addr address.Address) ([]*types.SignedVoucher, error) {
2019-08-09 21:42:56 +00:00
panic("nyi")
}