2019-07-25 22:15:03 +00:00
|
|
|
package state
|
2019-07-05 14:29:17 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2019-07-30 00:46:56 +00:00
|
|
|
"fmt"
|
2019-07-08 12:51:45 +00:00
|
|
|
|
2019-07-05 14:29:17 +00:00
|
|
|
"github.com/ipfs/go-cid"
|
|
|
|
hamt "github.com/ipfs/go-hamt-ipld"
|
2020-01-08 19:10:57 +00:00
|
|
|
logging "github.com/ipfs/go-log/v2"
|
2020-01-22 19:53:06 +00:00
|
|
|
"go.opencensus.io/trace"
|
2019-09-10 02:05:24 +00:00
|
|
|
"golang.org/x/xerrors"
|
2019-07-12 10:23:05 +00:00
|
|
|
|
2019-12-19 20:13:17 +00:00
|
|
|
"github.com/filecoin-project/go-address"
|
2019-10-18 04:47:41 +00:00
|
|
|
"github.com/filecoin-project/lotus/chain/actors"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
2019-07-05 14:29:17 +00:00
|
|
|
)
|
|
|
|
|
2019-07-25 22:15:03 +00:00
|
|
|
var log = logging.Logger("statetree")
|
|
|
|
|
2019-07-05 14:29:17 +00:00
|
|
|
type StateTree struct {
|
|
|
|
root *hamt.Node
|
2019-07-25 22:15:03 +00:00
|
|
|
Store *hamt.CborIpldStore
|
2019-07-05 14:29:17 +00:00
|
|
|
|
2019-07-12 10:23:05 +00:00
|
|
|
actorcache map[address.Address]*types.Actor
|
2019-07-05 14:29:17 +00:00
|
|
|
snapshot cid.Cid
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewStateTree(cst *hamt.CborIpldStore) (*StateTree, error) {
|
|
|
|
return &StateTree{
|
|
|
|
root: hamt.NewNode(cst),
|
2019-07-25 22:15:03 +00:00
|
|
|
Store: cst,
|
2019-07-12 10:23:05 +00:00
|
|
|
actorcache: make(map[address.Address]*types.Actor),
|
2019-07-05 14:29:17 +00:00
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func LoadStateTree(cst *hamt.CborIpldStore, c cid.Cid) (*StateTree, error) {
|
|
|
|
nd, err := hamt.LoadNode(context.Background(), cst, c)
|
|
|
|
if err != nil {
|
2019-07-30 16:04:36 +00:00
|
|
|
log.Errorf("loading hamt node %s failed: %s", c, err)
|
2019-07-05 14:29:17 +00:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return &StateTree{
|
|
|
|
root: nd,
|
2019-07-25 22:15:03 +00:00
|
|
|
Store: cst,
|
2019-07-12 10:23:05 +00:00
|
|
|
actorcache: make(map[address.Address]*types.Actor),
|
2019-07-05 14:29:17 +00:00
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
2019-07-12 10:23:05 +00:00
|
|
|
func (st *StateTree) SetActor(addr address.Address, act *types.Actor) error {
|
2019-11-15 16:39:43 +00:00
|
|
|
iaddr, err := st.LookupID(addr)
|
|
|
|
if err != nil {
|
|
|
|
return xerrors.Errorf("ID lookup failed: %w", err)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
2019-11-15 16:39:43 +00:00
|
|
|
addr = iaddr
|
2019-07-05 14:29:17 +00:00
|
|
|
|
|
|
|
cact, ok := st.actorcache[addr]
|
|
|
|
if ok {
|
|
|
|
if act == cact {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-09-23 09:45:22 +00:00
|
|
|
st.actorcache[addr] = act
|
|
|
|
|
2019-07-05 14:29:17 +00:00
|
|
|
return st.root.Set(context.TODO(), string(addr.Bytes()), act)
|
|
|
|
}
|
|
|
|
|
2019-11-15 16:39:43 +00:00
|
|
|
func (st *StateTree) LookupID(addr address.Address) (address.Address, error) {
|
|
|
|
if addr.Protocol() == address.ID {
|
|
|
|
return addr, nil
|
|
|
|
}
|
|
|
|
|
2019-10-21 18:12:11 +00:00
|
|
|
act, err := st.GetActor(actors.InitAddress)
|
2019-07-05 14:29:17 +00:00
|
|
|
if err != nil {
|
2019-09-10 02:05:24 +00:00
|
|
|
return address.Undef, xerrors.Errorf("getting init actor: %w", err)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
|
|
|
|
2019-07-12 23:52:25 +00:00
|
|
|
var ias actors.InitActorState
|
2019-07-25 22:15:03 +00:00
|
|
|
if err := st.Store.Get(context.TODO(), act.Head, &ias); err != nil {
|
2019-09-10 02:05:24 +00:00
|
|
|
return address.Undef, xerrors.Errorf("loading init actor state: %w", err)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
|
|
|
|
2019-07-25 22:15:03 +00:00
|
|
|
return ias.Lookup(st.Store, addr)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
|
|
|
|
2019-07-12 10:23:05 +00:00
|
|
|
func (st *StateTree) GetActor(addr address.Address) (*types.Actor, error) {
|
2019-07-30 00:46:56 +00:00
|
|
|
if addr == address.Undef {
|
|
|
|
return nil, fmt.Errorf("GetActor called on undefined address")
|
|
|
|
}
|
|
|
|
|
2019-11-15 16:39:43 +00:00
|
|
|
iaddr, err := st.LookupID(addr)
|
|
|
|
if err != nil {
|
|
|
|
if xerrors.Is(err, hamt.ErrNotFound) {
|
|
|
|
return nil, xerrors.Errorf("resolution lookup failed (%s): %w", addr, types.ErrActorNotFound)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
2019-11-15 16:39:43 +00:00
|
|
|
return nil, xerrors.Errorf("address resolution: %w", err)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
2019-11-15 16:39:43 +00:00
|
|
|
addr = iaddr
|
2019-07-05 14:29:17 +00:00
|
|
|
|
|
|
|
cact, ok := st.actorcache[addr]
|
|
|
|
if ok {
|
|
|
|
return cact, nil
|
|
|
|
}
|
|
|
|
|
2019-09-17 01:56:37 +00:00
|
|
|
var act types.Actor
|
2019-11-15 16:39:43 +00:00
|
|
|
err = st.root.Find(context.TODO(), string(addr.Bytes()), &act)
|
2019-07-05 14:29:17 +00:00
|
|
|
if err != nil {
|
|
|
|
if err == hamt.ErrNotFound {
|
2019-07-22 18:17:42 +00:00
|
|
|
return nil, types.ErrActorNotFound
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
2019-09-30 23:55:35 +00:00
|
|
|
return nil, xerrors.Errorf("hamt find failed: %w", err)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
st.actorcache[addr] = &act
|
|
|
|
|
|
|
|
return &act, nil
|
|
|
|
}
|
|
|
|
|
2020-01-22 19:53:06 +00:00
|
|
|
func (st *StateTree) Flush(ctx context.Context) (cid.Cid, error) {
|
|
|
|
ctx, span := trace.StartSpan(ctx, "stateTree.Flush")
|
|
|
|
defer span.End()
|
|
|
|
|
2019-07-05 14:29:17 +00:00
|
|
|
for addr, act := range st.actorcache {
|
2020-01-22 19:53:06 +00:00
|
|
|
if err := st.root.Set(ctx, string(addr.Bytes()), act); err != nil {
|
2019-07-05 14:29:17 +00:00
|
|
|
return cid.Undef, err
|
|
|
|
}
|
|
|
|
}
|
2019-07-12 10:23:05 +00:00
|
|
|
st.actorcache = make(map[address.Address]*types.Actor)
|
2019-07-05 14:29:17 +00:00
|
|
|
|
2020-01-22 19:53:06 +00:00
|
|
|
if err := st.root.Flush(ctx); err != nil {
|
2019-07-05 14:29:17 +00:00
|
|
|
return cid.Undef, err
|
|
|
|
}
|
|
|
|
|
2020-01-22 19:53:06 +00:00
|
|
|
return st.Store.Put(ctx, st.root)
|
2019-07-05 14:29:17 +00:00
|
|
|
}
|
|
|
|
|
2020-01-22 19:53:06 +00:00
|
|
|
func (st *StateTree) Snapshot(ctx context.Context) error {
|
|
|
|
ctx, span := trace.StartSpan(ctx, "stateTree.SnapShot")
|
|
|
|
defer span.End()
|
|
|
|
|
|
|
|
ss, err := st.Flush(ctx)
|
2019-07-05 14:29:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
st.snapshot = ss
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-07-12 10:23:05 +00:00
|
|
|
func (st *StateTree) RegisterNewAddress(addr address.Address, act *types.Actor) (address.Address, error) {
|
2019-07-05 14:29:17 +00:00
|
|
|
var out address.Address
|
2019-10-21 18:12:11 +00:00
|
|
|
err := st.MutateActor(actors.InitAddress, func(initact *types.Actor) error {
|
2019-07-12 23:52:25 +00:00
|
|
|
var ias actors.InitActorState
|
2019-07-25 22:15:03 +00:00
|
|
|
if err := st.Store.Get(context.TODO(), initact.Head, &ias); err != nil {
|
2019-07-05 14:29:17 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2019-07-25 22:15:03 +00:00
|
|
|
oaddr, err := ias.AddActor(st.Store, addr)
|
2019-07-05 14:29:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
out = oaddr
|
|
|
|
|
2019-07-25 22:15:03 +00:00
|
|
|
ncid, err := st.Store.Put(context.TODO(), &ias)
|
2019-07-05 14:29:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
initact.Head = ncid
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return address.Undef, err
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := st.SetActor(out, act); err != nil {
|
|
|
|
return address.Undef, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return out, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (st *StateTree) Revert() error {
|
2019-07-25 22:15:03 +00:00
|
|
|
nd, err := hamt.LoadNode(context.Background(), st.Store, st.snapshot)
|
2019-07-05 14:29:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
st.root = nd
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-07-12 10:23:05 +00:00
|
|
|
func (st *StateTree) MutateActor(addr address.Address, f func(*types.Actor) error) error {
|
2019-07-05 14:29:17 +00:00
|
|
|
act, err := st.GetActor(addr)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := f(act); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return st.SetActor(addr, act)
|
|
|
|
}
|