lotus/miner/testminer.go

54 lines
1.4 KiB
Go
Raw Normal View History

2019-09-23 15:27:30 +00:00
package miner
import (
"context"
2019-11-25 04:45:13 +00:00
2020-08-06 01:16:54 +00:00
lru "github.com/hashicorp/golang-lru"
ds "github.com/ipfs/go-datastore"
"github.com/filecoin-project/go-address"
2019-11-25 04:45:13 +00:00
"github.com/filecoin-project/lotus/api"
2019-11-30 23:17:50 +00:00
"github.com/filecoin-project/lotus/chain/gen"
2020-08-06 01:16:54 +00:00
"github.com/filecoin-project/lotus/chain/gen/slashfilter"
2020-07-28 16:16:46 +00:00
"github.com/filecoin-project/specs-actors/actors/abi"
2019-09-23 15:27:30 +00:00
)
2020-07-28 16:16:46 +00:00
type MineReq struct {
InjectNulls abi.ChainEpoch
Done func(bool, abi.ChainEpoch, error)
2020-07-28 16:16:46 +00:00
}
func NewTestMiner(nextCh <-chan MineReq, addr address.Address) func(api.FullNode, gen.WinningPoStProver) *Miner {
return func(api api.FullNode, epp gen.WinningPoStProver) *Miner {
2020-01-17 07:10:47 +00:00
arc, err := lru.NewARC(10000)
if err != nil {
panic(err)
}
2019-11-30 23:17:50 +00:00
m := &Miner{
2020-01-17 07:10:47 +00:00
api: api,
waitFunc: chanWaiter(nextCh),
epp: epp,
minedBlockHeights: arc,
address: addr,
2020-08-06 01:16:54 +00:00
sf: slashfilter.New(ds.NewMapDatastore()),
2019-09-23 15:27:30 +00:00
}
2019-11-30 23:17:50 +00:00
if err := m.Start(context.TODO()); err != nil {
2019-11-30 23:17:50 +00:00
panic(err)
}
return m
2019-09-23 15:27:30 +00:00
}
}
func chanWaiter(next <-chan MineReq) func(ctx context.Context, _ uint64) (func(bool, abi.ChainEpoch, error), abi.ChainEpoch, error) {
return func(ctx context.Context, _ uint64) (func(bool, abi.ChainEpoch, error), abi.ChainEpoch, error) {
2019-09-23 15:27:30 +00:00
select {
case <-ctx.Done():
2020-07-28 16:16:46 +00:00
return nil, 0, ctx.Err()
case req := <-next:
return req.Done, req.InjectNulls, nil
2019-09-23 15:27:30 +00:00
}
}
}