healthchecks.io integration

vendor/gitlab.com/etke.cc/go/healthchecks/client.go

@@ -0,0 +1,70 @@
+package healthchecks
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"strconv"
+)
+
+// Client for healthchecks
+type Client struct {
+	HTTP    *http.Client
+	log     func(string, error)
+	baseURL string
+	uuid    string
+	rid     string
+	done    chan bool
+}
+
+func (c *Client) call(operation, endpoint string, body ...io.Reader) {
+	var err error
+	var resp *http.Response
+	targetURL := fmt.Sprintf("%s/%s%s?rid=%s", c.baseURL, c.uuid, endpoint, c.rid)
+	if len(body) > 0 {
+		resp, err = c.HTTP.Post(targetURL, "text/plain; charset=utf-8", body[0])
+	} else {
+		resp, err = c.HTTP.Head(targetURL)
+	}
+	if err != nil {
+		c.log(operation, err)
+		return
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		respb, rerr := io.ReadAll(resp.Body)
+		if rerr != nil {
+			c.log(operation+":response", rerr)
+			return
+		}
+		rerr = fmt.Errorf(string(respb))
+		c.log(operation+":response", rerr)
+		return
+	}
+}
+
+// Start signal means the job started
+func (c *Client) Start(optionalBody ...io.Reader) {
+	c.call("start", "/start", optionalBody...)
+}
+
+// Success signal means the job has completed successfully (or, a continuously running process is still running and healthy).
+func (c *Client) Success(optionalBody ...io.Reader) {
+	c.call("success", "", optionalBody...)
+}
+
+// Fail signal means the job failed
+func (c *Client) Fail(optionalBody ...io.Reader) {
+	c.call("fail", "/fail", optionalBody...)
+}
+
+// Log signal just adds an event to the job log, without changing job status
+func (c *Client) Log(optionalBody ...io.Reader) {
+	c.call("log", "/log", optionalBody...)
+}
+
+// ExitStatus signal sends job's exit code (0-255)
+func (c *Client) ExitStatus(exitCode int, optionalBody ...io.Reader) {
+	c.call("exit status", "/"+strconv.Itoa(exitCode), optionalBody...)
+}