RepoMirrors
/
alertmanager
mirror of https://github.com/prometheus/alertmanager
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Clustering: Fix unsynchronised access 

Signed-off-by: gotjosh <josue@grafana.com>
This commit is contained in:
gotjosh 2021-04-13 13:51:07 +01:00
parent 54431be888
commit d0406d6295
No known key found for this signature in database
GPG Key ID: 731A782929412860
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cluster/cluster.go
View File

@ -536,7 +536,11 @@ func (p *Peer) peerUpdate(n *memberlist.Node) {
// AddState adds a new state that will be gossiped. It returns a channel to which // AddState adds a new state that will be gossiped. It returns a channel to which
// broadcast messages for the state can be sent. // broadcast messages for the state can be sent.
func (p *Peer) AddState(key string, s State, reg prometheus.Registerer) ClusterChannel { func (p *Peer) AddState(key string, s State, reg prometheus.Registerer) ClusterChannel {
p.mtx.Lock()
p.states[key] = s p.states[key] = s
p.mtx.Unlock()
send := func(b []byte) { send := func(b []byte) {
p.delegate.bcast.QueueBroadcast(simpleBroadcast(b)) p.delegate.bcast.QueueBroadcast(simpleBroadcast(b))
} }

5
cluster/delegate.go
View File

@ -157,7 +157,10 @@ func (d *delegate) NotifyMsg(b []byte) {
return return
} }
d.mtx.RLock()
s, ok := d.states[p.Key] s, ok := d.states[p.Key]
d.mtx.RUnlock()
if !ok { if !ok {
return return
} }
@ -179,6 +182,8 @@ func (d *delegate) GetBroadcasts(overhead, limit int) [][]byte {
// LocalState is called when gossip fetches local state. // LocalState is called when gossip fetches local state.
func (d *delegate) LocalState(_ bool) []byte { func (d *delegate) LocalState(_ bool) []byte {
d.mtx.RLock()
defer d.mtx.RUnlock()
all := &clusterpb.FullState{ all := &clusterpb.FullState{
Parts: make([]clusterpb.Part, 0, len(d.states)), Parts: make([]clusterpb.Part, 0, len(d.states)),
} }