diff --git a/miner/miner.go b/miner/miner.go index b95c658590..f8d87c9108 100644 --- a/miner/miner.go +++ b/miner/miner.go @@ -58,3 +58,18 @@ func (self *Miner) Stop() { func (self *Miner) HashRate() int64 { return self.worker.HashRate() } + +func (self *Miner) AddThreads(minerThreads int) int { + if minerThreads > 0 { + for i := 0; i < minerThreads; i++ { + self.worker.register(NewCpuMiner(i, self.pow)) + } + } + + return len(self.worker.agents) +} + +func (self *Miner) RemoveThreads(num int) int { + self.worker.unregister(num) + return len(self.worker.agents) +} diff --git a/miner/worker.go b/miner/worker.go index 29992b3275..845a856cf8 100644 --- a/miner/worker.go +++ b/miner/worker.go @@ -115,6 +115,17 @@ func (self *worker) register(agent Agent) { agent.SetWorkCh(self.recv) } +func (self *worker) unregister(numremove int) { + totagents := len(self.agents) + if totagents == 0 || numremove < 1 { + return + } + if numremove > totagents { + numremove = totagents + } + self.agents = self.agents[0 : totagents-numremove] +} + func (self *worker) update() { events := self.mux.Subscribe(core.ChainEvent{}, core.NewMinedBlockEvent{})