blob: fe8e5e5fe2213c080e3860fb95e69d35211f8825 (
plain) (
blame)
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
|
package metrics
// addrs keeps track on which addrs we listen, so we only start one listener, is
// prometheus is used in multiple Server Blocks.
type addrs struct {
a map[string]value
}
type value struct {
state int
f func() error
}
var uniqAddr addrs
func newAddress() addrs {
return addrs{a: make(map[string]value)}
}
func (a addrs) setAddress(addr string, f func() error) {
if a.a[addr].state == done {
return
}
a.a[addr] = value{todo, f}
}
// setAddressTodo sets addr to 'todo' again.
func (a addrs) setAddressTodo(addr string) {
v, ok := a.a[addr]
if !ok {
return
}
v.state = todo
a.a[addr] = v
}
// forEachTodo iterates for a and executes f for each element that is 'todo' and sets it to 'done'.
func (a addrs) forEachTodo() error {
for k, v := range a.a {
if v.state == todo {
v.f()
}
v.state = done
a.a[k] = v
}
return nil
}
const (
todo = 1
done = 2
)
|