From c36eb4788c3bb4fb448298cf7cc28b22c2364f94 Mon Sep 17 00:00:00 2001
From: JamesClonk <jamesclonk@jamesclonk.ch>
Date: Thu, 6 Oct 2016 16:22:20 +0200
Subject: [PATCH] add riemann_tag_keys option to specify which tags to send

Signed-off-by: Fabio Berchtold <fabio.berchtold@swisscom.com>
---
 etc/telegraf.conf                  |  4 ++--
 plugins/outputs/riemann/riemann.go | 26 ++++++++++++++++----------
 2 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/etc/telegraf.conf b/etc/telegraf.conf
index 3dc5cb183..08a1457fc 100644
--- a/etc/telegraf.conf
+++ b/etc/telegraf.conf
@@ -440,8 +440,8 @@
 #   separator = " "
 #   ## set measurement name as a Riemann tag instead of prepending it to the Riemann service name
 #   measurement_as_tag = false
-#   ## list of Riemann tags, if specified use these instead of any Telegraf tags
-#   tags = ["telegraf","custom_tag"]
+#   ## list of tag keys to specify, whose values get sent as Riemann tags. If empty, all Telegraf tag values will be sent to Riemann as tags.
+#   # riemann_tag_keys = ["telegraf","custom_tag"]
 
 
 ###############################################################################
diff --git a/plugins/outputs/riemann/riemann.go b/plugins/outputs/riemann/riemann.go
index 44602705c..57c89325e 100644
--- a/plugins/outputs/riemann/riemann.go
+++ b/plugins/outputs/riemann/riemann.go
@@ -16,7 +16,7 @@ type Riemann struct {
 	Transport        string
 	Separator        string
 	MeasurementAsTag bool
-	Tags             []string
+	RiemannTagKeys   []string
 
 	client *raidman.Client
 }
@@ -30,8 +30,8 @@ var sampleConfig = `
   separator = " "
   ## set measurement name as a Riemann tag instead of prepending it to the Riemann service name
   measurement_as_tag = false
-  ## list of Riemann tags, if specified use these instead of any Telegraf tags
-  tags = ["telegraf","custom_tag"]
+  ## list of tag keys to specify, whose values get sent as Riemann tags. If empty, all Telegraf tag values will be sent to Riemann as tags.
+  # riemann_tag_keys = ["telegraf","custom_tag"]
 `
 
 func (r *Riemann) Connect() error {
@@ -136,21 +136,27 @@ func (r *Riemann) attributes(name string, tags map[string]string) map[string]str
 }
 
 func (r *Riemann) tags(name string, tags map[string]string) []string {
-	if len(r.Tags) > 0 {
-		return r.Tags
+	var tagNames, tagValues []string
+
+	if r.MeasurementAsTag {
+		tagValues = append(tagValues, name)
 	}
 
-	var tagNames, tagValues []string
+	if len(r.RiemannTagKeys) > 0 {
+		for _, tagName := range r.RiemannTagKeys {
+			tagValue, ok := tags[tagName]
+			if ok {
+				tagValues = append(tagValues, tagValue)
+			}
+		}
+		return tagValues
+	}
 
 	for tagName := range tags {
 		tagNames = append(tagNames, tagName)
 	}
 	sort.Strings(tagNames)
 
-	if r.MeasurementAsTag {
-		tagValues = append(tagValues, name)
-	}
-
 	for _, tagName := range tagNames {
 		if tagName != "host" { // we'll skip the 'host' tag
 			tagValues = append(tagValues, tags[tagName])