Sumo Logic output plugin: carbon2 default to include field in metric

plugins/outputs/sumologic/sumologic.go

@@ -122,11 +122,19 @@ func (s *SumoLogic) SetSerializer(serializer serializers.Serializer) {
 		s.headers = make(map[string]string)
 	}
 
-	switch serializer.(type) {
+	switch sr := serializer.(type) {
 	case *carbon2.Serializer:
 		s.headers[contentTypeHeader] = carbon2ContentType
+
+		// In case Carbon2 is used and the metrics format was unset, default to
+		// include field in metric name.
+		if sr.IsMetricsFormatUnset() {
+			sr.SetMetricsFormat(carbon2.Carbon2FormatMetricIncludesField)
+		}
+
 	case *graphite.GraphiteSerializer:
 		s.headers[contentTypeHeader] = graphiteContentType
+
 	case *prometheus.Serializer:
 		s.headers[contentTypeHeader] = prometheusContentType
 

plugins/outputs/sumologic/sumologic_test.go

@@ -2,6 +2,7 @@ package sumologic
 
 import (
 	"bufio"
+	"bytes"
 	"compress/gzip"
 	"fmt"
 	"io"
@@ -20,7 +21,6 @@ import (
 	"github.com/influxdata/telegraf/config"
 	"github.com/influxdata/telegraf/internal"
 	"github.com/influxdata/telegraf/metric"
-	"github.com/influxdata/telegraf/plugins/serializers"
 	"github.com/influxdata/telegraf/plugins/serializers/carbon2"
 	"github.com/influxdata/telegraf/plugins/serializers/graphite"
 	"github.com/influxdata/telegraf/plugins/serializers/prometheus"
@@ -135,7 +135,7 @@ func TestMethod(t *testing.T) {
 				w.WriteHeader(http.StatusOK)
 			})
 
-			serializer, err := carbon2.NewSerializer(carbon2.Carbon2FormatFieldSeparate)
+			serializer, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldSeparate))
 			require.NoError(t, err)
 
 			plugin := tt.plugin()
@@ -212,7 +212,7 @@ func TestStatusCode(t *testing.T) {
 				w.WriteHeader(tt.statusCode)
 			})
 
-			serializer, err := carbon2.NewSerializer(carbon2.Carbon2FormatFieldSeparate)
+			serializer, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldSeparate))
 			require.NoError(t, err)
 
 			tt.plugin.SetSerializer(serializer)
@@ -226,77 +226,102 @@ func TestStatusCode(t *testing.T) {
 }
 
 func TestContentType(t *testing.T) {
-	ts := httptest.NewServer(http.NotFoundHandler())
-	defer ts.Close()
-
-	ts.Config.Handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		w.WriteHeader(http.StatusOK)
-	})
-
-	u, err := url.Parse(fmt.Sprintf("http://%s", ts.Listener.Addr().String()))
-	require.NoError(t, err)
-
-	carbon2Serializer, err := carbon2.NewSerializer(carbon2.Carbon2FormatFieldSeparate)
-	require.NoError(t, err)
-
 	tests := []struct {
-		name        string
-		plugin      func() *SumoLogic
-		expectedErr bool
-		serializer  serializers.Serializer
+		name         string
+		plugin       func() *SumoLogic
+		expectedBody []byte
 	}{
 		{
-			name: "carbon2 is supported",
+			name: "carbon2 (data format = field separate) is supported",
 			plugin: func() *SumoLogic {
 				s := Default()
-				s.URL = u.String()
 				s.headers = map[string]string{
 					contentTypeHeader: carbon2ContentType,
 				}
+				sr, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldSeparate))
+				require.NoError(t, err)
+				s.SetSerializer(sr)
+				return s
+			},
+			expectedBody: []byte("metric=cpu field=value  42 0\n"),
+		},
+		{
+			name: "carbon2 (data format unset) is supported and falls back to include field in metric name",
+			plugin: func() *SumoLogic {
+				s := Default()
+				s.headers = map[string]string{
+					contentTypeHeader: carbon2ContentType,
+				}
+				sr, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldEmpty))
+				require.NoError(t, err)
+				s.SetSerializer(sr)
 				return s
 			},
