Merge pull request #1954 from obscuren/regression-miner

miner: synchronise start / stop
This commit is contained in:
Jeffrey Wilcke 2015-10-31 20:00:34 +01:00
commit c841e39476
1 changed files with 12 additions and 5 deletions

View File

@ -20,6 +20,7 @@ import (
"errors"
"math/big"
"sync"
"sync/atomic"
"time"
"github.com/ethereum/ethash"
@ -45,6 +46,8 @@ type RemoteAgent struct {
hashrateMu sync.RWMutex
hashrate map[common.Hash]hashrate
running int32 // running indicates whether the agent is active. Call atomically
}
func NewRemoteAgent() *RemoteAgent {
@ -70,18 +73,22 @@ func (a *RemoteAgent) SetReturnCh(returnCh chan<- *Result) {
}
func (a *RemoteAgent) Start() {
if !atomic.CompareAndSwapInt32(&a.running, 0, 1) {
return
}
a.quit = make(chan struct{})
a.workCh = make(chan *Work, 1)
go a.maintainLoop()
}
func (a *RemoteAgent) Stop() {
if a.quit != nil {
close(a.quit)
}
if a.workCh != nil {
close(a.workCh)
if !atomic.CompareAndSwapInt32(&a.running, 1, 0) {
return
}
close(a.quit)
close(a.workCh)
}
// GetHashRate returns the accumulated hashrate of all identifier combined