package metrics

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

const tidbscalerComponent = "tidbscaler"

var (
	ActionCounter = prometheus.NewCounterVec(
		prometheus.CounterOpts{
			Subsystem: tidbscalerComponent,
			Name:      "action_total",
			Help:      "Counter of scaler action",
		},
		[]string{"action", "min_pool_size", "max_pool_size", "pool_ns"},
	)

	AbnormalTiDBGauge = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Subsystem: tidbscalerComponent,
			Name:      "abnormal_tidb",
			Help:      "Gauge of abnormal tidb indentified by manager",
		},
		[]string{"pool_ns"},
	)
)

func init() {
	// Register custom metrics with the global prometheus registry
	metrics.Registry.MustRegister(ActionCounter)
	metrics.Registry.MustRegister(AbnormalTiDBGauge)
}