types: refactor *memMarker.Count method

Signed-off-by: Simon Pasquier <spasquie@redhat.com>
This commit is contained in:
Simon Pasquier 2019-05-24 13:36:01 +02:00
parent 588b4dd17b
commit 27e99e9e35
1 changed files with 8 additions and 9 deletions

View File

@ -128,14 +128,14 @@ func (m *memMarker) registerMetrics(r prometheus.Registerer) {
// Count implements Marker. // Count implements Marker.
func (m *memMarker) Count(states ...AlertState) int { func (m *memMarker) Count(states ...AlertState) int {
count := 0
m.mtx.RLock() m.mtx.RLock()
defer m.mtx.RUnlock() defer m.mtx.RUnlock()
if len(states) == 0 { if len(states) == 0 {
count = len(m.m) return len(m.m)
} else { }
var count int
for _, status := range m.m { for _, status := range m.m {
for _, state := range states { for _, state := range states {
if status.State == state { if status.State == state {
@ -143,7 +143,6 @@ func (m *memMarker) Count(states ...AlertState) int {
} }
} }
} }
}
return count return count
} }