Use same timestamp for all objects in arrays in the json parser (#7412)

This commit is contained in:
presslab-us 2020-04-29 14:32:19 -04:00 committed by GitHub
parent 3163771778
commit e1b2ebe06d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 23 additions and 11 deletions

View File

@ -67,13 +67,13 @@ func New(config *Config) (*Parser, error) {
}, nil
}
func (p *Parser) parseArray(data []interface{}) ([]telegraf.Metric, error) {
func (p *Parser) parseArray(data []interface{}, timestamp time.Time) ([]telegraf.Metric, error) {
results := make([]telegraf.Metric, 0)
for _, item := range data {
switch v := item.(type) {
case map[string]interface{}:
metrics, err := p.parseObject(v)
metrics, err := p.parseObject(v, timestamp)
if err != nil {
if p.strict {
return nil, err
@ -90,7 +90,7 @@ func (p *Parser) parseArray(data []interface{}) ([]telegraf.Metric, error) {
return results, nil
}
func (p *Parser) parseObject(data map[string]interface{}) ([]telegraf.Metric, error) {
func (p *Parser) parseObject(data map[string]interface{}, timestamp time.Time) ([]telegraf.Metric, error) {
tags := make(map[string]string)
for k, v := range p.defaultTags {
tags[k] = v
@ -112,8 +112,7 @@ func (p *Parser) parseObject(data map[string]interface{}) ([]telegraf.Metric, er
}
}
//if time key is specified, set it to nTime
nTime := time.Now().UTC()
//if time key is specified, set timestamp to it
if p.timeKey != "" {
if p.timeFormat == "" {
err := fmt.Errorf("use of 'json_time_key' requires 'json_time_format'")
@ -125,7 +124,7 @@ func (p *Parser) parseObject(data map[string]interface{}) ([]telegraf.Metric, er
return nil, err
}
nTime, err = internal.ParseTimestamp(p.timeFormat, f.Fields[p.timeKey], p.timezone)
timestamp, err = internal.ParseTimestamp(p.timeFormat, f.Fields[p.timeKey], p.timezone)
if err != nil {
return nil, err
}
@ -133,13 +132,13 @@ func (p *Parser) parseObject(data map[string]interface{}) ([]telegraf.Metric, er
delete(f.Fields, p.timeKey)
//if the year is 0, set to current year
if nTime.Year() == 0 {
nTime = nTime.AddDate(time.Now().Year(), 0, 0)
if timestamp.Year() == 0 {
timestamp = timestamp.AddDate(time.Now().Year(), 0, 0)
}
}
tags, nFields := p.switchFieldToTag(tags, f.Fields)
metric, err := metric.New(name, tags, nFields, nTime)
metric, err := metric.New(name, tags, nFields, timestamp)
if err != nil {
return nil, err
}
@ -206,11 +205,12 @@ func (p *Parser) Parse(buf []byte) ([]telegraf.Metric, error) {
return nil, err
}
timestamp := time.Now().UTC()
switch v := data.(type) {
case map[string]interface{}:
return p.parseObject(v)
return p.parseObject(v, timestamp)
case []interface{}:
return p.parseArray(v)
return p.parseArray(v, timestamp)
default:
return nil, ErrWrongType
}

View File

@ -794,6 +794,18 @@ func TestTimeErrors(t *testing.T) {
require.Equal(t, fmt.Errorf("JSON time key could not be found"), err)
}
func TestShareTimestamp(t *testing.T) {
parser, err := New(&Config{
MetricName: "json_test",
})
require.NoError(t, err)
metrics, err := parser.Parse([]byte(validJSONArrayMultiple))
require.NoError(t, err)
require.Equal(t, 2, len(metrics))
require.Equal(t, true, metrics[0].Time() == metrics[1].Time())
}
func TestNameKey(t *testing.T) {
testString := `{
"a": 5,