lotus/chain/deals/state_store.go

package deals

import (
	"github.com/ipfs/go-cid"
	"github.com/ipfs/go-datastore"
	cbor "github.com/ipfs/go-ipld-cbor"
	"golang.org/x/xerrors"
)

type StateStore struct {
	ds datastore.Datastore
}

func (st *StateStore) Begin(i cid.Cid, s interface{}) error {
	k := datastore.NewKey(i.String())
	has, err := st.ds.Has(k)
	if err != nil {
		return err
	}
	if has {
		// TODO: uncomment after deals work
		//return xerrors.Errorf("Already tracking state for %s", i)
	}
	b, err := cbor.DumpObject(s)
	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)
}

// When this gets used anywhere else, migrate to reflect

func (st *StateStore) MutateMiner(i cid.Cid, mutator func(MinerDeal) (MinerDeal, error)) error {
	return st.mutate(i, minerMutator(mutator))
}

func minerMutator(m func(MinerDeal) (MinerDeal, error)) func([]byte) ([]byte, error) {
	return func(in []byte) ([]byte, error) {
		var cur MinerDeal
		err := cbor.DecodeInto(in, &cur)
		if err != nil {
			return nil, err
		}

		mutated, err := m(cur)
		if err != nil {
			return nil, err
		}

		return cbor.DumpObject(mutated)
	}
}

func (st *StateStore) MutateClient(i cid.Cid, mutator func(ClientDeal) (ClientDeal, error)) error {
	return st.mutate(i, clientMutator(mutator))
}

func clientMutator(m func(ClientDeal) (ClientDeal, error)) func([]byte) ([]byte, error) {
	return func(in []byte) ([]byte, error) {
		var cur ClientDeal
		err := cbor.DecodeInto(in, &cur)
		if err != nil {
			return nil, err
		}

		mutated, err := m(cur)
		if err != nil {
			return nil, err
		}

		return cbor.DumpObject(mutated)
	}
}

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)
}