146 lines
3.9 KiB
Go
146 lines
3.9 KiB
Go
package kit
|
|
|
|
import (
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/filecoin-project/go-state-types/big"
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/wallet"
|
|
"github.com/filecoin-project/lotus/node"
|
|
)
|
|
|
|
// DefaultPresealsPerBootstrapMiner is the number of preseals that every
|
|
// bootstrap miner has by default. It can be overridden through the
|
|
// PresealSectors option.
|
|
const DefaultPresealsPerBootstrapMiner = 2
|
|
|
|
const TestSpt = abi.RegisteredSealProof_StackedDrg2KiBV1_1
|
|
|
|
// nodeOpts is an options accumulating struct, where functional options are
|
|
// merged into.
|
|
type nodeOpts struct {
|
|
balance abi.TokenAmount
|
|
lite bool
|
|
sectors int
|
|
rpc bool
|
|
ownerKey *wallet.Key
|
|
extraNodeOpts []node.Option
|
|
|
|
subsystems MinerSubsystem
|
|
mainMiner *TestMiner
|
|
disableLibp2p bool
|
|
optBuilders []OptBuilder
|
|
proofType abi.RegisteredSealProof
|
|
}
|
|
|
|
// DefaultNodeOpts are the default options that will be applied to test nodes.
|
|
var DefaultNodeOpts = nodeOpts{
|
|
balance: big.Mul(big.NewInt(100000000), types.NewInt(build.FilecoinPrecision)),
|
|
sectors: DefaultPresealsPerBootstrapMiner,
|
|
proofType: abi.RegisteredSealProof_StackedDrg2KiBV1_1, // default _concrete_ proof type for non-genesis miners (notice the _1) for new actors versions.
|
|
}
|
|
|
|
// OptBuilder is used to create an option after some other node is already
|
|
// active. Takes all active nodes as a parameter.
|
|
type OptBuilder func(activeNodes []*TestFullNode) node.Option
|
|
|
|
// NodeOpt is a functional option for test nodes.
|
|
type NodeOpt func(opts *nodeOpts) error
|
|
|
|
func WithAllSubsystems() NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.subsystems = opts.subsystems.Add(SMarkets)
|
|
opts.subsystems = opts.subsystems.Add(SMining)
|
|
opts.subsystems = opts.subsystems.Add(SSealing)
|
|
opts.subsystems = opts.subsystems.Add(SSectorStorage)
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func WithSubsystems(systems ...MinerSubsystem) NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
for _, s := range systems {
|
|
opts.subsystems = opts.subsystems.Add(s)
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func DisableLibp2p() NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.disableLibp2p = true
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func MainMiner(m *TestMiner) NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.mainMiner = m
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// OwnerBalance specifies the balance to be attributed to a miner's owner
|
|
// account. Only relevant when creating a miner.
|
|
func OwnerBalance(balance abi.TokenAmount) NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.balance = balance
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// LiteNode specifies that this node will be a lite node. Only relevant when
|
|
// creating a fullnode.
|
|
func LiteNode() NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.lite = true
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// PresealSectors specifies the amount of preseal sectors to give to a miner
|
|
// at genesis. Only relevant when creating a miner.
|
|
func PresealSectors(sectors int) NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.sectors = sectors
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// ThroughRPC makes interactions with this node throughout the test flow through
|
|
// the JSON-RPC API.
|
|
func ThroughRPC() NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.rpc = true
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// OwnerAddr sets the owner address of a miner. Only relevant when creating
|
|
// a miner.
|
|
func OwnerAddr(wk *wallet.Key) NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.ownerKey = wk
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// ConstructorOpts are Lotus node constructor options that are passed as-is to
|
|
// the node.
|
|
func ConstructorOpts(extra ...node.Option) NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.extraNodeOpts = extra
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// ProofType sets the proof type for this node. If you're using new actor
|
|
// versions, this should be a _1 proof type.
|
|
func ProofType(proofType abi.RegisteredSealProof) NodeOpt {
|
|
return func(opts *nodeOpts) error {
|
|
opts.proofType = proofType
|
|
return nil
|
|
}
|
|
}
|