Make worker pool size, add autoscale option

This commit is contained in:
2015-10-27 03:09:19 +03:00
parent e0eb526af5
commit b0142de4a7
+11 -10
View File
@@ -19,6 +19,9 @@ type Satan struct {
DaemonStats stats.Publisher
Logger *log.Logger
DefaultNumWorkers int
AutoScale bool
daemons []Daemon
queue chan *task
runtimeStats stats.Manager
@@ -57,10 +60,6 @@ type task struct {
name string
}
const (
defaultNumWorkers = 10
)
var (
workerIndex uint64
)
@@ -68,11 +67,13 @@ var (
// Summon creates a new instance of Satan.
func Summon() *Satan {
return &Satan{
Logger: log.New(os.Stdout, "[daemons] ", log.LstdFlags),
queue: make(chan *task),
runtimeStats: stats.NewBasicStats(),
shutdownWorkers: make(chan struct{}),
shutdownSystem: make(chan struct{}),
Logger: log.New(os.Stdout, "[daemons] ", log.LstdFlags),
DefaultNumWorkers: 10,
AutoScale: false,
queue: make(chan *task),
runtimeStats: stats.NewBasicStats(),
shutdownWorkers: make(chan struct{}),
shutdownSystem: make(chan struct{}),
}
}
@@ -92,7 +93,7 @@ func (s *Satan) AddDaemon(d Daemon) {
// StartDaemons starts all registered daemons.
func (s *Satan) StartDaemons() {
s.addWorkers(defaultNumWorkers)
s.addWorkers(s.DefaultNumWorkers)
}
// StopDaemons stops all running daemons.