64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
package kit
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/filecoin-project/lotus/miner"
|
|
)
|
|
|
|
// BlockMiner is a utility that makes a test Miner Mine blocks on a timer.
|
|
type BlockMiner struct {
|
|
ctx context.Context
|
|
t *testing.T
|
|
miner TestStorageNode
|
|
blocktime time.Duration
|
|
done chan struct{}
|
|
|
|
Mine int64
|
|
Nulls int64
|
|
}
|
|
|
|
func NewBlockMiner(ctx context.Context, t *testing.T, miner TestStorageNode, blocktime time.Duration) *BlockMiner {
|
|
return &BlockMiner{
|
|
ctx: ctx,
|
|
t: t,
|
|
miner: miner,
|
|
blocktime: blocktime,
|
|
Mine: int64(1),
|
|
done: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
func (bm *BlockMiner) MineBlocks() {
|
|
time.Sleep(time.Second)
|
|
go func() {
|
|
defer close(bm.done)
|
|
for atomic.LoadInt64(&bm.Mine) == 1 {
|
|
select {
|
|
case <-bm.ctx.Done():
|
|
return
|
|
case <-time.After(bm.blocktime):
|
|
}
|
|
|
|
nulls := atomic.SwapInt64(&bm.Nulls, 0)
|
|
if err := bm.miner.MineOne(bm.ctx, miner.MineReq{
|
|
InjectNulls: abi.ChainEpoch(nulls),
|
|
Done: func(bool, abi.ChainEpoch, error) {},
|
|
}); err != nil {
|
|
bm.t.Error(err)
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (bm *BlockMiner) Stop() {
|
|
atomic.AddInt64(&bm.Mine, -1)
|
|
fmt.Println("shutting down mining")
|
|
<-bm.done
|
|
}
|