Add 'field name' config option to customize field name when using Value parser. (#8979)

This commit is contained in:
David Bennett 2021-03-11 21:53:32 -05:00 committed by GitHub
parent f32230259c
commit c905116ade
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 61 additions and 94 deletions

View File

@ -1342,6 +1342,8 @@ func (c *Config) getParserConfig(name string, tbl *ast.Table) (*parsers.Config,
c.getFieldStringSlice(tbl, "form_urlencoded_tag_keys", &pc.FormUrlencodedTagKeys) c.getFieldStringSlice(tbl, "form_urlencoded_tag_keys", &pc.FormUrlencodedTagKeys)
c.getFieldString(tbl, "value_field_name", &pc.ValueFieldName)
//for XML parser //for XML parser
if node, ok := tbl.Fields["xml"]; ok { if node, ok := tbl.Fields["xml"]; ok {
if subtbls, ok := node.([]*ast.Table); ok { if subtbls, ok := node.([]*ast.Table); ok {
@ -1468,7 +1470,7 @@ func (c *Config) missingTomlField(typ reflect.Type, key string) error {
"prefix", "prometheus_export_timestamp", "prometheus_sort_metrics", "prometheus_string_as_label", "prefix", "prometheus_export_timestamp", "prometheus_sort_metrics", "prometheus_string_as_label",
"separator", "splunkmetric_hec_routing", "splunkmetric_multimetric", "tag_keys", "separator", "splunkmetric_hec_routing", "splunkmetric_multimetric", "tag_keys",
"tagdrop", "tagexclude", "taginclude", "tagpass", "tags", "template", "templates", "tagdrop", "tagexclude", "taginclude", "tagpass", "tags", "template", "templates",
"wavefront_source_override", "wavefront_use_strict", "xml": "value_field_name", "wavefront_source_override", "wavefront_use_strict", "xml":
// ignore fields that are common to all plugins. // ignore fields that are common to all plugins.
default: default:

View File

@ -160,7 +160,7 @@ func TestCommandError(t *testing.T) {
} }
func TestExecCommandWithGlob(t *testing.T) { func TestExecCommandWithGlob(t *testing.T) {
parser, _ := parsers.NewValueParser("metric", "string", nil) parser, _ := parsers.NewValueParser("metric", "string", "", nil)
e := NewExec() e := NewExec()
e.Commands = []string{"/bin/ech* metric_value"} e.Commands = []string{"/bin/ech* metric_value"}
e.SetParser(parser) e.SetParser(parser)
@ -176,7 +176,7 @@ func TestExecCommandWithGlob(t *testing.T) {
} }
func TestExecCommandWithoutGlob(t *testing.T) { func TestExecCommandWithoutGlob(t *testing.T) {
parser, _ := parsers.NewValueParser("metric", "string", nil) parser, _ := parsers.NewValueParser("metric", "string", "", nil)
e := NewExec() e := NewExec()
e.Commands = []string{"/bin/echo metric_value"} e.Commands = []string{"/bin/echo metric_value"}
e.SetParser(parser) e.SetParser(parser)
@ -192,7 +192,7 @@ func TestExecCommandWithoutGlob(t *testing.T) {
} }
func TestExecCommandWithoutGlobAndPath(t *testing.T) { func TestExecCommandWithoutGlobAndPath(t *testing.T) {
parser, _ := parsers.NewValueParser("metric", "string", nil) parser, _ := parsers.NewValueParser("metric", "string", "", nil)
e := NewExec() e := NewExec()
e.Commands = []string{"echo metric_value"} e.Commands = []string{"echo metric_value"}
e.SetParser(parser) e.SetParser(parser)

View File

@ -257,7 +257,7 @@ func (c *FakeConsumerGroupClaim) Messages() <-chan *sarama.ConsumerMessage {
func TestConsumerGroupHandler_Lifecycle(t *testing.T) { func TestConsumerGroupHandler_Lifecycle(t *testing.T) {
acc := &testutil.Accumulator{} acc := &testutil.Accumulator{}
parser := &value.ValueParser{MetricName: "cpu", DataType: "int"} parser := value.NewValueParser("cpu", "int", "", nil)
cg := NewConsumerGroupHandler(acc, 1, parser) cg := NewConsumerGroupHandler(acc, 1, parser)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
@ -282,7 +282,7 @@ func TestConsumerGroupHandler_Lifecycle(t *testing.T) {
func TestConsumerGroupHandler_ConsumeClaim(t *testing.T) { func TestConsumerGroupHandler_ConsumeClaim(t *testing.T) {
acc := &testutil.Accumulator{} acc := &testutil.Accumulator{}
parser := &value.ValueParser{MetricName: "cpu", DataType: "int"} parser := value.NewValueParser("cpu", "int", "", nil)
cg := NewConsumerGroupHandler(acc, 1, parser) cg := NewConsumerGroupHandler(acc, 1, parser)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
@ -392,7 +392,7 @@ func TestConsumerGroupHandler_Handle(t *testing.T) {
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
acc := &testutil.Accumulator{} acc := &testutil.Accumulator{}
parser := &value.ValueParser{MetricName: "cpu", DataType: "int"} parser := value.NewValueParser("cpu", "int", "", nil)
cg := NewConsumerGroupHandler(acc, 1, parser) cg := NewConsumerGroupHandler(acc, 1, parser)
cg.MaxMessageLen = tt.maxMessageLen cg.MaxMessageLen = tt.maxMessageLen
cg.TopicTag = tt.topicTag cg.TopicTag = tt.topicTag

View File

@ -152,6 +152,9 @@ type Config struct {
// FormData configuration // FormData configuration
FormUrlencodedTagKeys []string `toml:"form_urlencoded_tag_keys"` FormUrlencodedTagKeys []string `toml:"form_urlencoded_tag_keys"`
// Value configuration
ValueFieldName string `toml:"value_field_name"`
// XML configuration // XML configuration
XMLConfig []XMLConfig `toml:"xml"` XMLConfig []XMLConfig `toml:"xml"`
} }
@ -182,7 +185,7 @@ func NewParser(config *Config) (Parser, error) {
) )
case "value": case "value":
parser, err = NewValueParser(config.MetricName, parser, err = NewValueParser(config.MetricName,
config.DataType, config.DefaultTags) config.DataType, config.ValueFieldName, config.DefaultTags)
case "influx": case "influx":
parser, err = NewInfluxParser() parser, err = NewInfluxParser()
case "nagios": case "nagios":
@ -291,13 +294,10 @@ func NewGraphiteParser(
func NewValueParser( func NewValueParser(
metricName string, metricName string,
dataType string, dataType string,
fieldName string,
defaultTags map[string]string, defaultTags map[string]string,
) (Parser, error) { ) (Parser, error) {
return &value.ValueParser{ return value.NewValueParser(metricName, dataType, fieldName, defaultTags), nil
MetricName: metricName,
DataType: dataType,
DefaultTags: defaultTags,
}, nil
} }
func NewCollectdParser( func NewCollectdParser(

View File

@ -15,6 +15,7 @@ type ValueParser struct {
MetricName string MetricName string
DataType string DataType string
DefaultTags map[string]string DefaultTags map[string]string
FieldName string
} }
func (v *ValueParser) Parse(buf []byte) ([]telegraf.Metric, error) { func (v *ValueParser) Parse(buf []byte) ([]telegraf.Metric, error) {
@ -46,7 +47,7 @@ func (v *ValueParser) Parse(buf []byte) ([]telegraf.Metric, error) {
return nil, err return nil, err
} }
fields := map[string]interface{}{"value": value} fields := map[string]interface{}{v.FieldName: value}
metric, err := metric.New(v.MetricName, v.DefaultTags, metric, err := metric.New(v.MetricName, v.DefaultTags,
fields, time.Now().UTC()) fields, time.Now().UTC())
if err != nil { if err != nil {
@ -73,3 +74,16 @@ func (v *ValueParser) ParseLine(line string) (telegraf.Metric, error) {
func (v *ValueParser) SetDefaultTags(tags map[string]string) { func (v *ValueParser) SetDefaultTags(tags map[string]string) {
v.DefaultTags = tags v.DefaultTags = tags
} }
func NewValueParser(metricName, dataType, fieldName string, defaultTags map[string]string) *ValueParser {
if fieldName == "" {
fieldName = "value"
}
return &ValueParser{
MetricName: metricName,
DataType: dataType,
DefaultTags: defaultTags,
FieldName: fieldName,
}
}

View File

@ -7,10 +7,7 @@ import (
) )
func TestParseValidValues(t *testing.T) { func TestParseValidValues(t *testing.T) {
parser := ValueParser{ parser := NewValueParser("value_test", "integer", "", nil)
MetricName: "value_test",
DataType: "integer",
}
metrics, err := parser.Parse([]byte("55")) metrics, err := parser.Parse([]byte("55"))
assert.NoError(t, err) assert.NoError(t, err)
assert.Len(t, metrics, 1) assert.Len(t, metrics, 1)
@ -20,10 +17,7 @@ func TestParseValidValues(t *testing.T) {
}, metrics[0].Fields()) }, metrics[0].Fields())
assert.Equal(t, map[string]string{}, metrics[0].Tags()) assert.Equal(t, map[string]string{}, metrics[0].Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "float", "", nil)
MetricName: "value_test",
DataType: "float",
}
metrics, err = parser.Parse([]byte("64")) metrics, err = parser.Parse([]byte("64"))
assert.NoError(t, err) assert.NoError(t, err)
assert.Len(t, metrics, 1) assert.Len(t, metrics, 1)
@ -33,10 +27,7 @@ func TestParseValidValues(t *testing.T) {
}, metrics[0].Fields()) }, metrics[0].Fields())
assert.Equal(t, map[string]string{}, metrics[0].Tags()) assert.Equal(t, map[string]string{}, metrics[0].Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "string", "", nil)
MetricName: "value_test",
DataType: "string",
}
metrics, err = parser.Parse([]byte("foobar")) metrics, err = parser.Parse([]byte("foobar"))
assert.NoError(t, err) assert.NoError(t, err)
assert.Len(t, metrics, 1) assert.Len(t, metrics, 1)
@ -46,10 +37,7 @@ func TestParseValidValues(t *testing.T) {
}, metrics[0].Fields()) }, metrics[0].Fields())
assert.Equal(t, map[string]string{}, metrics[0].Tags()) assert.Equal(t, map[string]string{}, metrics[0].Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "boolean", "", nil)
MetricName: "value_test",
DataType: "boolean",
}
metrics, err = parser.Parse([]byte("true")) metrics, err = parser.Parse([]byte("true"))
assert.NoError(t, err) assert.NoError(t, err)
assert.Len(t, metrics, 1) assert.Len(t, metrics, 1)
@ -61,10 +49,7 @@ func TestParseValidValues(t *testing.T) {
} }
func TestParseMultipleValues(t *testing.T) { func TestParseMultipleValues(t *testing.T) {
parser := ValueParser{ parser := NewValueParser("value_test", "integer", "", nil)
MetricName: "value_test",
DataType: "integer",
}
metrics, err := parser.Parse([]byte(`55 metrics, err := parser.Parse([]byte(`55
45 45
223 223
@ -80,11 +65,19 @@ func TestParseMultipleValues(t *testing.T) {
assert.Equal(t, map[string]string{}, metrics[0].Tags()) assert.Equal(t, map[string]string{}, metrics[0].Tags())
} }
func TestParseCustomFieldName(t *testing.T) {
parser := NewValueParser("value_test", "integer", "", nil)
parser.FieldName = "penguin"
metrics, err := parser.Parse([]byte(`55`))
assert.NoError(t, err)
assert.Equal(t, map[string]interface{}{
"penguin": int64(55),
}, metrics[0].Fields())
}
func TestParseLineValidValues(t *testing.T) { func TestParseLineValidValues(t *testing.T) {
parser := ValueParser{ parser := NewValueParser("value_test", "integer", "", nil)
MetricName: "value_test",
DataType: "integer",
}
metric, err := parser.ParseLine("55") metric, err := parser.ParseLine("55")
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, "value_test", metric.Name()) assert.Equal(t, "value_test", metric.Name())
@ -93,10 +86,7 @@ func TestParseLineValidValues(t *testing.T) {
}, metric.Fields()) }, metric.Fields())
assert.Equal(t, map[string]string{}, metric.Tags()) assert.Equal(t, map[string]string{}, metric.Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "float", "", nil)
MetricName: "value_test",
DataType: "float",
}
metric, err = parser.ParseLine("64") metric, err = parser.ParseLine("64")
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, "value_test", metric.Name()) assert.Equal(t, "value_test", metric.Name())
@ -105,10 +95,7 @@ func TestParseLineValidValues(t *testing.T) {
}, metric.Fields()) }, metric.Fields())
assert.Equal(t, map[string]string{}, metric.Tags()) assert.Equal(t, map[string]string{}, metric.Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "string", "", nil)
MetricName: "value_test",
DataType: "string",
}
metric, err = parser.ParseLine("foobar") metric, err = parser.ParseLine("foobar")
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, "value_test", metric.Name()) assert.Equal(t, "value_test", metric.Name())
@ -117,10 +104,7 @@ func TestParseLineValidValues(t *testing.T) {
}, metric.Fields()) }, metric.Fields())
assert.Equal(t, map[string]string{}, metric.Tags()) assert.Equal(t, map[string]string{}, metric.Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "boolean", "", nil)
MetricName: "value_test",
DataType: "boolean",
}
metric, err = parser.ParseLine("true") metric, err = parser.ParseLine("true")
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, "value_test", metric.Name()) assert.Equal(t, "value_test", metric.Name())
@ -131,59 +115,38 @@ func TestParseLineValidValues(t *testing.T) {
} }
func TestParseInvalidValues(t *testing.T) { func TestParseInvalidValues(t *testing.T) {
parser := ValueParser{ parser := NewValueParser("value_test", "integer", "", nil)
MetricName: "value_test",
DataType: "integer",
}
metrics, err := parser.Parse([]byte("55.0")) metrics, err := parser.Parse([]byte("55.0"))
assert.Error(t, err) assert.Error(t, err)
assert.Len(t, metrics, 0) assert.Len(t, metrics, 0)
parser = ValueParser{ parser = NewValueParser("value_test", "float", "", nil)
MetricName: "value_test",
DataType: "float",
}
metrics, err = parser.Parse([]byte("foobar")) metrics, err = parser.Parse([]byte("foobar"))
assert.Error(t, err) assert.Error(t, err)
assert.Len(t, metrics, 0) assert.Len(t, metrics, 0)
parser = ValueParser{ parser = NewValueParser("value_test", "boolean", "", nil)
MetricName: "value_test",
DataType: "boolean",
}
metrics, err = parser.Parse([]byte("213")) metrics, err = parser.Parse([]byte("213"))
assert.Error(t, err) assert.Error(t, err)
assert.Len(t, metrics, 0) assert.Len(t, metrics, 0)
} }
func TestParseLineInvalidValues(t *testing.T) { func TestParseLineInvalidValues(t *testing.T) {
parser := ValueParser{ parser := NewValueParser("value_test", "integer", "", nil)
MetricName: "value_test",
DataType: "integer",
}
_, err := parser.ParseLine("55.0") _, err := parser.ParseLine("55.0")
assert.Error(t, err) assert.Error(t, err)
parser = ValueParser{ parser = NewValueParser("value_test", "float", "", nil)
MetricName: "value_test",
DataType: "float",
}
_, err = parser.ParseLine("foobar") _, err = parser.ParseLine("foobar")
assert.Error(t, err) assert.Error(t, err)
parser = ValueParser{ parser = NewValueParser("value_test", "boolean", "", nil)
MetricName: "value_test",
DataType: "boolean",
}
_, err = parser.ParseLine("213") _, err = parser.ParseLine("213")
assert.Error(t, err) assert.Error(t, err)
} }
func TestParseValidValuesDefaultTags(t *testing.T) { func TestParseValidValuesDefaultTags(t *testing.T) {
parser := ValueParser{ parser := NewValueParser("value_test", "integer", "", nil)
MetricName: "value_test",
DataType: "integer",
}
parser.SetDefaultTags(map[string]string{"test": "tag"}) parser.SetDefaultTags(map[string]string{"test": "tag"})
metrics, err := parser.Parse([]byte("55")) metrics, err := parser.Parse([]byte("55"))
assert.NoError(t, err) assert.NoError(t, err)
@ -194,10 +157,7 @@ func TestParseValidValuesDefaultTags(t *testing.T) {
}, metrics[0].Fields()) }, metrics[0].Fields())
assert.Equal(t, map[string]string{"test": "tag"}, metrics[0].Tags()) assert.Equal(t, map[string]string{"test": "tag"}, metrics[0].Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "float", "", nil)
MetricName: "value_test",
DataType: "float",
}
parser.SetDefaultTags(map[string]string{"test": "tag"}) parser.SetDefaultTags(map[string]string{"test": "tag"})
metrics, err = parser.Parse([]byte("64")) metrics, err = parser.Parse([]byte("64"))
assert.NoError(t, err) assert.NoError(t, err)
@ -208,10 +168,7 @@ func TestParseValidValuesDefaultTags(t *testing.T) {
}, metrics[0].Fields()) }, metrics[0].Fields())
assert.Equal(t, map[string]string{"test": "tag"}, metrics[0].Tags()) assert.Equal(t, map[string]string{"test": "tag"}, metrics[0].Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "string", "", nil)
MetricName: "value_test",
DataType: "string",
}
parser.SetDefaultTags(map[string]string{"test": "tag"}) parser.SetDefaultTags(map[string]string{"test": "tag"})
metrics, err = parser.Parse([]byte("foobar")) metrics, err = parser.Parse([]byte("foobar"))
assert.NoError(t, err) assert.NoError(t, err)
@ -222,10 +179,7 @@ func TestParseValidValuesDefaultTags(t *testing.T) {
}, metrics[0].Fields()) }, metrics[0].Fields())
assert.Equal(t, map[string]string{"test": "tag"}, metrics[0].Tags()) assert.Equal(t, map[string]string{"test": "tag"}, metrics[0].Tags())
parser = ValueParser{ parser = NewValueParser("value_test", "boolean", "", nil)
MetricName: "value_test",
DataType: "boolean",
}
parser.SetDefaultTags(map[string]string{"test": "tag"}) parser.SetDefaultTags(map[string]string{"test": "tag"})
metrics, err = parser.Parse([]byte("true")) metrics, err = parser.Parse([]byte("true"))
assert.NoError(t, err) assert.NoError(t, err)
@ -238,10 +192,7 @@ func TestParseValidValuesDefaultTags(t *testing.T) {
} }
func TestParseValuesWithNullCharacter(t *testing.T) { func TestParseValuesWithNullCharacter(t *testing.T) {
parser := ValueParser{ parser := NewValueParser("value_test", "integer", "", nil)
MetricName: "value_test",
DataType: "integer",
}
metrics, err := parser.Parse([]byte("55\x00")) metrics, err := parser.Parse([]byte("55\x00"))
assert.NoError(t, err) assert.NoError(t, err)
assert.Len(t, metrics, 1) assert.Len(t, metrics, 1)