forked from mirror/client_golang
Merge "Add pushgateway support."
This commit is contained in:
commit
c21d91ceb9
|
@ -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.
|
||||||
|
|
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue