17277209982

Committed 27 Aug 2025 07:52PM UTC coverage: 13.246% (+8.4%) from 4.874%

Build # 17277209982

Build Type

Pull #3

github

Committed by

stillya

Commit Message

added prom integration

Pull Request Pull Request #3: Added metrics support

Run Details

78 of 293 new or added lines in 11 files covered. (26.62%)

5 existing lines in 4 files now uncovered.

111 of 838 relevant lines covered (13.25%)

0.14 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/pkg/metrics/collector.go

package metrics

import (
        "context"
        log "log/slog"

        "github.com/prometheus/client_golang/prometheus"
        "github.com/stillya/wg-relay/pkg/maps/metricsmap"
)

type MetricCollectorSource interface {
        Collect(ctx context.Context) ([]metricsmap.MetricData, error)
        Name() string
}

type BpfCollector struct {
        source MetricCollectorSource
        mode   string

        packetsDesc *prometheus.Desc
        bytesDesc   *prometheus.Desc
}

func NewBpfCollector(source MetricCollectorSource, mode string) *BpfCollector {
        return &BpfCollector{
                source: source,
                mode:   mode,
                packetsDesc: prometheus.NewDesc(
                        "wg_relay_packets",
                        "Current total number of WireGuard packets processed",
                        []string{"mode", "direction", "reason"},
                        nil,
                ),
                bytesDesc: prometheus.NewDesc(
                        "wg_relay_bytes",
                        "Current total bytes of WireGuard packets processed",
                        []string{"mode", "direction", "reason"},
                        nil,
                ),
        }
}

func (c *BpfCollector) Describe(ch chan<- *prometheus.Desc) {
        ch <- c.packetsDesc
        ch <- c.bytesDesc
}

func (c *BpfCollector) Collect(ch chan<- prometheus.Metric) {
        ctx := context.Background()
        metricsData, err := c.source.Collect(ctx)
        if err != nil {
                log.Error("Failed to collect metrics", "error", err)
                return
        }

        for _, metric := range metricsData {
                dirLabel := metricsmap.DirectionToString(metric.Key.Dir)
                reasonLabel := metricsmap.ReasonToString(metric.Key.Reason)

                ch <- prometheus.MustNewConstMetric(
                        c.packetsDesc,
                        prometheus.GaugeValue,
                        float64(metric.Value.Packets),
                        c.mode, dirLabel, reasonLabel,
                )

                ch <- prometheus.MustNewConstMetric(
                        c.bytesDesc,
                        prometheus.GaugeValue,
                        float64(metric.Value.Bytes),
                        c.mode, dirLabel, reasonLabel,
                )
        }
}

1	package metrics
2
3	import (
4	"context"
5	log "log/slog"
6
7	"github.com/prometheus/client_golang/prometheus"
8	"github.com/stillya/wg-relay/pkg/maps/metricsmap"
9	)
10
11	type MetricCollectorSource interface {
12	Collect(ctx context.Context) ([]metricsmap.MetricData, error)
13	Name() string
14	}
15
16	type BpfCollector struct {
17	source MetricCollectorSource
18	mode string
19
20	packetsDesc *prometheus.Desc
21	bytesDesc *prometheus.Desc
22	}
23
NEW 24	func NewBpfCollector(source MetricCollectorSource, mode string) *BpfCollector {	×
NEW 25	return &BpfCollector{	×
NEW 26	source: source,	×
NEW 27	mode: mode,	×
NEW 28	packetsDesc: prometheus.NewDesc(	×
NEW 29	"wg_relay_packets",	×
NEW 30	"Current total number of WireGuard packets processed",	×
NEW 31	[]string{"mode", "direction", "reason"},	×
NEW 32	nil,	×
NEW 33	),	×
NEW 34	bytesDesc: prometheus.NewDesc(	×
NEW 35	"wg_relay_bytes",	×
NEW 36	"Current total bytes of WireGuard packets processed",	×
NEW 37	[]string{"mode", "direction", "reason"},	×
NEW 38	nil,	×
NEW 39	),	×
NEW 40	}	×
NEW 41	}	×
42
NEW 43	func (c BpfCollector) Describe(ch chan<- prometheus.Desc) {	×
NEW 44	ch <- c.packetsDesc	×
NEW 45	ch <- c.bytesDesc	×
NEW 46	}	×
47
NEW 48	func (c *BpfCollector) Collect(ch chan<- prometheus.Metric) {	×
NEW 49	ctx := context.Background()	×
NEW 50	metricsData, err := c.source.Collect(ctx)	×
NEW 51	if err != nil {	×
NEW 52	log.Error("Failed to collect metrics", "error", err)	×
NEW 53	return	×
NEW 54	}	×
55
NEW 56	for _, metric := range metricsData {	×
NEW 57	dirLabel := metricsmap.DirectionToString(metric.Key.Dir)	×
NEW 58	reasonLabel := metricsmap.ReasonToString(metric.Key.Reason)	×
NEW 59		×
NEW 60	ch <- prometheus.MustNewConstMetric(	×
NEW 61	c.packetsDesc,	×
NEW 62	prometheus.GaugeValue,	×
NEW 63	float64(metric.Value.Packets),	×
NEW 64	c.mode, dirLabel, reasonLabel,	×
NEW 65	)	×
NEW 66		×
NEW 67	ch <- prometheus.MustNewConstMetric(	×
NEW 68	c.bytesDesc,	×
NEW 69	prometheus.GaugeValue,	×
NEW 70	float64(metric.Value.Bytes),	×
NEW 71	c.mode, dirLabel, reasonLabel,	×
NEW 72	)	×
NEW 73	}	×
74	}

stillya / wg-relay / 17277209982

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous