Added "name" parameter to NATS output plugin (#8429)

influxdata · Nov 27, 2020 · f5d5a51 · f5d5a51
1 parent 4090c77
commit f5d5a51
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/plugins/outputs/nats/README.md b/plugins/outputs/nats/README.md
@@ -6,6 +6,10 @@ This plugin writes to a (list of) specified NATS instance(s).
 [[outputs.nats]]
   ## URLs of NATS servers
   servers = ["nats://localhost:4222"]
+
+  ## Optional client name
+  # name = ""
+
   ## Optional credentials
   # username = ""
   # password = ""

diff --git a/plugins/outputs/nats/nats.go b/plugins/outputs/nats/nats.go
@@ -15,6 +15,7 @@ import (
 type NATS struct {
 	Servers     []string `toml:"servers"`
 	Secure      bool     `toml:"secure"`
+	Name        string   `toml:"name"`
 	Username    string   `toml:"username"`
 	Password    string   `toml:"password"`
 	Credentials string   `toml:"credentials"`
@@ -30,6 +31,9 @@ var sampleConfig = `
   ## URLs of NATS servers
   servers = ["nats://localhost:4222"]
 
+  ## Optional client name
+  # name = ""
+
   ## Optional credentials
   # username = ""
   # password = ""
@@ -73,6 +77,10 @@ func (n *NATS) Connect() error {
 		opts = append(opts, nats.UserInfo(n.Username, n.Password))
 	}
 
+	if n.Name != "" {
+		opts = append(opts, nats.Name(n.Name))
+	}
+
 	if n.Secure {
 		tlsConfig, err := n.ClientConfig.TLSConfig()
 		if err != nil {

diff --git a/plugins/outputs/nats/nats_test.go b/plugins/outputs/nats/nats_test.go
@@ -17,6 +17,7 @@ func TestConnectAndWrite(t *testing.T) {
 	s, _ := serializers.NewInfluxSerializer()
 	n := &NATS{
 		Servers:    server,
+		Name:       "telegraf",
 		Subject:    "telegraf",
 		serializer: s,
 	}