package metrics

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

const tidbPodEvicterComponent = "tidbpodevicter"

var (
	TryEvictCounter = prometheus.NewCounterVec(
		prometheus.CounterOpts{
			Subsystem: tidbPodEvicterComponent,
			Name:      "try_evict_total",
			Help:      "Counter of tidbpodevicter try evict",
		},
		[]string{"evict_type", "node", "nodegroup"},
	)
	EvictCounter = prometheus.NewCounterVec(
		prometheus.CounterOpts{
			Subsystem: tidbPodEvicterComponent,
			Name:      "evict_total",
			Help:      "Counter of tidbpodevicter evict",
		},
		[]string{"evict_type", "node", "nodegroup"},
	)
	ForceEvictCounter = prometheus.NewCounterVec(
		prometheus.CounterOpts{
			Subsystem: tidbPodEvicterComponent,
			Name:      "force_evict_total",
			Help:      "Counter of tidbpodevicter force evict",
		},
		[]string{"evict_type", "node", "nodegroup"},
	)
)

func init() {
	// Register custom metrics with the global prometheus registry
	metrics.Registry.MustRegister(TryEvictCounter)
	metrics.Registry.MustRegister(EvictCounter)
	metrics.Registry.MustRegister(ForceEvictCounter)
}