-			serializer:  carbon2Serializer,
-			expectedErr: false,
+			expectedBody: []byte("metric=cpu_value  42 0\n"),
+		},
+		{
+			name: "carbon2 (data format = metric includes field) is supported",
+			plugin: func() *SumoLogic {
+				s := Default()
+				s.headers = map[string]string{
+					contentTypeHeader: carbon2ContentType,
+				}
+				sr, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatMetricIncludesField))
+				require.NoError(t, err)
+				s.SetSerializer(sr)
+				return s
+			},
+			expectedBody: []byte("metric=cpu_value  42 0\n"),
 		},
 		{
 			name: "graphite is supported",
 			plugin: func() *SumoLogic {
 				s := Default()
-				s.URL = u.String()
 				s.headers = map[string]string{
 					contentTypeHeader: graphiteContentType,
 				}
+				s.SetSerializer(&graphite.GraphiteSerializer{})
 				return s
 			},
-			serializer:  &graphite.GraphiteSerializer{},
-			expectedErr: false,
 		},
 		{
 			name: "prometheus is supported",
 			plugin: func() *SumoLogic {
 				s := Default()
-				s.URL = u.String()
 				s.headers = map[string]string{
 					contentTypeHeader: prometheusContentType,
 				}
+				s.SetSerializer(&prometheus.Serializer{})
 				return s
 			},
-			serializer:  &prometheus.Serializer{},
-			expectedErr: false,
 		},
 	}
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			plugin := tt.plugin()
-
-			plugin.SetSerializer(tt.serializer)
+			var body bytes.Buffer
+			ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				gz, err := gzip.NewReader(r.Body)
+				require.NoError(t, err)
+				io.Copy(&body, gz)
+				w.WriteHeader(http.StatusOK)
+			}))
+			defer ts.Close()
 
-			err := plugin.Connect()
+			u, err := url.Parse(fmt.Sprintf("http://%s", ts.Listener.Addr().String()))
 			require.NoError(t, err)
 
+			plugin := tt.plugin()
+			plugin.URL = u.String()
+
+			require.NoError(t, plugin.Connect())
+
 			err = plugin.Write([]telegraf.Metric{getMetric(t)})
 			require.NoError(t, err)
+
+			if tt.expectedBody != nil {
+				require.Equal(t, string(tt.expectedBody), body.String())
+			}
 		})
 	}
 }
@@ -338,7 +363,7 @@ func TestContentEncodingGzip(t *testing.T) {
 				w.WriteHeader(http.StatusNoContent)
 			})
 
-			serializer, err := carbon2.NewSerializer(carbon2.Carbon2FormatFieldSeparate)
+			serializer, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldSeparate))
 			require.NoError(t, err)
 
 			plugin := tt.plugin()
@@ -374,7 +399,7 @@ func TestDefaultUserAgent(t *testing.T) {
 			MaxRequstBodySize: Default().MaxRequstBodySize,
 		}
 
-		serializer, err := carbon2.NewSerializer(carbon2.Carbon2FormatFieldSeparate)
+		serializer, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldSeparate))
 		require.NoError(t, err)
 
 		plugin.SetSerializer(serializer)
@@ -627,7 +652,7 @@ func TestMaxRequestBodySize(t *testing.T) {
 				w.WriteHeader(http.StatusOK)
 			})
 
-			serializer, err := carbon2.NewSerializer(carbon2.Carbon2FormatFieldSeparate)
+			serializer, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldSeparate))
 			require.NoError(t, err)
 
 			plugin := tt.plugin()
