147 lines
3.8 KiB
Go
147 lines
3.8 KiB
Go
package metrics
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"lostak.dev/pve-exporter/proxmox"
|
|
)
|
|
|
|
// PVE subscription state collector.
|
|
type PveSubscriptionCollector struct {
|
|
apiClient *proxmox.PveApiClient // PVE API client instance.
|
|
registry *TTLRegistry // TTL metrics registry.
|
|
|
|
info *TTLGaugeVec // Node subscription info prometheus gauge.
|
|
status *TTLGaugeVec // Node subscription status prometheus gauge.
|
|
nextDueDate *TTLGaugeVec // Node subscription next due date prometheus gauge.
|
|
regDate *TTLGaugeVec // Node subscription registration date prometheus gauge.
|
|
sockets *TTLGaugeVec // Node subscription sockets count prometheus gauge.
|
|
}
|
|
|
|
// Create new instance of PVE cluster state collector.
|
|
func NewPveSubscriptionCollector(apiClient *proxmox.PveApiClient, registry *TTLRegistry) *PveSubscriptionCollector {
|
|
c := PveSubscriptionCollector{apiClient: apiClient}
|
|
c.registry = registry
|
|
|
|
// Node subscription info.
|
|
c.info = NewTTLGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Name: "pve_node_subscription_info",
|
|
Help: "Node subscription info.",
|
|
},
|
|
[]string{"cluster", "node", "productname", "serverid"},
|
|
1*time.Minute,
|
|
)
|
|
c.registry.Register(c.info)
|
|
|
|
// Node subscription status.
|
|
c.status = NewTTLGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Name: "pve_node_subscription_status",
|
|
Help: "Node subscription status.",
|
|
},
|
|
[]string{"cluster", "node"},
|
|
1*time.Minute,
|
|
)
|
|
c.registry.Register(c.status)
|
|
|
|
// Node subscription registration date.
|
|
c.regDate = NewTTLGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Name: "pve_node_subscription_regdate",
|
|
Help: "Node subscription registration date.",
|
|
},
|
|
[]string{"cluster", "node"},
|
|
1*time.Minute,
|
|
)
|
|
c.registry.Register(c.regDate)
|
|
|
|
// Node subscription next due date.
|
|
c.nextDueDate = NewTTLGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Name: "pve_node_subscription_nextduedate",
|
|
Help: "Node subscription next due date.",
|
|
},
|
|
[]string{"cluster", "node"},
|
|
1*time.Minute,
|
|
)
|
|
c.registry.Register(c.nextDueDate)
|
|
|
|
// Node subscription count of sockets.
|
|
c.sockets = NewTTLGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Name: "pve_node_subscription_sockets",
|
|
Help: "Node subscription count of sockets.",
|
|
},
|
|
[]string{"cluster", "node"},
|
|
1*time.Minute,
|
|
)
|
|
c.registry.Register(c.sockets)
|
|
|
|
return &c
|
|
}
|
|
|
|
// PveMetricsCollector interface implementation.
|
|
func (c *PveSubscriptionCollector) CollectMetrics() error {
|
|
cluster, err := c.apiClient.GetClusterStatus()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, node := range cluster.NodeStatuses {
|
|
labels := prometheus.Labels{
|
|
"cluster": cluster.GetClusterName(),
|
|
"node": node.Name,
|
|
}
|
|
|
|
// Node subscription.
|
|
subscription, err := c.apiClient.GetNodeSubscription(node.Name)
|
|
if err != nil {
|
|
return err
|
|
} else {
|
|
// Subscription info.
|
|
subsLabels := prometheus.Labels{
|
|
"cluster": cluster.GetClusterName(),
|
|
"node": node.Name,
|
|
"productname": subscription.ProductName,
|
|
"serverid": subscription.ServerID,
|
|
}
|
|
|
|
c.info.With(subsLabels).Set(1)
|
|
|
|
// Subscription state.
|
|
c.status.With(labels).Set(subscription.GetActiveNumeric())
|
|
|
|
// Subscription sockets count.
|
|
c.sockets.With(labels).Set(float64(subscription.Sockets))
|
|
|
|
// Subscription registered date.
|
|
if subscription.Registration != "" {
|
|
registrationTime, err := time.Parse("2006-01-02 15:04:05", subscription.Registration)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.regDate.With(labels).Set(float64(registrationTime.Unix()))
|
|
}
|
|
|
|
// Subscription due date.
|
|
if subscription.NextDueDate != "" {
|
|
nextDueTime, err := time.Parse("2006-01-02", subscription.NextDueDate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.nextDueDate.With(labels).Set(float64(nextDueTime.Unix()))
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// PveMetricsCollector interface implementation.
|
|
func (c *PveSubscriptionCollector) GetName() string {
|
|
return "Node Subscription"
|
|
}
|