Skip to content

Commit

Permalink
Introduce v1 actors
Browse files Browse the repository at this point in the history
  • Loading branch information
arajasek committed Sep 21, 2020
1 parent 286fe04 commit 2b22715
Show file tree
Hide file tree
Showing 21 changed files with 1,092 additions and 1 deletion.
8 changes: 8 additions & 0 deletions chain/actors/builtin/account/account.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import (
"github.com/filecoin-project/go-address"
"github.com/filecoin-project/go-state-types/cbor"
builtin0 "github.com/filecoin-project/specs-actors/actors/builtin"
builtin1 "github.com/filecoin-project/specs-actors/v2/actors/builtin"

"github.com/filecoin-project/lotus/chain/actors/adt"
"github.com/filecoin-project/lotus/chain/types"
Expand All @@ -20,6 +21,13 @@ func Load(store adt.Store, act *types.Actor) (State, error) {
return nil, err
}
return &out, nil
case builtin1.AccountActorCodeID:
out := state1{store: store}
err := store.Get(store.Context(), act.Head, &out)
if err != nil {
return nil, err
}
return &out, nil
}
return nil, xerrors.Errorf("unknown actor code %s", act.Code)
}
Expand Down
18 changes: 18 additions & 0 deletions chain/actors/builtin/account/v1.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
package account

import (
"github.com/filecoin-project/go-address"
"github.com/filecoin-project/lotus/chain/actors/adt"
"github.com/filecoin-project/specs-actors/v2/actors/builtin/account"
)

var _ State = (*state1)(nil)

type state1 struct {
account.State
store adt.Store
}

func (s *state1) PubkeyAddress() (address.Address, error) {
return s.Address, nil
}
5 changes: 5 additions & 0 deletions chain/actors/builtin/builtin.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ import (
"fmt"

smoothing0 "github.com/filecoin-project/specs-actors/actors/util/smoothing"
smoothing1 "github.com/filecoin-project/specs-actors/v2/actors/util/smoothing"

"github.com/filecoin-project/go-state-types/network"
)
Expand All @@ -30,3 +31,7 @@ type FilterEstimate = smoothing0.FilterEstimate
func FromV0FilterEstimate(v0 smoothing0.FilterEstimate) FilterEstimate {
return (FilterEstimate)(v0)
}

func FromV1FilterEstimate(v1 smoothing1.FilterEstimate) FilterEstimate {
return (FilterEstimate)(v1)
}
8 changes: 8 additions & 0 deletions chain/actors/builtin/init/init.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import (
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/go-state-types/cbor"
builtin0 "github.com/filecoin-project/specs-actors/actors/builtin"
builtin1 "github.com/filecoin-project/specs-actors/v2/actors/builtin"

"github.com/filecoin-project/lotus/chain/actors/adt"
"github.com/filecoin-project/lotus/chain/types"
Expand All @@ -24,6 +25,13 @@ func Load(store adt.Store, act *types.Actor) (State, error) {
return nil, err
}
return &out, nil
case builtin1.InitActorCodeID:
out := state1{store: store}
err := store.Get(store.Context(), act.Head, &out)
if err != nil {
return nil, err
}
return &out, nil
}
return nil, xerrors.Errorf("unknown actor code %s", act.Code)
}
Expand Down
47 changes: 47 additions & 0 deletions chain/actors/builtin/init/v1.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
package init

import (
"github.com/filecoin-project/go-address"
"github.com/filecoin-project/go-state-types/abi"
cbg "github.com/whyrusleeping/cbor-gen"

"github.com/filecoin-project/lotus/chain/actors/adt"
"github.com/filecoin-project/lotus/node/modules/dtypes"

init_ "github.com/filecoin-project/specs-actors/v2/actors/builtin/init"
adt1 "github.com/filecoin-project/specs-actors/v2/actors/util/adt"
)

var _ State = (*state1)(nil)

type state1 struct {
init_.State
store adt.Store
}

func (s *state1) ResolveAddress(address address.Address) (address.Address, bool, error) {
return s.State.ResolveAddress(s.store, address)
}

func (s *state1) MapAddressToNewID(address address.Address) (address.Address, error) {
return s.State.MapAddressToNewID(s.store, address)
}

func (s *state1) ForEachActor(cb func(id abi.ActorID, address address.Address) error) error {
addrs, err := adt1.AsMap(s.store, s.State.AddressMap)
if err != nil {
return err
}
var actorID cbg.CborInt
return addrs.ForEach(&actorID, func(key string) error {
addr, err := address.NewFromBytes([]byte(key))
if err != nil {
return err
}
return cb(abi.ActorID(actorID), addr)
})
}

func (s *state1) NetworkName() (dtypes.NetworkName, error) {
return dtypes.NetworkName(s.State.NetworkName), nil
}
8 changes: 8 additions & 0 deletions chain/actors/builtin/market/market.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import (
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/go-state-types/cbor"
builtin0 "github.com/filecoin-project/specs-actors/actors/builtin"
builtin1 "github.com/filecoin-project/specs-actors/v2/actors/builtin"
"github.com/ipfs/go-cid"
cbg "github.com/whyrusleeping/cbor-gen"

Expand All @@ -25,6 +26,13 @@ func Load(store adt.Store, act *types.Actor) (st State, err error) {
return nil, err
}
return &out, nil
case builtin1.StorageMarketActorCodeID:
out := state1{store: store}
err := store.Get(store.Context(), act.Head, &out)
if err != nil {
return nil, err
}
return &out, nil
}
return nil, xerrors.Errorf("unknown actor code %s", act.Code)
}
Expand Down
186 changes: 186 additions & 0 deletions chain/actors/builtin/market/v1.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,186 @@
package market

