68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package metrics
|
|
|
|
import (
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
"lostak.dev/pve-exporter/proxmox"
|
|
)
|
|
|
|
// PVE SDN state collector.
|
|
type PveSdnCollector struct {
|
|
apiClient *proxmox.PveApiClient // PVE API client instance.
|
|
|
|
state *prometheus.GaugeVec // SDN state prometheus gauge.
|
|
}
|
|
|
|
// Create new instance of PVE SDN collector.
|
|
func NewPveSdnCollector(apiClient *proxmox.PveApiClient) *PveSdnCollector {
|
|
c := PveSdnCollector{apiClient: apiClient}
|
|
|
|
// SDN Up state.
|
|
c.state = promauto.NewGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Name: "pve_sdn_state",
|
|
Help: "Node software defined network state.",
|
|
},
|
|
[]string{"cluster", "node", "sdn", "sdn_id"},
|
|
)
|
|
|
|
return &c
|
|
}
|
|
|
|
// PveMetricsCollector interface implementation.
|
|
func (c *PveSdnCollector) CollectMetrics() error {
|
|
cluster, err := c.apiClient.GetClusterStatus()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
resources, err := c.apiClient.GetClusterResources()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.state.Reset()
|
|
|
|
for _, node := range cluster.NodeStatuses {
|
|
sdns := resources.FindNodeSDN(node.Name)
|
|
if len(*sdns) > 0 {
|
|
for _, sdn := range *sdns {
|
|
labels := prometheus.Labels{
|
|
"cluster": cluster.GetClusterName(),
|
|
"node": node.Name,
|
|
"sdn": sdn.SDN,
|
|
"sdn_id": sdn.ID,
|
|
}
|
|
|
|
c.state.With(labels).Set(sdn.GetStatusNumeric())
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// PveMetricsCollector interface implementation.
|
|
func (c *PveSdnCollector) GetName() string {
|
|
return "SDN"
|
|
}
|