package metrics

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

const ExportComponent = "export"

var (
	ExportRUGauge = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Namespace: ExportComponent,
			Name:      "ru_cost_counter",
			Help:      "Total number of export ru.",
		},
		[]string{"cluster_id", "export_id", "keyspace_id"},
	)
	FailedGetExportSizeCounter = prometheus.NewCounterVec(
		prometheus.CounterOpts{
			Namespace: ExportComponent,
			Name:      "failed_get_export_size_counter",
			Help:      "Counter of failed to get export size.",
		},
		[]string{"cluster_id", "export_id", "keyspace_id"},
	)
)

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