telegraf/plugins/serializers/json/json.go

74 lines
1.5 KiB
Go
Raw Normal View History

2016-03-17 17:50:39 +00:00
package json
import (
"encoding/json"
"time"
2016-03-17 17:50:39 +00:00
"github.com/influxdata/telegraf"
)
type serializer struct {
TimestampUnits time.Duration
2016-03-17 17:50:39 +00:00
}
func NewSerializer(timestampUnits time.Duration) (*serializer, error) {
s := &serializer{
TimestampUnits: truncateDuration(timestampUnits),
}
return s, nil
}
func (s *serializer) Serialize(metric telegraf.Metric) ([]byte, error) {
m := s.createObject(metric)
serialized, err := json.Marshal(m)
2016-03-17 17:50:39 +00:00
if err != nil {
return []byte{}, err
2016-03-17 17:50:39 +00:00
}
serialized = append(serialized, '\n')
2016-03-17 17:50:39 +00:00
return serialized, nil
2016-03-17 17:50:39 +00:00
}
func (s *serializer) SerializeBatch(metrics []telegraf.Metric) ([]byte, error) {
objects := make([]interface{}, 0, len(metrics))
for _, metric := range metrics {
m := s.createObject(metric)
objects = append(objects, m)
}
obj := map[string]interface{}{
"metrics": objects,
}
serialized, err := json.Marshal(obj)
if err != nil {
return []byte{}, err
}
return serialized, nil
}
func (s *serializer) createObject(metric telegraf.Metric) map[string]interface{} {
m := make(map[string]interface{}, 4)
m["tags"] = metric.Tags()
m["fields"] = metric.Fields()
m["name"] = metric.Name()
m["timestamp"] = metric.Time().UnixNano() / int64(s.TimestampUnits)
return m
}
func truncateDuration(units time.Duration) time.Duration {
// Default precision is 1s
if units <= 0 {
return time.Second
}
// Search for the power of ten less than the duration
d := time.Nanosecond
for {
if d*10 > units {
return d
}
d = d * 10
}
}