cost-model/modules/prometheus-source/pkg/prom/validate.go

package prom

import (
	"fmt"

	prometheus "github.com/prometheus/client_golang/api"
)

const UpQuery = "up"

// PrometheusMetadata represents a validation result for prometheus/thanos running
// opencost.
type PrometheusMetadata struct {
	Running            bool `json:"running"`
	KubecostDataExists bool `json:"kubecostDataExists"`
}

// Validate tells the model what data prometheus has on it.
func Validate(cli prometheus.Client, config *OpenCostPrometheusConfig) (*PrometheusMetadata, error) {
	if IsThanos(cli) {
		return validate(cli, validationQueryFor(config), config)
	}

	return validate(cli, validationQueryFor(config), config)
}

func validationQueryFor(config *OpenCostPrometheusConfig) string {
	if config.Offset != "" {
		return fmt.Sprintf("%s offset %s", UpQuery, config.Offset)
	}

	return UpQuery
}

// validate executes the prometheus query against the provided client.
func validate(cli prometheus.Client, q string, config *OpenCostPrometheusConfig) (*PrometheusMetadata, error) {
	ctx := NewContext(cli, config)

	resUp, _, err := ctx.QuerySync(q)
	if err != nil {
		return &PrometheusMetadata{
			Running:            false,
			KubecostDataExists: false,
		}, err
	}

	if len(resUp) == 0 {
		return &PrometheusMetadata{
			Running:            false,
			KubecostDataExists: false,
		}, fmt.Errorf("no running jobs on Prometheus at %s", ctx.QueryURL().Path)
	}

	for _, result := range resUp {
		job, err := result.GetString("job")
		if err != nil {
			return &PrometheusMetadata{
				Running:            false,
				KubecostDataExists: false,
			}, fmt.Errorf("up query does not have job names")
		}

		if job == config.JobName {
			return &PrometheusMetadata{
				Running:            true,
				KubecostDataExists: true,
			}, err
		}
	}

	return &PrometheusMetadata{
		Running:            true,
		KubecostDataExists: false,
	}, nil
}