import (
"bytes"

"github.com/filecoin-project/go-address"
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/lotus/chain/actors/adt"
"github.com/filecoin-project/lotus/chain/types"
"github.com/filecoin-project/specs-actors/v2/actors/builtin/market"
adt1 "github.com/filecoin-project/specs-actors/v2/actors/util/adt"
cbg "github.com/whyrusleeping/cbor-gen"
)

var _ State = (*state1)(nil)

type state1 struct {
market.State
store adt.Store
}

func (s *state1) TotalLocked() (abi.TokenAmount, error) {
fml := types.BigAdd(s.TotalClientLockedCollateral, s.TotalProviderLockedCollateral)
fml = types.BigAdd(fml, s.TotalClientStorageFee)
return fml, nil
}

func (s *state1) BalancesChanged(otherState State) bool {
otherState1, ok := otherState.(*state1)
if !ok {
// there's no way to compare different versions of the state, so let's
// just say that means the state of balances has changed
return true
}
return !s.State.EscrowTable.Equals(otherState1.State.EscrowTable) || !s.State.LockedTable.Equals(otherState1.State.LockedTable)
}

func (s *state1) StatesChanged(otherState State) bool {
otherState1, ok := otherState.(*state1)
if !ok {
// there's no way to compare different versions of the state, so let's
// just say that means the state of balances has changed
return true
}
return !s.State.States.Equals(otherState1.State.States)
}

func (s *state1) States() (DealStates, error) {
stateArray, err := adt1.AsArray(s.store, s.State.States)
if err != nil {
return nil, err
}
return &dealStates1{stateArray}, nil
}

func (s *state1) ProposalsChanged(otherState State) bool {
otherState1, ok := otherState.(*state1)
if !ok {
// there's no way to compare different versions of the state, so let's
// just say that means the state of balances has changed
return true
}
return !s.State.Proposals.Equals(otherState1.State.Proposals)
}

func (s *state1) Proposals() (DealProposals, error) {
proposalArray, err := adt1.AsArray(s.store, s.State.Proposals)
if err != nil {
return nil, err
}
return &dealProposals1{proposalArray}, nil
}

func (s *state1) EscrowTable() (BalanceTable, error) {
bt, err := adt1.AsBalanceTable(s.store, s.State.EscrowTable)
if err != nil {
return nil, err
}
return &balanceTable1{bt}, nil
}

func (s *state1) LockedTable() (BalanceTable, error) {
bt, err := adt1.AsBalanceTable(s.store, s.State.LockedTable)
if err != nil {
return nil, err
}
return &balanceTable1{bt}, nil
}

func (s *state1) VerifyDealsForActivation(
minerAddr address.Address, deals []abi.DealID, currEpoch, sectorExpiry abi.ChainEpoch,
) (weight, verifiedWeight abi.DealWeight, err error) {
w, vw, _, err := market.ValidateDealsForActivation(&s.State, s.store, deals, minerAddr, sectorExpiry, currEpoch)
return w, vw, err
}

type balanceTable1 struct {
*adt1.BalanceTable
}

func (bt *balanceTable1) ForEach(cb func(address.Address, abi.TokenAmount) error) error {
asMap := (*adt1.Map)(bt.BalanceTable)
var ta abi.TokenAmount
return asMap.ForEach(&ta, func(key string) error {
a, err := address.NewFromBytes([]byte(key))
if err != nil {
return err
}
return cb(a, ta)
})
}

type dealStates1 struct {
adt.Array
}

func (s *dealStates1) Get(dealID abi.DealID) (*DealState, bool, error) {
var deal1 market.DealState
found, err := s.Array.Get(uint64(dealID), &deal1)
if err != nil {
return nil, false, err
}
if !found {
return nil, false, nil
}
deal := fromV1DealState(deal1)
return &deal, true, nil
}

func (s *dealStates1) decode(val *cbg.Deferred) (*DealState, error) {
var ds1 market.DealState
if err := ds1.UnmarshalCBOR(bytes.NewReader(val.Raw)); err != nil {
return nil, err
}
ds := fromV1DealState(ds1)
return &ds, nil
}

func (s *dealStates1) array() adt.Array {
return s.Array
}

func fromV1DealState(v1 market.DealState) DealState {
return (DealState)(v1)
}

type dealProposals1 struct {
adt.Array
}

func (s *dealProposals1) Get(dealID abi.DealID) (*DealProposal, bool, error) {
var proposal1 market.DealProposal
found, err := s.Array.Get(uint64(dealID), &proposal1)
if err != nil {
return nil, false, err
}
if !found {
return nil, false, nil
}
proposal := fromV1DealProposal(proposal1)
return &proposal, true, nil
}

func (s *dealProposals1) ForEach(cb func(dealID abi.DealID, dp DealProposal) error) error {
var dp1 market.DealProposal
return s.Array.ForEach(&dp1, func(idx int64) error {
return cb(abi.DealID(idx), fromV1DealProposal(dp1))
})
}

func (s *dealProposals1) decode(val *cbg.Deferred) (*DealProposal, error) {
var dp1 market.DealProposal
if err := dp1.UnmarshalCBOR(bytes.NewReader(val.Raw)); err != nil {
return nil, err
}
dp := fromV1DealProposal(dp1)
return &dp, nil
}

func (s *dealProposals1) array() adt.Array {
return s.Array
}

func fromV1DealProposal(v1 market.DealProposal) DealProposal {
return (DealProposal)(v1)
}

0 comments on commit 2b22715

Please sign in to comment.