From f14d2277327f50feaf03f4e4b2337a80bfea098d Mon Sep 17 00:00:00 2001 From: Jonathan Thurman Date: Wed, 19 Feb 2020 10:34:39 -0800 Subject: [PATCH] fix(newrelic): Add option to set ServiceName in Config --- newrelic/newrelic.go | 11 +++++++++++ newrelic/newrelic_test.go | 8 ++++++++ 2 files changed, 19 insertions(+) diff --git a/newrelic/newrelic.go b/newrelic/newrelic.go index fe9049955..a08677f7a 100644 --- a/newrelic/newrelic.go +++ b/newrelic/newrelic.go @@ -119,6 +119,17 @@ func ConfigUserAgent(ua string) ConfigOption { } } +// ConfigServiceName sets the service name logged +func ConfigServiceName(name string) ConfigOption { + return func(cfg *config.Config) error { + if name != "" { + cfg.ServiceName = name + } + + return nil + } +} + // ConfigBaseURL sets the base URL used to make requests to the REST API V2. func ConfigBaseURL(url string) ConfigOption { return func(cfg *config.Config) error { diff --git a/newrelic/newrelic_test.go b/newrelic/newrelic_test.go index 0a4fcf690..183c214ed 100644 --- a/newrelic/newrelic_test.go +++ b/newrelic/newrelic_test.go @@ -98,6 +98,14 @@ func TestNew_optionUserAgent(t *testing.T) { assert.NoError(t, err) } +func TestNew_optionServiceName(t *testing.T) { + t.Parallel() + + nr, err := New(ConfigAPIKey(testAPIkey), ConfigServiceName("my-service")) + assert.NotNil(t, nr) + assert.NoError(t, err) +} + func TestNew_optionBaseURL(t *testing.T) { t.Parallel()