Allow the timeout to be adjusted

3 files changed, 6 insertions, 5 deletions

diff --git a/cmd/main/http.go b/cmd/main/http.go
index 94d6ae5..fd6169e 100644
--- a/cmd/main/http.go
+++ b/cmd/main/http.go
@@ -12,9 +12,9 @@ type httpCheck struct {
 	Code int
 }
 
-func httpStatusWorker(jobs <-chan httpCheck, msg chan<- *jobResponse) {
+func httpStatusWorker(jobs <-chan httpCheck, msg chan<- *jobResponse, timeout int) {
 	for hc := range jobs {
-		err := checkup.HttpStatusOK(hc.URL, hc.Code)
+		err := checkup.HttpStatusOK(hc.URL, timeout, hc.Code)
 		if err != nil {
 			s := fmt.Sprintf("Checking: %s, %v",
 				hc.URL, err)
@@ -36,7 +36,7 @@ func checkStatus(conf *Config) map[string]*jobResponse {
 	msgs := make(chan *jobResponse)
 
 	for i := 1; i <= conf.Workers; i++ {
-		go httpStatusWorker(hc, msgs)
+		go httpStatusWorker(hc, msgs, conf.HTTPTimeout)
 	}
 
 	go func() {
diff --git a/cmd/main/main.go b/cmd/main/main.go
index 34fc6f4..1b264a3 100644
--- a/cmd/main/main.go
+++ b/cmd/main/main.go
@@ -31,6 +31,7 @@ type Config struct {
 	StatusChecks       map[string]*int    `yaml:"StatusChecks"`
 	Workers            int                `yaml:"Workers"`
 	Interval           int                `yaml:"Interval"`
+	HTTPTimeout        int                `yaml:"HTTPTimeout"`
 }
 
 func ReadConfig(fn string, conf *Config) error {
diff --git a/http.go b/http.go
index 24e0b4d..daf8945 100644
--- a/http.go
+++ b/http.go
@@ -6,8 +6,8 @@ import (
 	"time"
 )
 
-func HttpStatusOK(url string, status int) error {
-	client := &http.Client{Timeout: 5 * time.Second}
+func HttpStatusOK(url string, timeout, status int) error {
+	client := &http.Client{Timeout: time.Duration(timeout) * time.Second}
 
 	resp, err := client.Get(url)