50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/czerwonk/bird_exporter/bgp"
|
|
"github.com/czerwonk/bird_exporter/device"
|
|
"github.com/czerwonk/bird_exporter/direct"
|
|
"github.com/czerwonk/bird_exporter/kernel"
|
|
"github.com/czerwonk/bird_exporter/ospf"
|
|
"github.com/czerwonk/bird_exporter/protocol"
|
|
"github.com/czerwonk/bird_exporter/static"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
type MetricExporter interface {
|
|
Describe(ch chan<- *prometheus.Desc)
|
|
Export(p *protocol.Protocol, ch chan<- prometheus.Metric)
|
|
}
|
|
|
|
type MetricCollector struct {
|
|
protocols []*protocol.Protocol
|
|
exporters map[int]MetricExporter
|
|
}
|
|
|
|
func NewMetricCollectorForProtocols(protocols []*protocol.Protocol) *MetricCollector {
|
|
e := map[int]MetricExporter{
|
|
protocol.BGP: &bgp.BgpMetricExporter{},
|
|
protocol.Device: &device.DeviceMetricExporter{},
|
|
protocol.Direct: &direct.DirectMetricExporter{},
|
|
protocol.Kernel: &kernel.KernelMetricExporter{},
|
|
protocol.OSPF: &ospf.OspfMetricExporter{},
|
|
protocol.Static: &static.StaticMetricExporter{},
|
|
}
|
|
|
|
return &MetricCollector{protocols: protocols, exporters: e}
|
|
}
|
|
|
|
func (m *MetricCollector) Describe(ch chan<- *prometheus.Desc) {
|
|
for _, v := range m.exporters {
|
|
v.Describe(ch)
|
|
}
|
|
}
|
|
|
|
func (m *MetricCollector) Collect(ch chan<- prometheus.Metric) {
|
|
for _, p := range m.protocols {
|
|
if p.Proto != protocol.PROTO_UNKNOWN {
|
|
m.exporters[p.Proto].Export(p, ch)
|
|
}
|
|
}
|
|
}
|