[Ajat|Baskara] set sampleRate from config

gojek · Jul 1, 2019 · 8805074 · 8805074
1 parent 5e77be3
commit 8805074
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/pkg/plugins/statsd_collector.go b/pkg/plugins/statsd_collector.go
@@ -1,6 +1,5 @@
 package plugins
 
-
 import (
 	"github.com/cactus/go-statsd-client/statsd"
 )
@@ -21,5 +20,6 @@ type StatsdCollectorConfig struct {
 }
 
 func InitializeStatsdCollector(config *StatsdCollectorConfig) (*StatsdCollectorClient, error) {
-	return &StatsdCollectorClient{}, nil
-}
+	sampleRate := config.SampleRate
+	return &StatsdCollectorClient{sampleRate: sampleRate}, nil
+}
diff --git a/pkg/plugins/statsd_collector_test.go b/pkg/plugins/statsd_collector_test.go
@@ -11,6 +11,12 @@ func TestInitializeStatsdCollector(t *testing.T) {
 	scc, err := InitializeStatsdCollector(&StatsdCollectorConfig{})
 	assert.Nil(t, err)
 	assert.NotNil(t, scc)
+
+	// Test sampleRate
+	scc, err = InitializeStatsdCollector(&StatsdCollectorConfig{SampleRate: 5})
+	assert.Nil(t, err)
+	assert.NotNil(t, scc)
+	assert.Equal(t, float32(5), scc.sampleRate)
 }
 
 type mockStatsdClient struct {