mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-27 05:57:28 +00:00
525a3a7cbb
Co-authored-by: Alexey Sharp <alexeysharp@Alexeys-iMac.local>
68 lines
1.6 KiB
Go
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
|
|
}
|