Run requests concurrently

1 files changed, 56 insertions, 0 deletions

diff --git a/cmd/main/http.go b/cmd/main/http.go
new file mode 100644
index 0000000..ce31495
--- /dev/null
+++ b/cmd/main/http.go
@@ -0,0 +1,56 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+
+	"riedstra.dev/go/checkup"
+)
+
+type httpCheck struct {
+	URL  string
+	Code int
+}
+
+func httpStatusWorker(jobs <-chan httpCheck, msg chan<- *string) {
+	for hc := range jobs {
+		err := checkup.HttpStatusOK(hc.URL, hc.Code)
+		if err != nil {
+			s := fmt.Sprintf("Checking: %s, %v\n",
+				hc.URL, err)
+			msg <- &s
+			continue
+		}
+		msg <- nil
+	}
+}
+
+func checkStatus(conf *Config) {
+	buf := &bytes.Buffer{}
+
+	hc := make(chan httpCheck)
+	msgs := make(chan *string)
+
+	for i := 1; i <= conf.Workers; i++ {
+		go httpStatusWorker(hc, msgs)
+	}
+
+	go func() {
+		for url, code := range conf.StatusChecks {
+			if code == nil {
+				code = &conf.ExpectedStatusCode
+			}
+			hc <- httpCheck{url, *code}
+		}
+		close(hc)
+	}()
+
+	for i := 0; i < len(conf.StatusChecks); i++ {
+		m := <-msgs
+		if m != nil {
+			buf.Write([]byte(*m))
+		}
+	}
+
+	notify(conf, buf.Bytes())
+}