cinder.go

package main

import (
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/common/log"
	"gopkg.in/niedbalski/goose.v3/cinder"
	"gopkg.in/niedbalski/goose.v3/client"
	"net/http"
	"net/url"
)

type CinderExporter struct {
	BaseOpenStackExporter
	Client *cinder.Client
}

var defaultCinderMetrics = []Metric{
	{Name: "volumes"},
	{Name: "snapshots"},
	{Name: "agent_state", Labels: []string{"hostname", "service", "adminState", "zone"}},
}

func NewCinderExporter(client client.AuthenticatingClient, prefix string, config *Cloud) (*CinderExporter, error) {
	endpoint := client.EndpointsForRegion(config.Region)["volumev3"]
	endpointUrl, err := url.Parse(endpoint)

	if err != nil {
		return nil, err
	}

	exporter := CinderExporter{BaseOpenStackExporter{
		Name:   "cinder",
		Prefix: prefix,
		Config: config,
	}, cinder.NewClient(client.TenantId(), endpointUrl,
		cinder.SetAuthHeaderFn(client.Token,
			http.DefaultClient.Do),
	)}

	for _, metric := range defaultCinderMetrics {
		exporter.AddMetric(metric.Name, metric.Labels, nil)
	}

	return &exporter, nil
}

func (exporter *CinderExporter) Describe(ch chan<- *prometheus.Desc) {
	for _, metric := range exporter.Metrics {
		ch <- metric
	}
}

func (exporter *CinderExporter) Collect(ch chan<- prometheus.Metric) {
	log.Infoln("Fetching volumes info")
	volumes, err := exporter.Client.GetVolumesSimple()
	if err != nil {
		log.Errorf("%s", err)
	}

	log.Infoln("Fetching services state information")
	services, err := exporter.Client.GetServices()
	if err != nil {
		log.Errorf("%s", err)
	}

	for _, service := range services.Services {
		var state int = 0
		if service.State == "up" {
			state = 1
		}
		ch <- prometheus.MustNewConstMetric(exporter.Metrics["agent_state"],
			prometheus.CounterValue, float64(state), service.Host, service.Binary, service.Status, service.Zone)
	}

	log.Infoln("Fetching volumes information")
	ch <- prometheus.MustNewConstMetric(exporter.Metrics["volumes"],
		prometheus.GaugeValue, float64(len(volumes.Volumes)))

	log.Infoln("Fetching snapshots information")
	snapshots, err := exporter.Client.GetSnapshotsSimple()
	if err != nil {
		log.Errorf("%s", err)
	}

	ch <- prometheus.MustNewConstMetric(exporter.Metrics["snapshots"],
		prometheus.GaugeValue, float64(len(snapshots.Snapshots)))
}