57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
|
package test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"sync/atomic"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/lotus/miner"
|
||
|
|
"github.com/filecoin-project/specs-actors/actors/abi"
|
||
|
|
)
|
||
|
|
|
||
|
|
type BlockMiner struct {
|
||
|
|
ctx context.Context
|
||
|
|
t *testing.T
|
||
|
|
miner TestStorageNode
|
||
|
|
blocktime time.Duration
|
||
|
|
mine int64
|
||
|
|
nulls int64
|
||
|
|
done chan struct{}
|
||
|
|
}
|
||
|
|
|
||
|
|
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 {
|
||
|
|
time.Sleep(bm.blocktime)
|
||
|
|
nulls := atomic.SwapInt64(&bm.nulls, 0)
|
||
|
|
if err := bm.miner.MineOne(bm.ctx, miner.MineReq{
|
||
|
|
InjectNulls: abi.ChainEpoch(nulls),
|
||
|
|
Done: func(bool, error) {},
|
||
|
|
}); err != nil {
|
||
|
|
bm.t.Error(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (bm *BlockMiner) Stop() {
|
||
|
|
atomic.AddInt64(&bm.mine, -1)
|
||
|
|
fmt.Println("shutting down mining")
|
||
|
|
<-bm.done
|
||
|
|
}
|