Initial commit

metrics/pve_cluster_state_collector.go

@@ -0,0 +1,59 @@
+package metrics
+
+import (
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promauto"
+	"lostak.dev/pve-exporter/proxmox"
+)
+
+// PVE cluster state collector.
+type PveClusterStateCollector struct {
+	apiClient *proxmox.PveApiClient // PVE API client instance.
+
+	nodes   *prometheus.GaugeVec // Count of nodes prometheus gauge.
+	quorate *prometheus.GaugeVec // Cluster quorum state prometheus gauge.
+}
+
+// Create new instance of PVE cluster state collector.
+func NewPveClusterStateCollector(apiClient *proxmox.PveApiClient) *PveClusterStateCollector {
+	c := PveClusterStateCollector{apiClient: apiClient}
+
+	// Cluster meta gauge.
+	c.nodes = promauto.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name: "pve_cluster_nodes",
+			Help: "Cluster nodes count.",
+		},
+		[]string{"cluster"},
+	)
+
+	// Cluster quorate gauge.
+	c.quorate = promauto.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name: "pve_cluster_quorate",
+			Help: "Cluster quorum state.",
+		},
+		[]string{"cluster"},
+	)
+
+	return &c
+}
+
+// PveMetricsCollector interface implementation.
+func (c *PveClusterStateCollector) CollectMetrics() error {
+	cluster, err := c.apiClient.GetClusterStatus()
+	if err != nil {
+		return err
+	}
+
+	l := prometheus.Labels{"cluster": cluster.Name}
+	c.nodes.With(l).Set(float64(cluster.Nodes))
+	c.quorate.With(l).Set(float64(cluster.Quorate))
+
+	return nil
+}
+
+// PveMetricsCollector interface implementation.
+func (c *PveClusterStateCollector) GetName() string {
+	return "Cluster State"
+}