localhots/shezmu
1
0
Fork 0
Code Releases Activity

Use auto increment value as worker index

Browse Source
This commit is contained in:
Gregory Eremin 2015-10-24 01:04:47 +03:00
parent 53f2cec247
commit 182cfa3e40
1 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
satan.go
View File

@ -5,6 +5,7 @@ import (
"log" "log"
"runtime/debug" "runtime/debug"
"sync" "sync"
"sync/atomic"
"time" "time"
) )
@ -55,6 +56,10 @@ const (
defaultNumWorkers = 10 defaultNumWorkers = 10
) )
var (
workerIndex uint64
)
// Summon creates a new instance of Satan. // Summon creates a new instance of Satan.
func Summon() *Satan { func Summon() *Satan {
return &Satan{ return &Satan{
@ -81,11 +86,7 @@ func (s *Satan) AddDaemon(d Daemon) {
// StartDaemons starts all registered daemons. // StartDaemons starts all registered daemons.
func (s *Satan) StartDaemons() { func (s *Satan) StartDaemons() {
for i := 0; i < defaultNumWorkers; i++ { s.addWorkers(defaultNumWorkers)
go func(i int) {
s.runWorker(i)
}(i)
}
} }
// StopDaemons stops all running daemons. // StopDaemons stops all running daemons.
@ -107,9 +108,23 @@ func (s *Satan) StopDaemons() {
log.Printf("Task processing latency statistics:\n%s\n", s.latency.snapshot()) log.Printf("Task processing latency statistics:\n%s\n", s.latency.snapshot())
} }
func (s *Satan) runWorker(i int) { func (s *Satan) addWorkers(num int) {
for i := 0; i < num; i++ {
go s.runWorker()
}
}
func (s *Satan) stopWorkers(num int) {
for i := 0; i < num; i++ {
s.shutdownWorkers <- struct{}{}
}
}
func (s *Satan) runWorker() {
s.wgWorkers.Add(1) s.wgWorkers.Add(1)
defer s.wgWorkers.Done() defer s.wgWorkers.Done()
i := atomic.AddUint64(&workerIndex, 1)
log.Printf("Starting worker #%d", i+1) log.Printf("Starting worker #%d", i+1)
defer log.Printf("Worker #%d has stopped", i+1) defer log.Printf("Worker #%d has stopped", i+1)