add sandboxSet Prometheus metrics

pkg/controller/sandboxset/metrics.go

@@ -0,0 +1,40 @@
+package sandboxset
+
+import (
+	"github.com/prometheus/client_golang/prometheus"
+	"sigs.k8s.io/controller-runtime/pkg/metrics"
+)
+
+var (
+	// SandboxSetReplicas tracks the total number of replicas in each SandboxSet
+	SandboxSetReplicas = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name: "sandboxset_replicas",
+			Help: "Current number of replicas in the SandboxSet",
+		},
+		[]string{"namespace", "name"},
+	)
+
+	// SandboxSetAvailableReplicas tracks the number of available replicas in each SandboxSet
+	SandboxSetAvailableReplicas = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name: "sandboxset_available_replicas",
+			Help: "Current number of available replicas in the SandboxSet",
+		},
+		[]string{"namespace", "name"},
+	)
+
+	// SandboxSetDesiredReplicas tracks the desired number of replicas in each SandboxSet
+	SandboxSetDesiredReplicas = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Name: "sandboxset_desired_replicas",
+			Help: "Desired number of replicas in the SandboxSet",
+		},
+		[]string{"namespace", "name"},
+	)
+)
+
+func init() {
+	// Register custom metrics with the global prometheus registry
+	metrics.Registry.MustRegister(SandboxSetReplicas, SandboxSetAvailableReplicas, SandboxSetDesiredReplicas)
+}

pkg/controller/sandboxset/sandboxset_controller.go

@@ -108,6 +108,10 @@ func (r *Reconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Resu
 		if apierrors.IsNotFound(err) {
 			scaleUpExpectation.DeleteExpectations(req.String())
 			scaleDownExpectation.DeleteExpectations(req.String())
+			// Remove metrics when sandboxset is deleted
+			SandboxSetReplicas.DeleteLabelValues(req.Namespace, req.Name)
+			SandboxSetAvailableReplicas.DeleteLabelValues(req.Namespace, req.Name)
+			SandboxSetDesiredReplicas.DeleteLabelValues(req.Namespace, req.Name)
 			return ctrl.Result{}, nil
 		}
 		return ctrl.Result{}, err
@@ -337,6 +341,10 @@ func (r *Reconciler) updateSandboxSetStatus(ctx context.Context, newStatus agent
 	err := r.Status().Update(ctx, clone)
 	if err == nil {
 		log.Info("update sandboxset status success", "status", utils.DumpJson(newStatus))
+		// Update metrics for availableReplicas and replicas
+		SandboxSetReplicas.WithLabelValues(sbs.Namespace, sbs.Name).Set(float64(newStatus.Replicas))
+		SandboxSetAvailableReplicas.WithLabelValues(sbs.Namespace, sbs.Name).Set(float64(newStatus.AvailableReplicas))
+		SandboxSetDesiredReplicas.WithLabelValues(sbs.Namespace, sbs.Name).Set(float64(sbs.Spec.Replicas))
 	} else {
 		log.Error(err, "update sandboxset status failed")
 	}