add Count case

gojek · Jul 1, 2019 · eb99107 · eb99107
1 parent ec32050
commit eb99107
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/metrics/statsd_collector.go b/pkg/metrics/statsd_collector.go
@@ -55,7 +55,7 @@ func InitializeStatsdCollector(config *StatsdCollectorConfig, collectorName stri
 	return nil
 }
 
-var formatter = func(on string) string {
+func formatter(on string) string {
 	return fmt.Sprintf("%s.%s", instance.collectorName, on)
 }
 
@@ -70,6 +70,10 @@ func Update(updateOption UpdateOption) {
 		break
 	case Gauge:
 		err = instance.client.Gauge(formatter(updateOption.Name), int64(updateOption.NumValue), instance.sampleRate)
+		break
+	case Count:
+		err = instance.client.Inc(formatter(updateOption.Name), 1, instance.sampleRate)
+		break
 	}
 	if err != nil {
 		logger.Error(err)