Adding upstream version 1.34.4.

Signed-off-by: Daniel Baumann <daniel@debian.org>

plugins/outputs/prometheus_client/v2/collector.go

@@ -0,0 +1,98 @@
+package v2
+
+import (
+	"sync"
+	"time"
+
+	"github.com/prometheus/client_golang/prometheus"
+	dto "github.com/prometheus/client_model/go"
+
+	"github.com/influxdata/telegraf"
+	serializers_prometheus "github.com/influxdata/telegraf/plugins/serializers/prometheus"
+)
+
+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           *serializers_prometheus.Collection
+}
+
+func NewCollector(expire time.Duration, stringsAsLabel, exportTimestamp bool, typeMapping serializers_prometheus.MetricTypes) *Collector {
+	cfg := serializers_prometheus.FormatConfig{
+		StringAsLabel:   stringsAsLabel,
+		ExportTimestamp: exportTimestamp,
+		TypeMappings:    typeMapping,
+	}
+
+	return &Collector{
+		expireDuration: expire,
+		coll:           serializers_prometheus.NewCollection(cfg),
+	}
+}
+
+func (*Collector) Describe(_ 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.
+}
+
+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, time.Now())
+	}
+
+	// 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
+}