97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package v2
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
serializer "github.com/influxdata/telegraf/plugins/serializers/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
dto "github.com/prometheus/client_model/go"
|
|
)
|
|
|
|
type Metric struct {
|
|
family *dto.MetricFamily
|
|
metric *dto.Metric
|
|
}
|
|
|
|
func (m *Metric) Desc() *prometheus.Desc {
|
|
labelNames := make([]string, 0, len(m.metric.Label))
|
|
for _, label := range m.metric.Label {
|
|
labelNames = append(labelNames, *label.Name)
|
|
}
|
|
|
|
desc := prometheus.NewDesc(*m.family.Name, *m.family.Help, labelNames, nil)
|
|
|
|
return desc
|
|
}
|
|
|
|
func (m *Metric) Write(out *dto.Metric) error {
|
|
out.Label = m.metric.Label
|
|
out.Counter = m.metric.Counter
|
|
out.Untyped = m.metric.Untyped
|
|
out.Gauge = m.metric.Gauge
|
|
out.Histogram = m.metric.Histogram
|
|
out.Summary = m.metric.Summary
|
|
out.TimestampMs = m.metric.TimestampMs
|
|
return nil
|
|
}
|
|
|
|
type Collector struct {
|
|
sync.Mutex
|
|
expireDuration time.Duration
|
|
coll *serializer.Collection
|
|
}
|
|
|
|
func NewCollector(expire time.Duration, stringsAsLabel bool) *Collector {
|
|
config := serializer.FormatConfig{}
|
|
if stringsAsLabel {
|
|
config.StringHandling = serializer.StringAsLabel
|
|
}
|
|
return &Collector{
|
|
expireDuration: expire,
|
|
coll: serializer.NewCollection(config),
|
|
}
|
|
}
|
|
|
|
func (c *Collector) Describe(ch chan<- *prometheus.Desc) {
|
|
// Sending no descriptor at all marks the Collector as "unchecked",
|
|
// i.e. no checks will be performed at registration time, and the
|
|
// Collector may yield any Metric it sees fit in its Collect method.
|
|
return
|
|
}
|
|
|
|
func (c *Collector) Collect(ch chan<- prometheus.Metric) {
|
|
c.Lock()
|
|
defer c.Unlock()
|
|
|
|
// Expire metrics, doing this on Collect ensure metrics are removed even if no
|
|
// new metrics are added to the output.
|
|
if c.expireDuration != 0 {
|
|
c.coll.Expire(time.Now(), c.expireDuration)
|
|
}
|
|
|
|
for _, family := range c.coll.GetProto() {
|
|
for _, metric := range family.Metric {
|
|
ch <- &Metric{family: family, metric: metric}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *Collector) Add(metrics []telegraf.Metric) error {
|
|
c.Lock()
|
|
defer c.Unlock()
|
|
|
|
for _, metric := range metrics {
|
|
c.coll.Add(metric)
|
|
}
|
|
|
|
// Expire metrics, doing this on Add ensure metrics are removed even if no
|
|
// one is querying the data.
|
|
if c.expireDuration != 0 {
|
|
c.coll.Expire(time.Now(), c.expireDuration)
|
|
}
|
|
|
|
return nil
|
|
}
|