localhots/shezmu
1
0
Fork 0
Code Releases Activity

Make worker pool size, add autoscale option

Browse Source
This commit is contained in:
Gregory Eremin 2015-10-27 03:09:19 +03:00
parent e0eb526af5
commit b0142de4a7
1 changed files with 11 additions and 10 deletions

11
satan.go

@ -19,6 +19,9 @@ type Satan struct {
DaemonStats stats.Publisher DaemonStats stats.Publisher
Logger *log.Logger Logger *log.Logger
DefaultNumWorkers int
AutoScale bool
daemons []Daemon daemons []Daemon
queue chan *task queue chan *task
runtimeStats stats.Manager runtimeStats stats.Manager
@ -57,10 +60,6 @@ type task struct {
name string name string
} }
const (
defaultNumWorkers = 10
)
var ( var (
workerIndex uint64 workerIndex uint64
) )
@ -69,6 +68,8 @@ var (
func Summon() *Satan { func Summon() *Satan {
return &Satan{ return &Satan{
Logger: log.New(os.Stdout, "[daemons] ", log.LstdFlags), Logger: log.New(os.Stdout, "[daemons] ", log.LstdFlags),
DefaultNumWorkers: 10,
AutoScale: false,
queue: make(chan *task), queue: make(chan *task),
runtimeStats: stats.NewBasicStats(), runtimeStats: stats.NewBasicStats(),
shutdownWorkers: make(chan struct{}), shutdownWorkers: make(chan struct{}),
@ -92,7 +93,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() {
s.addWorkers(defaultNumWorkers) s.addWorkers(s.DefaultNumWorkers)
} }
// StopDaemons stops all running daemons. // StopDaemons stops all running daemons.