certusone
/
wormhole
mirror of https://github.com/certusone/wormhole.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

node/pkg/p2p: use mutex for error counters 

Change-Id: Idde862e034c567b7ac2d5648bec0b3505f032b0c
This commit is contained in:
Leo 2021-08-10 18:18:52 +02:00
parent 212e04a72d
commit 3f81840e69
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
bridge/pkg/p2p/registry.go
View File

@ -4,7 +4,6 @@ import (
gossipv1 "github.com/certusone/wormhole/bridge/pkg/proto/gossip/v1" gossipv1 "github.com/certusone/wormhole/bridge/pkg/proto/gossip/v1"
"github.com/certusone/wormhole/bridge/pkg/vaa" "github.com/certusone/wormhole/bridge/pkg/vaa"
"sync" "sync"
"sync/atomic"
) )
// The p2p package implements a simple global metrics registry singleton for node status values transmitted on-chain. // The p2p package implements a simple global metrics registry singleton for node status values transmitted on-chain.
@ -15,8 +14,9 @@ type registry struct {
// Mapping of chain IDs to network status messages. // Mapping of chain IDs to network status messages.
networkStats map[vaa.ChainID]*gossipv1.Heartbeat_Network networkStats map[vaa.ChainID]*gossipv1.Heartbeat_Network
// Atomic per-chain error counters // Per-chain error counters
errorCounters map[vaa.ChainID]uint64 errorCounters map[vaa.ChainID]uint64
errorCounterMu sync.Mutex
// Value of Heartbeat.guardian_addr. // Value of Heartbeat.guardian_addr.
guardianAddress string guardianAddress string
@ -51,11 +51,13 @@ func (r *registry) SetNetworkStats(chain vaa.ChainID, data *gossipv1.Heartbeat_N
} }
func (r *registry) AddErrorCount(chain vaa.ChainID, delta uint64) { func (r *registry) AddErrorCount(chain vaa.ChainID, delta uint64) {
ctr := r.errorCounters[chain] r.errorCounterMu.Lock()
atomic.AddUint64(&ctr, delta) defer r.errorCounterMu.Unlock()
r.errorCounters[chain] += 1
} }
func (r *registry) GetErrorCount(chain vaa.ChainID) uint64 { func (r *registry) GetErrorCount(chain vaa.ChainID) uint64 {
ctr := r.errorCounters[chain] r.errorCounterMu.Lock()
return atomic.LoadUint64(&ctr) defer r.errorCounterMu.Unlock()
return r.errorCounters[chain]
} }