Skip to content

Commit

Permalink
Makin telemetry creator functions private
Browse files Browse the repository at this point in the history
Those function should not have been public in the first place and there
signature should not be part of the public API we support. Users
shouldn't have any reason to use them as they can control everything
through the client options.
  • Loading branch information
hush-hush committed Oct 9, 2020
1 parent 870d284 commit ff410a4
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 12 deletions.
4 changes: 2 additions & 2 deletions statsd/statsd.go
Expand Up @@ -332,10 +332,10 @@ func newWithWriter(w statsdWriter, o *Options, writerName string) (*Client, erro

if o.Telemetry {
if o.TelemetryAddr == "" {
c.telemetry = NewTelemetryClient(&c, writerName, o.DevMode)
c.telemetry = newTelemetryClient(&c, writerName, o.DevMode)
} else {
var err error
c.telemetry, err = NewTelemetryClientWithCustomAddr(&c, writerName, o.DevMode, o.TelemetryAddr, bufferPool)
c.telemetry, err = newTelemetryClientWithCustomAddr(&c, writerName, o.DevMode, o.TelemetryAddr, bufferPool)
if err != nil {
return nil, err
}
Expand Down
6 changes: 3 additions & 3 deletions statsd/telemetry.go
Expand Up @@ -29,21 +29,21 @@ type telemetryClient struct {
devMode bool
}

func NewTelemetryClient(c *Client, transport string, devMode bool) *telemetryClient {
func newTelemetryClient(c *Client, transport string, devMode bool) *telemetryClient {
return &telemetryClient{
c: c,
tags: append(c.Tags, clientTelemetryTag, clientVersionTelemetryTag, "client_transport:"+transport),
devMode: devMode,
}
}

func NewTelemetryClientWithCustomAddr(c *Client, transport string, devMode bool, telemetryAddr string, pool *bufferPool) (*telemetryClient, error) {
func newTelemetryClientWithCustomAddr(c *Client, transport string, devMode bool, telemetryAddr string, pool *bufferPool) (*telemetryClient, error) {
telemetryWriter, _, err := resolveAddr(telemetryAddr)
if err != nil {
return nil, fmt.Errorf("Could not resolve telemetry address: %v", err)
}

t := NewTelemetryClient(c, transport, devMode)
t := newTelemetryClient(c, transport, devMode)

// Creating a custom sender/worker with 1 worker in mutex mode for the
// telemetry that share the same bufferPool.
Expand Down
14 changes: 7 additions & 7 deletions statsd/telemetry_test.go
Expand Up @@ -46,7 +46,7 @@ func TestNewTelemetry(t *testing.T) {
client, err := New("localhost:8125", WithoutTelemetry(), WithNamespace("test_namespace"))
require.Nil(t, err)

telemetry := NewTelemetryClient(client, "test_transport", false)
telemetry := newTelemetryClient(client, "test_transport", false)
assert.NotNil(t, telemetry)

assert.Equal(t, telemetry.c, client)
Expand Down Expand Up @@ -95,7 +95,7 @@ func TestTelemetry(t *testing.T) {
client, err := New("localhost:8125", WithoutTelemetry())
require.Nil(t, err)

telemetry := NewTelemetryClient(client, "test_transport", false)
telemetry := newTelemetryClient(client, "test_transport", false)
testTelemetry(t, telemetry, basicExpectedMetrics, basicExpectedTags)
}

Expand All @@ -112,7 +112,7 @@ func TestTelemetryDevMode(t *testing.T) {
expectedMetrics[k] = v
}

telemetry := NewTelemetryClient(client, "test_transport", true)
telemetry := newTelemetryClient(client, "test_transport", true)
testTelemetry(t, telemetry, expectedMetrics, basicExpectedTags)
}

Expand All @@ -121,7 +121,7 @@ func TestTelemetryChannelMode(t *testing.T) {
client, err := New("localhost:8125", WithoutTelemetry(), WithChannelMode())
require.Nil(t, err)

telemetry := NewTelemetryClient(client, "test_transport", false)
telemetry := newTelemetryClient(client, "test_transport", false)
testTelemetry(t, telemetry, basicExpectedMetrics, basicExpectedTags)
}

Expand All @@ -133,7 +133,7 @@ func TestTelemetryWithGlobalTags(t *testing.T) {
client, err := New("localhost:8125", WithoutTelemetry(), WithTags([]string{"tag1", "tag2"}))
require.Nil(t, err)

telemetry := NewTelemetryClient(client, "test_transport", false)
telemetry := newTelemetryClient(client, "test_transport", false)

expectedTelemetryTags := append([]string{"tag1", "tag2", "env:test"}, basicExpectedTags...)
testTelemetry(t, telemetry, basicExpectedMetrics, expectedTelemetryTags)
Expand All @@ -144,7 +144,7 @@ func TestTelemetryWithAggregation(t *testing.T) {
client, err := New("localhost:8125", WithoutTelemetry(), WithClientSideAggregation())
require.Nil(t, err)

telemetry := NewTelemetryClient(client, "test_transport", false)
telemetry := newTelemetryClient(client, "test_transport", false)

expectedMetrics := map[string]int64{
"datadog.dogstatsd.client.aggregated_context": 5,
Expand All @@ -161,7 +161,7 @@ func TestTelemetryWithAggregationDevMode(t *testing.T) {
client, err := New("localhost:8125", WithoutTelemetry(), WithClientSideAggregation(), WithDevMode())
require.Nil(t, err)

telemetry := NewTelemetryClient(client, "test_transport", true)
telemetry := newTelemetryClient(client, "test_transport", true)

expectedMetrics := map[string]int64{
"datadog.dogstatsd.client.aggregated_context": 5,
Expand Down

0 comments on commit ff410a4

Please sign in to comment.