erigon-pulse/turbo/builder/block_builder.go
ledgerwatch 525a3a7cbb
Fix for empty blocks being proposed (#5749)
Co-authored-by: Alexey Sharp <alexeysharp@Alexeys-iMac.local>
2022-10-15 00:12:38 +01:00

68 lines
1.6 KiB
Go

package builder
import (
"sync"
"sync/atomic"
"time"
"github.com/ledgerwatch/erigon/core"
"github.com/ledgerwatch/erigon/core/types"
"github.com/ledgerwatch/log/v3"
)
type BlockBuilderFunc func(param *core.BlockBuilderParameters, interrupt *int32) (*types.Block, error)
// BlockBuilder wraps a goroutine that builds Proof-of-Stake payloads (PoS "mining")
type BlockBuilder struct {
emptyHeader *types.Header
interrupt int32
syncCond *sync.Cond
block *types.Block
err error
}
func NewBlockBuilder(build BlockBuilderFunc, param *core.BlockBuilderParameters, emptyHeader *types.Header) *BlockBuilder {
b := new(BlockBuilder)
b.emptyHeader = emptyHeader
b.syncCond = sync.NewCond(new(sync.Mutex))
go func() {
log.Debug("Building block...")
t := time.Now()
block, err := build(param, &b.interrupt)
log.Debug("Built block", "hash", block.Hash(), "height", block.NumberU64(), "txs", len(block.Transactions()), "gas used %", 100*float64(block.GasUsed())/float64(block.GasLimit()), "time", time.Since(t))
b.syncCond.L.Lock()
defer b.syncCond.L.Unlock()
b.block = block
b.err = err
b.syncCond.Broadcast()
}()
return b
}
func (b *BlockBuilder) Stop() *types.Block {
atomic.StoreInt32(&b.interrupt, 1)
b.syncCond.L.Lock()
defer b.syncCond.L.Unlock()
for b.block == nil && b.err == nil {
b.syncCond.Wait()
}
if b.err != nil {
log.Error("BlockBuilder", "err", b.err)
return types.NewBlock(b.emptyHeader, nil, nil, nil)
}
return b.block
}
func (b *BlockBuilder) Block() *types.Block {
b.syncCond.L.Lock()
defer b.syncCond.L.Unlock()
return b.block
}