From 3a59e44a4c362d01dcdb5d424c4d9d3d8efd3bcc Mon Sep 17 00:00:00 2001
From: Brad Rydzewski <brad.rydzewski@gmail.com>
Date: Wed, 26 Feb 2014 14:20:47 -0700
Subject: [PATCH] fixed issue #136

---
 cmd/droned/drone.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/cmd/droned/drone.go b/cmd/droned/drone.go
index 399682d38..27f05f7d8 100644
--- a/cmd/droned/drone.go
+++ b/cmd/droned/drone.go
@@ -42,6 +42,10 @@ var (
 	// optional flags for tls listener
 	sslcert string
 	sslkey  string
+
+	// build will timeout after N milliseconds.
+	// this will default to 500 minutes (6 hours)
+	timeout time.Duration
 )
 
 func main() {
@@ -52,6 +56,7 @@ func main() {
 	flag.StringVar(&datasource, "datasource", "drone.sqlite", "")
 	flag.StringVar(&sslcert, "sslcert", "", "")
 	flag.StringVar(&sslkey, "sslkey", "", "")
+	flag.DurationVar(&timeout, "timeout", 300*time.Minute, "")
 	flag.Parse()
 
 	// validate the TLS arguments
@@ -122,7 +127,7 @@ func setupStatic() {
 
 // setup routes for serving dynamic content.
 func setupHandlers() {
-	queueRunner := queue.NewBuildRunner(docker.New(), 300*time.Second)
+	queueRunner := queue.NewBuildRunner(docker.New(), timeout)
 	queue := queue.Start(runtime.NumCPU(), queueRunner)
 
 	hookHandler := handler.NewHookHandler(queue)