set num of workers using command line params

This commit is contained in:
Mohit Gupta 2014-03-20 15:58:06 -07:00
parent 3b3da0424e
commit 2b90f0a180
2 changed files with 7 additions and 1 deletions

View file

@ -49,6 +49,10 @@ var (
// commit sha for the current build.
version string
// Number of concurrent build workers to run
// default to number of CPUs on machine
workers int
)
func main() {
@ -60,6 +64,7 @@ func main() {
flag.StringVar(&sslcert, "sslcert", "", "")
flag.StringVar(&sslkey, "sslkey", "", "")
flag.DurationVar(&timeout, "timeout", 300*time.Minute, "")
flag.IntVar(&workers, "workers", runtime.NumCPU(), "")
flag.Parse()
// validate the TLS arguments
@ -135,7 +140,7 @@ func setupStatic() {
// setup routes for serving dynamic content.
func setupHandlers() {
queueRunner := queue.NewBuildRunner(docker.New(), timeout)
queue := queue.Start(runtime.NumCPU(), queueRunner)
queue := queue.Start(workers, queueRunner)
hookHandler := handler.NewHookHandler(queue)

View file

@ -6,5 +6,6 @@
# -driver="sqlite3":
# -path="":
# -port=":8080":
# -workers="4":
#
#DRONED_OPTS="--port=:80"