aboutsummaryrefslogtreecommitdiff
path: root/internal/pushgateway/client.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/pushgateway/client.go')
-rw-r--r--internal/pushgateway/client.go66
1 files changed, 66 insertions, 0 deletions
diff --git a/internal/pushgateway/client.go b/internal/pushgateway/client.go
new file mode 100644
index 00000000..49907cee
--- /dev/null
+++ b/internal/pushgateway/client.go
@@ -0,0 +1,66 @@
+package pushgateway
+
+import (
+ "bytes"
+ "context"
+ "crypto/tls"
+ "encoding/json"
+ "fmt"
+ "net/http"
+ "time"
+
+ "github.com/opentracing/opentracing-go"
+)
+
+type httpClient struct {
+ hc *http.Client
+}
+
+// NewHTTPClient creates a new Push Gateway client.
+func NewHTTPClient(disableTLSValidation bool) Client {
+ hc := &http.Client{
+ Timeout: 30 * time.Second,
+ Transport: &http.Transport{
+ DisableKeepAlives: true,
+ TLSClientConfig: &tls.Config{
+ InsecureSkipVerify: disableTLSValidation,
+ },
+ },
+ }
+ return &httpClient{hc: hc}
+}
+
+func (h *httpClient) Notify(ctx context.Context, url string, req *NotifyRequest, resp *NotifyResponse) error {
+ span, ctx := opentracing.StartSpanFromContext(ctx, "Notify")
+ defer span.Finish()
+
+ body, err := json.Marshal(req)
+ if err != nil {
+ return err
+ }
+ hreq, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(body))
+ if err != nil {
+ return err
+ }
+ hreq.Header.Set("Content-Type", "application/json")
+
+ hresp, err := h.hc.Do(hreq)
+ if err != nil {
+ return err
+ }
+
+ //nolint:errcheck
+ defer hresp.Body.Close()
+
+ if hresp.StatusCode == http.StatusOK {
+ return json.NewDecoder(hresp.Body).Decode(resp)
+ }
+
+ var errorBody struct {
+ Message string `json:"message"`
+ }
+ if err := json.NewDecoder(hresp.Body).Decode(&errorBody); err == nil {
+ return fmt.Errorf("push gateway: %d from %s: %s", hresp.StatusCode, url, errorBody.Message)
+ }
+ return fmt.Errorf("push gateway: %d from %s", hresp.StatusCode, url)
+}