143 lines
2.6 KiB
Go
143 lines
2.6 KiB
Go
package deals
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/filecoin-project/lotus/lib/cborrpc"
|
|
"github.com/ipfs/go-cid"
|
|
"github.com/ipfs/go-datastore"
|
|
"github.com/ipfs/go-datastore/query"
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
type StateStore struct {
|
|
ds datastore.Datastore
|
|
}
|
|
|
|
func (st *StateStore) Begin(i cid.Cid, state interface{}) error {
|
|
k := datastore.NewKey(i.String())
|
|
has, err := st.ds.Has(k)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if has {
|
|
return xerrors.Errorf("Already tracking state for %s", i)
|
|
}
|
|
|
|
b, err := cborrpc.Dump(state)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return st.ds.Put(k, b)
|
|
}
|
|
|
|
func (st *StateStore) End(i cid.Cid) error {
|
|
k := datastore.NewKey(i.String())
|
|
has, err := st.ds.Has(k)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !has {
|
|
return xerrors.Errorf("No state for %s", i)
|
|
}
|
|
return st.ds.Delete(k)
|
|
}
|
|
|
|
func (st *StateStore) mutate(i cid.Cid, mutator func([]byte) ([]byte, error)) error {
|
|
k := datastore.NewKey(i.String())
|
|
has, err := st.ds.Has(k)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !has {
|
|
return xerrors.Errorf("No state for %s", i)
|
|
}
|
|
|
|
cur, err := st.ds.Get(k)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mutated, err := mutator(cur)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return st.ds.Put(k, mutated)
|
|
}
|
|
|
|
type MinerStateStore struct {
|
|
StateStore
|
|
}
|
|
|
|
func (st *MinerStateStore) MutateMiner(i cid.Cid, mutator func(*MinerDeal) error) error {
|
|
return st.mutate(i, minerMutator(mutator))
|
|
}
|
|
|
|
func minerMutator(m func(*MinerDeal) error) func([]byte) ([]byte, error) {
|
|
return func(in []byte) ([]byte, error) {
|
|
var deal MinerDeal
|
|
err := cborrpc.ReadCborRPC(bytes.NewReader(in), &deal)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := m(&deal); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return cborrpc.Dump(deal)
|
|
}
|
|
}
|
|
|
|
type ClientStateStore struct {
|
|
StateStore
|
|
}
|
|
|
|
func (st *ClientStateStore) MutateClient(i cid.Cid, mutator func(*ClientDeal) error) error {
|
|
return st.mutate(i, clientMutator(mutator))
|
|
}
|
|
|
|
func clientMutator(m func(*ClientDeal) error) func([]byte) ([]byte, error) {
|
|
return func(in []byte) ([]byte, error) {
|
|
var deal ClientDeal
|
|
err := cborrpc.ReadCborRPC(bytes.NewReader(in), &deal)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := m(&deal); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return cborrpc.Dump(deal)
|
|
}
|
|
}
|
|
|
|
func (st *ClientStateStore) ListClient() ([]ClientDeal, error) {
|
|
var out []ClientDeal
|
|
|
|
res, err := st.ds.Query(query.Query{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer res.Close()
|
|
|
|
for {
|
|
res, ok := res.NextSync()
|
|
if !ok {
|
|
break
|
|
}
|
|
|
|
var deal ClientDeal
|
|
err := cborrpc.ReadCborRPC(bytes.NewReader(res.Value), &deal)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
out = append(out, deal)
|
|
}
|
|
|
|
return out, nil
|
|
}
|