aboutsummaryrefslogtreecommitdiff
path: root/plugin/metrics/addr.go
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
)