/
unregisterer.go
63 lines (55 loc) · 1.4 KB
/
unregisterer.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
package util
import "github.com/prometheus/client_golang/prometheus"
// Unregisterer is a Prometheus Registerer that can unregister all collectors
// passed to it.
type Unregisterer struct {
wrap prometheus.Registerer
cs map[prometheus.Collector]struct{}
}
// WrapWithUnregisterer wraps a prometheus Registerer with capabilities to
// unregister all collectors.
func WrapWithUnregisterer(reg prometheus.Registerer) *Unregisterer {
return &Unregisterer{
wrap: reg,
cs: make(map[prometheus.Collector]struct{}),
}
}
// Register implements prometheus.Registerer.
func (u *Unregisterer) Register(c prometheus.Collector) error {
if u.wrap == nil {
return nil
}
err := u.wrap.Register(c)
if err != nil {
return err
}
u.cs[c] = struct{}{}
return nil
}
// MustRegister implements prometheus.Registerer.
func (u *Unregisterer) MustRegister(cs ...prometheus.Collector) {
for _, c := range cs {
if err := u.Register(c); err != nil {
panic(err)
}
}
}
// Unregister implements prometheus.Registerer.
func (u *Unregisterer) Unregister(c prometheus.Collector) bool {
if u.wrap != nil && u.wrap.Unregister(c) {
delete(u.cs, c)
return true
}
return false
}
// UnregisterAll unregisters all collectors that were registered through the
// Reigsterer.
func (u *Unregisterer) UnregisterAll() bool {
success := true
for c := range u.cs {
if !u.Unregister(c) {
success = false
}
}
return success
}