plugeth/miner/worker.go

376 lines
8.7 KiB
Go
Raw Normal View History

2015-02-04 13:52:59 +00:00
package miner
import (
"fmt"
"math/big"
"sort"
"sync"
"sync/atomic"
2015-02-28 22:09:49 +00:00
"time"
2015-02-04 13:52:59 +00:00
2015-03-18 12:00:01 +00:00
"github.com/ethereum/go-ethereum/common"
2015-02-04 13:52:59 +00:00
"github.com/ethereum/go-ethereum/core"
2015-03-23 17:27:05 +00:00
"github.com/ethereum/go-ethereum/core/state"
2015-02-04 13:52:59 +00:00
"github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/event"
2015-03-01 15:09:59 +00:00
"github.com/ethereum/go-ethereum/logger"
2015-04-04 21:04:19 +00:00
"github.com/ethereum/go-ethereum/logger/glog"
2015-02-04 13:52:59 +00:00
"github.com/ethereum/go-ethereum/pow"
"gopkg.in/fatih/set.v0"
)
2015-03-01 15:09:59 +00:00
var jsonlogger = logger.NewJsonLogger()
2015-02-04 13:52:59 +00:00
type environment struct {
totalUsedGas *big.Int
state *state.StateDB
coinbase *state.StateObject
block *types.Block
2015-03-23 11:12:49 +00:00
family *set.Set
2015-02-04 13:52:59 +00:00
uncles *set.Set
}
func env(block *types.Block, eth core.Backend) *environment {
state := state.New(block.Root(), eth.StateDb())
2015-02-04 13:52:59 +00:00
env := &environment{
totalUsedGas: new(big.Int),
state: state,
block: block,
2015-03-23 11:12:49 +00:00
family: set.New(),
2015-02-04 13:52:59 +00:00
uncles: set.New(),
coinbase: state.GetOrNewStateObject(block.Coinbase()),
}
return env
}
type Work struct {
2015-02-28 19:58:37 +00:00
Number uint64
2015-03-03 20:04:31 +00:00
Nonce uint64
2015-02-28 19:58:37 +00:00
MixDigest []byte
SeedHash []byte
}
2015-02-04 23:05:47 +00:00
type Agent interface {
2015-02-09 15:20:34 +00:00
Work() chan<- *types.Block
SetReturnCh(chan<- *types.Block)
2015-02-09 15:20:34 +00:00
Stop()
2015-02-14 15:52:14 +00:00
Start()
2015-03-20 16:42:09 +00:00
GetHashRate() int64
2015-02-04 23:05:47 +00:00
}
2015-02-04 13:52:59 +00:00
type worker struct {
mu sync.Mutex
2015-02-09 15:20:34 +00:00
agents []Agent
recv chan *types.Block
2015-02-04 13:52:59 +00:00
mux *event.TypeMux
quit chan struct{}
pow pow.PoW
atWork int64
2015-02-04 13:52:59 +00:00
eth core.Backend
chain *core.ChainManager
proc *core.BlockProcessor
2015-03-18 12:00:01 +00:00
coinbase common.Address
extra []byte
2015-02-04 13:52:59 +00:00
2015-04-07 10:32:55 +00:00
currentMu sync.Mutex
current *environment
2015-03-23 11:12:49 +00:00
uncleMu sync.Mutex
possibleUncles map[common.Hash]*types.Block
2015-04-07 10:32:55 +00:00
txQueueMu sync.Mutex
txQueue map[common.Hash]*types.Transaction
mining int64
2015-02-04 13:52:59 +00:00
}
2015-03-18 12:00:01 +00:00
func newWorker(coinbase common.Address, eth core.Backend) *worker {
2015-04-07 10:32:55 +00:00
worker := &worker{
2015-03-23 11:12:49 +00:00
eth: eth,
mux: eth.EventMux(),
recv: make(chan *types.Block),
2015-03-23 11:12:49 +00:00
chain: eth.ChainManager(),
proc: eth.BlockProcessor(),
possibleUncles: make(map[common.Hash]*types.Block),
coinbase: coinbase,
2015-04-07 10:32:55 +00:00
txQueue: make(map[common.Hash]*types.Transaction),
2015-02-09 15:20:34 +00:00
}
2015-04-07 10:32:55 +00:00
go worker.update()
go worker.wait()
worker.quit = make(chan struct{})
worker.commitNewWork()
return worker
2015-02-09 15:20:34 +00:00
}
2015-04-07 10:32:55 +00:00
func (self *worker) pendingState() *state.StateDB {
self.currentMu.Lock()
defer self.currentMu.Unlock()
return self.current.state
}
2015-04-07 10:32:55 +00:00
func (self *worker) pendingBlock() *types.Block {
self.currentMu.Lock()
defer self.currentMu.Unlock()
2015-04-07 10:32:55 +00:00
return self.current.block
}
func (self *worker) start() {
2015-02-14 15:52:14 +00:00
// spin up agents
for _, agent := range self.agents {
agent.Start()
}
2015-04-07 10:32:55 +00:00
atomic.StoreInt64(&self.mining, 1)
2015-02-09 15:20:34 +00:00
}
func (self *worker) stop() {
2015-04-07 10:32:55 +00:00
atomic.StoreInt64(&self.mining, 0)
2015-04-07 10:32:55 +00:00
atomic.StoreInt64(&self.atWork, 0)
2015-02-09 15:20:34 +00:00
}
func (self *worker) register(agent Agent) {
2015-02-04 13:52:59 +00:00
self.agents = append(self.agents, agent)
agent.SetReturnCh(self.recv)
2015-02-04 13:52:59 +00:00
}
func (self *worker) update() {
2015-04-07 10:32:55 +00:00
events := self.mux.Subscribe(core.ChainHeadEvent{}, core.ChainSideEvent{}, core.TxPreEvent{})
2015-02-04 13:52:59 +00:00
2015-02-28 22:09:49 +00:00
timer := time.NewTicker(2 * time.Second)
2015-02-04 13:52:59 +00:00
out:
for {
select {
case event := <-events.Chan():
switch ev := event.(type) {
case core.ChainHeadEvent:
2015-02-14 15:52:14 +00:00
self.commitNewWork()
2015-03-23 15:14:33 +00:00
case core.ChainSideEvent:
2015-03-23 11:12:49 +00:00
self.uncleMu.Lock()
self.possibleUncles[ev.Block.Hash()] = ev.Block
self.uncleMu.Unlock()
2015-04-07 10:32:55 +00:00
case core.TxPreEvent:
if atomic.LoadInt64(&self.mining) == 0 {
self.commitNewWork()
}
2015-02-04 13:52:59 +00:00
}
2015-02-04 13:52:59 +00:00
case <-self.quit:
2015-02-14 15:52:14 +00:00
// stop all agents
for _, agent := range self.agents {
agent.Stop()
}
2015-02-04 13:52:59 +00:00
break out
2015-02-28 22:09:49 +00:00
case <-timer.C:
if glog.V(logger.Debug) {
2015-04-04 21:04:19 +00:00
glog.Infoln("Hash rate:", self.HashRate(), "Khash")
}
// XXX In case all mined a possible uncle
2015-04-07 10:32:55 +00:00
if atomic.LoadInt64(&self.atWork) == 0 && atomic.LoadInt64(&self.mining) == 1 {
self.commitNewWork()
}
2015-02-04 13:52:59 +00:00
}
}
events.Unsubscribe()
2015-02-04 13:52:59 +00:00
}
2015-02-09 15:20:34 +00:00
func (self *worker) wait() {
for {
for block := range self.recv {
atomic.AddInt64(&self.atWork, -1)
if block == nil {
continue
}
if err := self.chain.InsertChain(types.Blocks{block}); err == nil {
for _, uncle := range block.Uncles() {
delete(self.possibleUncles, uncle.Hash())
2015-02-14 15:52:14 +00:00
}
self.mux.Post(core.NewMinedBlockEvent{block})
2015-04-04 21:04:19 +00:00
glog.V(logger.Info).Infof("🔨 Mined block #%v", block.Number())
2015-04-02 10:58:17 +00:00
jsonlogger.LogJson(&logger.EthMinerNewBlock{
BlockHash: block.Hash().Hex(),
BlockNumber: block.Number(),
ChainHeadHash: block.ParentHeaderHash.Hex(),
BlockPrevHash: block.ParentHeaderHash.Hex(),
})
} else {
self.commitNewWork()
}
2015-02-09 15:20:34 +00:00
}
}
}
func (self *worker) push() {
2015-04-07 10:32:55 +00:00
if atomic.LoadInt64(&self.mining) == 1 {
2015-02-14 15:52:14 +00:00
self.current.block.Header().GasUsed = self.current.totalUsedGas
self.current.block.SetRoot(self.current.state.Root())
2015-02-04 13:52:59 +00:00
2015-02-14 15:52:14 +00:00
// push new work to agents
for _, agent := range self.agents {
atomic.AddInt64(&self.atWork, 1)
agent.Work() <- self.current.block.Copy()
}
2015-02-04 13:52:59 +00:00
}
}
2015-04-07 10:32:55 +00:00
func (self *worker) makeCurrent() {
block := self.chain.NewBlock(self.coinbase)
if block.Time() == self.chain.CurrentBlock().Time() {
block.Header().Time++
}
block.Header().Extra = self.extra
self.current = env(block, self.eth)
2015-03-23 11:12:49 +00:00
for _, ancestor := range self.chain.GetAncestors(block, 7) {
self.current.family.Add(ancestor.Hash())
}
2015-02-04 13:52:59 +00:00
parent := self.chain.GetBlock(self.current.block.ParentHash())
self.current.coinbase.SetGasPool(core.CalcGasLimit(parent, self.current.block))
2015-04-07 10:32:55 +00:00
}
func (self *worker) commitNewWork() {
self.mu.Lock()
defer self.mu.Unlock()
self.uncleMu.Lock()
defer self.uncleMu.Unlock()
self.currentMu.Lock()
defer self.currentMu.Unlock()
self.makeCurrent()
2015-02-04 13:52:59 +00:00
transactions := self.eth.TxPool().GetTransactions()
sort.Sort(types.TxByNonce{transactions})
// Keep track of transactions which return errors so they can be removed
2015-03-23 15:35:44 +00:00
var (
remove types.Transactions
tcount = 0
)
gasLimit:
2015-03-14 11:43:10 +00:00
for i, tx := range transactions {
2015-02-04 13:52:59 +00:00
err := self.commitTransaction(tx)
switch {
case core.IsNonceErr(err):
fallthrough
case core.IsInvalidTxErr(err):
// Remove invalid transactions
2015-03-18 12:00:01 +00:00
from, _ := tx.From()
self.chain.TxState().RemoveNonce(from, tx.Nonce())
2015-02-04 13:52:59 +00:00
remove = append(remove, tx)
2015-04-04 21:04:19 +00:00
2015-04-07 12:57:04 +00:00
if glog.V(logger.Debug) {
2015-04-04 21:04:19 +00:00
glog.Infof("TX (%x) failed, will be removed: %v\n", tx.Hash().Bytes()[:4], err)
2015-04-07 12:57:04 +00:00
glog.Infoln(tx)
2015-04-04 21:04:19 +00:00
}
case state.IsGasLimitErr(err):
2015-04-07 12:57:04 +00:00
glog.V(logger.Debug).Infof("Gas limit reached for block. %d TXs included in this block\n", i)
2015-02-09 15:20:34 +00:00
// Break on gas limit
break gasLimit
2015-03-23 15:35:44 +00:00
default:
tcount++
2015-02-04 13:52:59 +00:00
}
}
self.eth.TxPool().RemoveSet(remove)
2015-03-23 17:27:05 +00:00
var (
uncles []*types.Header
badUncles []common.Hash
)
2015-03-23 11:12:49 +00:00
for hash, uncle := range self.possibleUncles {
2015-03-23 15:14:33 +00:00
if len(uncles) == 2 {
2015-03-23 11:12:49 +00:00
break
}
if err := self.commitUncle(uncle.Header()); err != nil {
2015-04-07 12:57:04 +00:00
glog.V(logger.Debug).Infof("Bad uncle found and will be removed (%x)\n", hash[:4])
2015-04-04 21:04:19 +00:00
glog.V(logger.Debug).Infoln(uncle)
2015-03-23 17:27:05 +00:00
badUncles = append(badUncles, hash)
2015-03-23 11:12:49 +00:00
} else {
2015-04-07 12:57:04 +00:00
glog.V(logger.Debug).Infof("commiting %x as uncle\n", hash[:4])
2015-03-23 15:14:33 +00:00
uncles = append(uncles, uncle.Header())
2015-03-23 11:12:49 +00:00
}
}
2015-04-07 12:57:04 +00:00
// We only care about logging if we're actually mining
if atomic.LoadInt64(&self.mining) == 1 {
glog.V(logger.Info).Infof("commit new work on block %v with %d txs & %d uncles\n", self.current.block.Number(), tcount, len(uncles))
}
2015-03-23 17:27:05 +00:00
for _, hash := range badUncles {
delete(self.possibleUncles, hash)
}
2015-03-23 15:35:44 +00:00
2015-03-23 15:14:33 +00:00
self.current.block.SetUncles(uncles)
2015-03-23 11:12:49 +00:00
core.AccumulateRewards(self.current.state, self.current.block)
2015-02-04 13:52:59 +00:00
2015-04-01 21:58:26 +00:00
self.current.state.Update()
2015-04-07 10:32:55 +00:00
2015-02-09 15:20:34 +00:00
self.push()
2015-02-04 13:52:59 +00:00
}
var (
inclusionReward = new(big.Int).Div(core.BlockReward, big.NewInt(32))
_uncleReward = new(big.Int).Mul(core.BlockReward, big.NewInt(15))
uncleReward = new(big.Int).Div(_uncleReward, big.NewInt(16))
)
func (self *worker) commitUncle(uncle *types.Header) error {
2015-03-18 12:00:01 +00:00
if self.current.uncles.Has(uncle.Hash()) {
2015-02-04 13:52:59 +00:00
// Error not unique
return core.UncleError("Uncle not unique")
}
2015-03-18 12:00:01 +00:00
self.current.uncles.Add(uncle.Hash())
2015-02-04 13:52:59 +00:00
2015-03-23 11:12:49 +00:00
if !self.current.family.Has(uncle.ParentHash) {
2015-02-04 13:52:59 +00:00
return core.UncleError(fmt.Sprintf("Uncle's parent unknown (%x)", uncle.ParentHash[0:4]))
}
2015-03-23 11:12:49 +00:00
if self.current.family.Has(uncle.Hash()) {
return core.UncleError(fmt.Sprintf("Uncle already in family (%x)", uncle.Hash()))
2015-02-04 13:52:59 +00:00
}
return nil
}
func (self *worker) commitTransaction(tx *types.Transaction) error {
snap := self.current.state.Copy()
receipt, _, err := self.proc.ApplyTransaction(self.current.coinbase, self.current.state, self.current.block, tx, self.current.totalUsedGas, true)
if err != nil && (core.IsNonceErr(err) || state.IsGasLimitErr(err) || core.IsInvalidTxErr(err)) {
self.current.state.Set(snap)
2015-02-04 13:52:59 +00:00
return err
}
self.current.block.AddTransaction(tx)
self.current.block.AddReceipt(receipt)
return nil
}
2015-02-28 22:09:49 +00:00
func (self *worker) HashRate() int64 {
var tot int64
for _, agent := range self.agents {
2015-03-20 16:42:09 +00:00
tot += agent.GetHashRate()
2015-02-28 22:09:49 +00:00
}
return tot
}