Add pushgateway support.

Change-Id: I4730b150ac84ae38939b16effaf4b2ad4afa5bc0
This commit is contained in:
Bjoern Rabenstein 2014-07-01 20:20:42 +02:00
parent 998774096c
commit 0afe1a813e
2 changed files with 48 additions and 1 deletions

View File

@ -82,7 +82,7 @@ func ExampleGaugeFunc() {
}, },
func() float64 { return float64(runtime.NumGoroutine()) }, func() float64 { return float64(runtime.NumGoroutine()) },
)); err == nil { )); 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 // Note that the count of goroutines is a gauge (and not a counter) as
// it can go up and down. // it can go up and down.

View File

@ -26,6 +26,7 @@ import (
"hash/fnv" "hash/fnv"
"io" "io"
"net/http" "net/http"
"net/url"
"sort" "sort"
"sync" "sync"
@ -179,6 +180,26 @@ func EnableCollectChecks(b bool) {
defRegistry.collectChecksEnabled = b 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 // 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 // certain encoding. It returns the number of bytes written and any error
// encountered. Note that ext.WriteDelimited and text.MetricFamilyToText are // encountered. Note that ext.WriteDelimited and text.MetricFamilyToText are
@ -318,6 +339,32 @@ func (r *registry) Unregister(c Collector) bool {
return true 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) { func (r *registry) ServeHTTP(w http.ResponseWriter, req *http.Request) {
enc, contentType := chooseEncoder(req) enc, contentType := chooseEncoder(req)
buf := r.getBuf() buf := r.getBuf()