@@ -659,7 +684,7 @@ func TestTryingToSendEmptyMetricsDoesntFail(t *testing.T) {
 	plugin := Default()
 	plugin.URL = u.String()
 
-	serializer, err := carbon2.NewSerializer(carbon2.Carbon2FormatFieldSeparate)
+	serializer, err := carbon2.NewSerializer(string(carbon2.Carbon2FormatFieldSeparate))
 	require.NoError(t, err)
 	plugin.SetSerializer(serializer)
 

plugins/serializers/carbon2/carbon2.go

@@ -12,35 +12,29 @@ import (
 type format string
 
 const (
-	Carbon2FormatFieldSeparate       string = "field_separate"
-	Carbon2FormatMetricIncludesField string = "metric_includes_field"
-
-	formatFieldSeparate       = format(Carbon2FormatFieldSeparate)
-	formatMetricIncludesField = format(Carbon2FormatMetricIncludesField)
+	Carbon2FormatFieldEmpty          = format("")
+	Carbon2FormatFieldSeparate       = format("field_separate")
+	Carbon2FormatMetricIncludesField = format("metric_includes_field")
 )
 
-var formats = map[string]format{
-	// Field separate is the default when no format specified.
-	"":                               formatFieldSeparate,
-	Carbon2FormatFieldSeparate:       formatFieldSeparate,
-	Carbon2FormatMetricIncludesField: formatMetricIncludesField,
+var formats = map[format]struct{}{
+	Carbon2FormatFieldEmpty:          {},
+	Carbon2FormatFieldSeparate:       {},
+	Carbon2FormatMetricIncludesField: {},
 }
 
 type Serializer struct {
 	metricsFormat format
 }
 
-func NewSerializer(f string) (*Serializer, error) {
-	var (
-		ok            bool
-		metricsFormat format
-	)
-	if metricsFormat, ok = formats[f]; !ok {
+func NewSerializer(metricsFormat string) (*Serializer, error) {
+	var f = format(metricsFormat)
+	if _, ok := formats[f]; !ok {
 		return nil, fmt.Errorf("unknown carbon2 format: %s", f)
 	}
 
 	return &Serializer{
-		metricsFormat: metricsFormat,
+		metricsFormat: f,
 	}, nil
 }
 
@@ -58,17 +52,22 @@ func (s *Serializer) SerializeBatch(metrics []telegraf.Metric) ([]byte, error) {
 
 func (s *Serializer) createObject(metric telegraf.Metric) []byte {
 	var m bytes.Buffer
+	metricsFormat := s.getMetricsFormat()
+
 	for fieldName, fieldValue := range metric.Fields() {
 		if !isNumeric(fieldValue) {
 			continue
 		}
 
-		switch s.metricsFormat {
-		case formatFieldSeparate:
+		switch metricsFormat {
+		// Field separate is the default when no format specified.
+		case Carbon2FormatFieldEmpty:
+		case Carbon2FormatFieldSeparate:
 			m.WriteString(serializeMetricFieldSeparate(
 				metric.Name(), fieldName,
 			))
-		case formatMetricIncludesField:
+
+		case Carbon2FormatMetricIncludesField:
 			m.WriteString(serializeMetricIncludeField(
 				metric.Name(), fieldName,
 			))
@@ -93,6 +92,18 @@ func (s *Serializer) createObject(metric telegraf.Metric) []byte {
 	return m.Bytes()
 }
 
+func (s *Serializer) SetMetricsFormat(f format) {
+	s.metricsFormat = f
+}
+
+func (s *Serializer) getMetricsFormat() format {
+	return s.metricsFormat
+}
+
+func (s *Serializer) IsMetricsFormatUnset() bool {
+	return s.metricsFormat == Carbon2FormatFieldEmpty
+}
+
 func serializeMetricFieldSeparate(name, fieldName string) string {
 	return fmt.Sprintf("metric=%s field=%s ",
 		strings.Replace(name, " ", "_", -1),