process: Use registry collector for V1 data (#1814)

Signed-off-by: Jan-Otto Kröpke <mail@jkroepke.de>
This commit is contained in:
Jan-Otto Kröpke
2024-12-21 22:58:47 +01:00
committed by GitHub
parent 39c929eefe
commit a9f8b3b722
158 changed files with 7793 additions and 7748 deletions

View File

@@ -25,7 +25,7 @@ import (
"github.com/alecthomas/kingpin/v2"
"github.com/prometheus-community/windows_exporter/internal/headers/iphlpapi"
"github.com/prometheus-community/windows_exporter/internal/mi"
"github.com/prometheus-community/windows_exporter/internal/perfdata"
"github.com/prometheus-community/windows_exporter/internal/pdh"
"github.com/prometheus-community/windows_exporter/internal/types"
"github.com/prometheus/client_golang/prometheus"
"golang.org/x/sys/windows"
@@ -49,8 +49,10 @@ var ConfigDefaults = Config{
type Collector struct {
config Config
perfDataCollector4 *perfdata.Collector
perfDataCollector6 *perfdata.Collector
perfDataCollector4 *pdh.Collector
perfDataCollector6 *pdh.Collector
perfDataObject4 []perfDataCounterValues
perfDataObject6 []perfDataCounterValues
connectionFailures *prometheus.Desc
connectionsActive *prometheus.Desc
@@ -116,26 +118,14 @@ func (c *Collector) Close() error {
}
func (c *Collector) Build(_ *slog.Logger, _ *mi.Session) error {
counters := []string{
connectionFailures,
connectionsActive,
connectionsEstablished,
connectionsPassive,
connectionsReset,
segmentsPerSec,
segmentsReceivedPerSec,
segmentsRetransmittedPerSec,
segmentsSentPerSec,
}
var err error
c.perfDataCollector4, err = perfdata.NewCollector("TCPv4", nil, counters)
c.perfDataCollector4, err = pdh.NewCollector[perfDataCounterValues]("TCPv4", nil)
if err != nil {
return fmt.Errorf("failed to create TCPv4 collector: %w", err)
}
c.perfDataCollector6, err = perfdata.NewCollector("TCPv6", nil, counters)
c.perfDataCollector6, err = pdh.NewCollector[perfDataCounterValues]("TCPv6", nil)
if err != nil {
return fmt.Errorf("failed to create TCPv6 collector: %w", err)
}
@@ -224,84 +214,76 @@ func (c *Collector) Collect(ch chan<- prometheus.Metric) error {
}
func (c *Collector) collect(ch chan<- prometheus.Metric) error {
data, err := c.perfDataCollector4.Collect()
err := c.perfDataCollector4.Collect(&c.perfDataObject4)
if err != nil {
return fmt.Errorf("failed to collect TCPv4 metrics: %w", err)
return fmt.Errorf("failed to collect TCPv4 metrics[0]. %w", err)
}
if _, ok := data[perfdata.InstanceEmpty]; !ok {
return errors.New("no data for TCPv4")
}
c.writeTCPCounters(ch, c.perfDataObject4, []string{"ipv4"})
c.writeTCPCounters(ch, data[perfdata.InstanceEmpty], []string{"ipv4"})
data, err = c.perfDataCollector6.Collect()
err = c.perfDataCollector6.Collect(&c.perfDataObject6)
if err != nil {
return fmt.Errorf("failed to collect TCPv6 metrics: %w", err)
return fmt.Errorf("failed to collect TCPv6 metrics[0]. %w", err)
}
if _, ok := data[perfdata.InstanceEmpty]; !ok {
return errors.New("no data for TCPv6")
}
c.writeTCPCounters(ch, data[perfdata.InstanceEmpty], []string{"ipv6"})
c.writeTCPCounters(ch, c.perfDataObject6, []string{"ipv6"})
return nil
}
func (c *Collector) writeTCPCounters(ch chan<- prometheus.Metric, metrics map[string]perfdata.CounterValue, labels []string) {
func (c *Collector) writeTCPCounters(ch chan<- prometheus.Metric, metrics []perfDataCounterValues, labels []string) {
ch <- prometheus.MustNewConstMetric(
c.connectionFailures,
prometheus.CounterValue,
metrics[connectionFailures].FirstValue,
metrics[0].ConnectionFailures,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.connectionsActive,
prometheus.CounterValue,
metrics[connectionsActive].FirstValue,
metrics[0].ConnectionsActive,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.connectionsEstablished,
prometheus.GaugeValue,
metrics[connectionsEstablished].FirstValue,
metrics[0].ConnectionsEstablished,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.connectionsPassive,
prometheus.CounterValue,
metrics[connectionsPassive].FirstValue,
metrics[0].ConnectionsPassive,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.connectionsReset,
prometheus.CounterValue,
metrics[connectionsReset].FirstValue,
metrics[0].ConnectionsReset,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.segmentsTotal,
prometheus.CounterValue,
metrics[segmentsPerSec].FirstValue,
metrics[0].SegmentsPerSec,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.segmentsReceivedTotal,
prometheus.CounterValue,
metrics[segmentsReceivedPerSec].FirstValue,
metrics[0].SegmentsReceivedPerSec,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.segmentsRetransmittedTotal,
prometheus.CounterValue,
metrics[segmentsRetransmittedPerSec].FirstValue,
metrics[0].SegmentsRetransmittedPerSec,
labels...,
)
ch <- prometheus.MustNewConstMetric(
c.segmentsSentTotal,
prometheus.CounterValue,
metrics[segmentsSentPerSec].FirstValue,
metrics[0].SegmentsSentPerSec,
labels...,
)
}

View File

@@ -19,14 +19,14 @@ package tcp
// - https://msdn.microsoft.com/en-us/library/aa394341(v=vs.85).aspx
// The TCPv6 performance object uses the same fields.
// https://learn.microsoft.com/en-us/dotnet/api/system.net.networkinformation.tcpstate?view=net-8.0.
const (
connectionFailures = "Connection Failures"
connectionsActive = "Connections Active"
connectionsEstablished = "Connections Established"
connectionsPassive = "Connections Passive"
connectionsReset = "Connections Reset"
segmentsPerSec = "Segments/sec"
segmentsReceivedPerSec = "Segments Received/sec"
segmentsRetransmittedPerSec = "Segments Retransmitted/sec"
segmentsSentPerSec = "Segments Sent/sec"
)
type perfDataCounterValues struct {
ConnectionFailures float64 `perfdata:"Connection Failures"`
ConnectionsActive float64 `perfdata:"Connections Active"`
ConnectionsEstablished float64 `perfdata:"Connections Established"`
ConnectionsPassive float64 `perfdata:"Connections Passive"`
ConnectionsReset float64 `perfdata:"Connections Reset"`
SegmentsPerSec float64 `perfdata:"Segments/sec"`
SegmentsReceivedPerSec float64 `perfdata:"Segments Received/sec"`
SegmentsRetransmittedPerSec float64 `perfdata:"Segments Retransmitted/sec"`
SegmentsSentPerSec float64 `perfdata:"Segments Sent/sec"`
}