hcloud-pricing-exporter/fetcher/fetcher.go

package fetcher

import (
	"context"
	"fmt"

	"github.com/hetznercloud/hcloud-go/v2/hcloud"
	"github.com/prometheus/client_golang/prometheus"
)

var (
	ctx = context.Background()
)

// Fetcher defines a common interface for types that fetch pricing data from the HCloud API.
type Fetcher interface {
	// GetHourly returns the prometheus collector that collects pricing data for hourly expenses.
	GetHourly() *prometheus.GaugeVec
	// GetMonthly returns the prometheus collector that collects pricing data for monthly expenses.
	GetMonthly() *prometheus.GaugeVec
	// Run executes a new data fetching cycle and updates the prometheus exposed collectors.
	Run(*hcloud.Client) error
}

type baseFetcher struct {
	pricing          *PriceProvider
	hourly           *prometheus.GaugeVec
	monthly          *prometheus.GaugeVec
	additionalLabels []string
}

func (fetcher baseFetcher) GetHourly() *prometheus.GaugeVec {
	return fetcher.hourly
}

func (fetcher baseFetcher) GetMonthly() *prometheus.GaugeVec {
	return fetcher.monthly
}

func newBase(pricing *PriceProvider, resource string, baselabels []string, additionalLabels ...string) *baseFetcher {
	labels := append([]string{"name"}, baselabels...)
	labels = append(labels, additionalLabels...)

	hourlyGaugeOpts := prometheus.GaugeOpts{
		Namespace: "hcloud",
		Subsystem: "pricing",
		Name:      fmt.Sprintf("%s_hourly", resource),
		Help:      fmt.Sprintf("The cost of the resource %s per hour", resource),
	}
	monthlyGaugeOpts := prometheus.GaugeOpts{
		Namespace: "hcloud",
		Subsystem: "pricing",
		Name:      fmt.Sprintf("%s_monthly", resource),
		Help:      fmt.Sprintf("The cost of the resource %s per month", resource),
	}

	return &baseFetcher{
		pricing:          pricing,
		hourly:           prometheus.NewGaugeVec(hourlyGaugeOpts, labels),
		monthly:          prometheus.NewGaugeVec(monthlyGaugeOpts, labels),
		additionalLabels: additionalLabels,
	}
}

// Fetchers defines a type for a slice of fetchers that should be handled together.
type Fetchers []Fetcher

// RegisterCollectors registers all collectors of the contained fetchers into the passed registry.
func (fetchers Fetchers) RegisterCollectors(registry *prometheus.Registry) {
	for _, fetcher := range fetchers {
		registry.MustRegister(
			fetcher.GetHourly(),
			fetcher.GetMonthly(),
		)
	}
}

// Run executes all contained fetchers and returns a single error, even when multiple failures occurred.
func (fetchers Fetchers) Run(client *hcloud.Client) error {
	errors := prometheus.MultiError{}
	for _, fetcher := range fetchers {
		fetcher.GetHourly().Reset()
		fetcher.GetMonthly().Reset()

		if err := fetcher.Run(client); err != nil {
			errors.Append(err)
		}
	}

	if len(errors) > 0 {
		return errors
	}
	return nil
}

// MustRun executes all contained fetchers and panics if any of them threw an error.
func (fetchers Fetchers) MustRun(client *hcloud.Client) {
	if err := fetchers.Run(client); err != nil {
		panic(err)
	}
}