upda/server/service_prometheus.go

package server

import (
	"github.com/Depado/ginprom"
	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
)

type prometheusService struct {
	router     *gin.Engine
	prometheus *ginprom.Prometheus
	config     *prometheusConfig
}

func newPrometheusService(r *gin.Engine, c *prometheusConfig) *prometheusService {
	var p *ginprom.Prometheus

	if c.enabled {
		if c.secureTokenEnabled {
			p = ginprom.New(
				ginprom.Engine(r),
				ginprom.Namespace(Name),
				ginprom.Subsystem(""),
				ginprom.Path(c.path),
				ginprom.Ignore(c.path),
				ginprom.Token(c.secureToken),
			)
		} else {
			p = ginprom.New(
				ginprom.Engine(r),
				ginprom.Namespace(Name),
				ginprom.Subsystem(""),
				ginprom.Ignore(c.path),
				ginprom.Path(c.path),
			)
		}
	}

	return &prometheusService{
		prometheus: p,
		config:     c,
	}
}

func (s *prometheusService) init() {
	if !s.config.enabled {
		return
	}

	var err error

	err = s.registerGaugeNoLabels(metricUpdatesTotal, metricUpdatesTotalHelp)
	err = s.registerGaugeNoLabels(metricUpdatesPending, metricUpdatesPendingHelp)
	err = s.registerGaugeNoLabels(metricUpdatesIgnored, metricUpdatesIgnoredHelp)
	err = s.registerGaugeNoLabels(metricUpdatesApproved, metricUpdatesApprovedHelp)
	err = s.registerGauge(metricUpdates, metricUpdatesHelp, []string{"application", "provider", "host"})

	err = s.registerGaugeNoLabels(metricWebhooks, metricWebhooksHelp)
	err = s.registerGaugeNoLabels(metricEvents, metricEventsHelp)

	if err != nil {
		zap.L().Sugar().Fatalf("Cannot initialize service. Reason: %v", err)
	}
}

func (s *prometheusService) registerGaugeNoLabels(name string, help string) error {
	return s.registerGauge(name, help, make([]string, 0))
}

func (s *prometheusService) registerGauge(name string, help string, labels []string) error {
	if !s.config.enabled {
		return nil
	}

	if name == "" || help == "" {
		return errorValidationNotBlank
	}

	s.prometheus.AddCustomGauge(name, help, labels)

	return nil
}

func (s *prometheusService) registerCounterNoLabels(name string, help string) error {
	return s.registerCounter(name, help, make([]string, 0))
}

func (s *prometheusService) registerCounter(name string, help string, labels []string) error {
	if !s.config.enabled {
		return nil
	}

	if name == "" || help == "" {
		return errorValidationNotBlank
	}

	s.prometheus.AddCustomCounter(name, help, labels)

	return nil
}

func (s *prometheusService) setGaugeNoLabels(name string, value float64) error {
	return s.setGauge(name, make([]string, 0), value)
}

func (s *prometheusService) setGauge(name string, labelValues []string, value float64) error {
	if !s.config.enabled {
		return nil
	}

	if name == "" {
		return errorValidationNotBlank
	}

	if err := s.prometheus.SetGaugeValue(name, labelValues, value); err != nil {
		return err
	}

	return nil
}

func (s *prometheusService) increaseCounterNoLabels(name string) error {
	return s.increaseCounter(name, make([]string, 0))
}

func (s *prometheusService) increaseCounter(name string, labelValues []string) error {
	if !s.config.enabled {
		return nil
	}

	if name == "" {
		return errorValidationNotBlank
	}

	if err := s.prometheus.IncrementCounterValue(name, labelValues); err != nil {
		return err
	}

	return nil
}