From 0afe1a813ed19d3ba2b6a487c84f2610c6f49e62 Mon Sep 17 00:00:00 2001
From: Bjoern Rabenstein <bjoern@soundcloud.com>
Date: Tue, 1 Jul 2014 20:20:42 +0200
Subject: [PATCH] Add pushgateway support.

Change-Id: I4730b150ac84ae38939b16effaf4b2ad4afa5bc0
---
 prometheus/examples_test.go |  2 +-
 prometheus/registry.go      | 47 +++++++++++++++++++++++++++++++++++++
 2 files changed, 48 insertions(+), 1 deletion(-)

diff --git a/prometheus/examples_test.go b/prometheus/examples_test.go
index f9b1299..14e853d 100644
--- a/prometheus/examples_test.go
+++ b/prometheus/examples_test.go
@@ -82,7 +82,7 @@ func ExampleGaugeFunc() {
 		},
 		func() float64 { return float64(runtime.NumGoroutine()) },
 	)); err == nil {
-		fmt.Println("GaugeFunc 'goroutines_count' registered.\n")
+		fmt.Println("GaugeFunc 'goroutines_count' registered.")
 	}
 	// Note that the count of goroutines is a gauge (and not a counter) as
 	// it can go up and down.
diff --git a/prometheus/registry.go b/prometheus/registry.go
index 32e43cb..4c13b40 100644
--- a/prometheus/registry.go
+++ b/prometheus/registry.go
@@ -26,6 +26,7 @@ import (
 	"hash/fnv"
 	"io"
 	"net/http"
+	"net/url"
 	"sort"
 	"sync"
 
@@ -179,6 +180,26 @@ func EnableCollectChecks(b bool) {
 	defRegistry.collectChecksEnabled = b
 }
 
+// Push triggers a metric collection and pushes all collected metrics to the
+// Pushgateway specified by addr. See the Pushgateway documentation for detailed
+// implications of the job and instance parameter. instance can be left
+// empty. The Pushgateway will then use the client's IP number instead. Use just
+// host:port or ip:port ass addr. (Don't add 'http://' or any path.)
+//
+// Note that all previously pushed metrics with the same job and instance will
+// be replaced with the metrics pushed by this call. (It uses HTTP method 'PUT'
+// to push to the Pushgateway.)
+func Push(job, instance, addr string) error {
+	return defRegistry.Push(job, instance, addr, "PUT")
+}
+
+// PushAdd works like Push, but only previously pushed metrics with the same
+// name (and the same job and instance) will be replaced. (It uses HTTP method
+// 'POST' to push to the Pushgateway.)
+func PushAdd(job, instance, addr string) error {
+	return defRegistry.Push(job, instance, addr, "POST")
+}
+
 // encoder is a function that writes a dto.MetricFamily to an io.Writer in a
 // certain encoding. It returns the number of bytes written and any error
 // encountered.  Note that ext.WriteDelimited and text.MetricFamilyToText are
@@ -318,6 +339,32 @@ func (r *registry) Unregister(c Collector) bool {
 	return true
 }
 
+func (r *registry) Push(job, instance, addr, method string) error {
+	u := fmt.Sprintf("http://%s/metrics/jobs/%s", addr, url.QueryEscape(job))
+	if instance != "" {
+		u += "/instances/" + url.QueryEscape(instance)
+	}
+	buf := r.getBuf()
+	defer r.giveBuf(buf)
+	if _, err := r.writePB(buf, text.WriteProtoDelimited); err != nil {
+		if r.panicOnCollectError {
+			panic(err)
+		}
+		return err
+	}
+	req, err := http.NewRequest(method, u, buf)
+	req.Header.Set("Content-Type", DelimitedTelemetryContentType)
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != 202 {
+		return fmt.Errorf("unexpected status code %d while pushing to %s", resp.StatusCode, u)
+	}
+	return nil
+}
+
 func (r *registry) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 	enc, contentType := chooseEncoder(req)
 	buf := r.getBuf()