feat(tracing): add operator.impvm.reconcile and operator.impvm.schedule spans

Author: syscod3

internal/controller/impvm_controller.go

@@ -20,6 +20,9 @@ import (
 	"context"
 	"time"
 
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/trace"
 	corev1 "k8s.io/api/core/v1"
 	apierrors "k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -34,6 +37,7 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/reconcile"
 
 	impdevv1alpha1 "github.com/syscode-labs/imp/api/v1alpha1"
+	"github.com/syscode-labs/imp/internal/tracing"
 )
 
 const (
@@ -58,14 +62,22 @@ type ImpVMReconciler struct {
 // +kubebuilder:rbac:groups=imp.dev,resources=clusterimpconfigs,verbs=get;list;watch
 // +kubebuilder:rbac:groups=core,resources=events,verbs=create;patch
 
-func (r *ImpVMReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
+func (r *ImpVMReconciler) Reconcile(ctx context.Context, req ctrl.Request) (result ctrl.Result, err error) {
 	log := logf.FromContext(ctx)
 
 	vm := &impdevv1alpha1.ImpVM{}
 	if err := r.Get(ctx, req.NamespacedName, vm); err != nil {
 		return ctrl.Result{}, client.IgnoreNotFound(err)
 	}
 
+	ctx, span := otel.Tracer("imp.operator").Start(ctx, "operator.impvm.reconcile",
+		trace.WithAttributes(
+			attribute.String("vm.name", req.Name),
+			attribute.String("vm.namespace", req.Namespace),
+			attribute.String("vm.phase", string(vm.Status.Phase)),
+		))
+	defer func() { tracing.RecordError(span, err); span.End() }()
+
 	// 1. Handle deletion
 	if !vm.DeletionTimestamp.IsZero() {
 		return r.handleDeletion(ctx, vm)
@@ -103,6 +115,7 @@ func (r *ImpVMReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl
 		// Assign to node
 		specPatch := client.MergeFrom(vm.DeepCopy())
 		vm.Spec.NodeName = nodeName
+		tracing.InjectToVM(ctx, vm, span)
 		if err := r.Patch(ctx, vm, specPatch); err != nil {
 			return ctrl.Result{}, err
 		}

internal/controller/impvm_scheduler.go

@@ -21,12 +21,16 @@ import (
 	"sort"
 	"strings"
 
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/trace"
 	corev1 "k8s.io/api/core/v1"
 	apierrors "k8s.io/apimachinery/pkg/api/errors"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	logf "sigs.k8s.io/controller-runtime/pkg/log"
 
 	impdevv1alpha1 "github.com/syscode-labs/imp/api/v1alpha1"
+	"github.com/syscode-labs/imp/internal/tracing"
 )
 
 const labelImpEnabled = "imp/enabled"
@@ -60,7 +64,14 @@ func sumUsedResources(ctx context.Context, c client.Client, vms []impdevv1alpha1
 
 // schedule selects a node for vm using a capacity-aware least-loaded strategy.
 // Returns "" and no error when no suitable node is available.
-func (r *ImpVMReconciler) schedule(ctx context.Context, vm *impdevv1alpha1.ImpVM) (string, error) {
+func (r *ImpVMReconciler) schedule(ctx context.Context, vm *impdevv1alpha1.ImpVM) (nodeName string, err error) {
+	ctx, span := otel.Tracer("imp.operator").Start(ctx, "operator.impvm.schedule",
+		trace.WithAttributes(
+			attribute.String("vm.name", vm.Name),
+			attribute.String("vm.namespace", vm.Namespace),
+		))
+	defer func() { tracing.RecordError(span, err); span.End() }()
+
 	log := logf.FromContext(ctx)
 
 	// 1. List nodes with imp/enabled=true