-
Notifications
You must be signed in to change notification settings - Fork 0
/
metrics.go
64 lines (56 loc) · 1.83 KB
/
metrics.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package metrics
import (
"net/http"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/collectors"
"github.com/prometheus/client_golang/prometheus/promhttp"
)
var reg *prometheus.Registry
func init() {
reg = prometheus.NewRegistry()
reg.Register(collectors.NewBuildInfoCollector())
reg.Register(collectors.NewGoCollector())
reg.Register(collectors.NewProcessCollector(collectors.ProcessCollectorOpts{
Namespace: "process",
}))
}
func Register(c ...prometheus.Collector) {
for _, coll := range c {
reg.Register(coll)
}
}
func Unregister(c ...prometheus.Collector) {
for _, coll := range c {
reg.Unregister(coll)
}
}
func Listen(host string) {
httpHandler := promhttp.HandlerFor(reg, promhttp.HandlerOpts{
ErrorLog: nil,
ErrorHandling: 0,
Registry: nil,
DisableCompression: false,
MaxRequestsInFlight: 0,
Timeout: 0,
EnableOpenMetrics: false,
})
mux := http.NewServeMux()
mux.Handle("/metrics", httpHandler)
go http.ListenAndServe(host, mux)
}