75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package stmgr
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/ipfs/go-cid"
|
|
cbor "github.com/ipfs/go-ipld-cbor"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
|
|
"github.com/filecoin-project/lotus/chain/state"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
)
|
|
|
|
func (sm *StateManager) ParentStateTsk(ctx context.Context, tsk types.TipSetKey) (*state.StateTree, error) {
|
|
ts, err := sm.cs.GetTipSetFromKey(ctx, tsk)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("loading tipset %s: %w", tsk, err)
|
|
}
|
|
return sm.ParentState(ts)
|
|
}
|
|
|
|
func (sm *StateManager) ParentState(ts *types.TipSet) (*state.StateTree, error) {
|
|
cst := cbor.NewCborStore(sm.cs.StateBlockstore())
|
|
state, err := state.LoadStateTree(cst, sm.parentState(ts))
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("load state tree: %w", err)
|
|
}
|
|
|
|
return state, nil
|
|
}
|
|
|
|
func (sm *StateManager) parentState(ts *types.TipSet) cid.Cid {
|
|
if ts == nil {
|
|
ts = sm.cs.GetHeaviestTipSet()
|
|
}
|
|
|
|
return ts.ParentState()
|
|
}
|
|
|
|
func (sm *StateManager) StateTree(st cid.Cid) (*state.StateTree, error) {
|
|
cst := cbor.NewCborStore(sm.cs.StateBlockstore())
|
|
state, err := state.LoadStateTree(cst, st)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("load state tree: %w", err)
|
|
}
|
|
|
|
return state, nil
|
|
}
|
|
|
|
func (sm *StateManager) LoadActor(_ context.Context, addr address.Address, ts *types.TipSet) (*types.Actor, error) {
|
|
state, err := sm.ParentState(ts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return state.GetActor(addr)
|
|
}
|
|
|
|
func (sm *StateManager) LoadActorTsk(ctx context.Context, addr address.Address, tsk types.TipSetKey) (*types.Actor, error) {
|
|
state, err := sm.ParentStateTsk(ctx, tsk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return state.GetActor(addr)
|
|
}
|
|
|
|
func (sm *StateManager) LoadActorRaw(_ context.Context, addr address.Address, st cid.Cid) (*types.Actor, error) {
|
|
state, err := sm.StateTree(st)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return state.GetActor(addr)
|
|
}
|