package metrics

import (
	"github.com/prometheus/client_golang/prometheus"
	"sigs.k8s.io/controller-runtime/pkg/metrics"
)

const tidbautoscalingComponent = "tidbautoscaling"

var (
	ScaleCounter = prometheus.NewCounterVec(
		prometheus.CounterOpts{
			Subsystem: tidbautoscalingComponent,
			Name:      "scale_total",
			Help:      "Counter of scale",
		},
		[]string{"direction", "pool_ns"},
	)
	TidbPodNumGauge = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Subsystem: tidbautoscalingComponent,
			Name:      "tidbpod_num",
			Help:      "Gauge of tidb pods number",
		},
		[]string{"pool_ns"},
	)
	ScaleDownTokenNumGauge = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Subsystem: tidbautoscalingComponent,
			Name:      "scale_down_token_num",
			Help:      "Gauge of scale down token number",
		},
		[]string{"pool_ns"},
	)
	ActiveTokenNumGauge = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Subsystem: tidbautoscalingComponent,
			Name:      "active_token_num",
			Help:      "Gauge of active token number",
		},
		[]string{"pool_ns"},
	)
)

func init() {
	// Register custom metrics with the global prometheus registry
	metrics.Registry.MustRegister(ScaleCounter)
	metrics.Registry.MustRegister(TidbPodNumGauge)
	metrics.Registry.MustRegister(ScaleDownTokenNumGauge)
	metrics.Registry.MustRegister(ActiveTokenNumGauge)
}