types: refactor *memMarker.Count method

Signed-off-by: Simon Pasquier <spasquie@redhat.com>

types/types.go

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