target-allocator: Add support for scrape-classes

Signed-off-by: Christian Ciach <christian.ciach@gmail.com>

Author: ChristianCiach

cmd/otel-allocator/internal/config/config.go

@@ -89,6 +89,7 @@ type PrometheusCRConfig struct {
 	ProbeNamespaceSelector          *metav1.LabelSelector         `yaml:"probe_namespace_selector,omitempty"`
 	ScrapeProtocols                 []monitoringv1.ScrapeProtocol `yaml:"scrape_protocols,omitempty"`
 	ScrapeInterval                  model.Duration                `yaml:"scrape_interval,omitempty"`
+	ScrapeClasses                   []monitoringv1.ScrapeClass    `yaml:"scrape_classes,omitempty"`
 }
 
 type HTTPSServerConfig struct {

cmd/otel-allocator/internal/watcher/promOperator.go

@@ -84,6 +84,7 @@ func NewPrometheusCRWatcher(
 				ServiceDiscoveryRole:            &serviceDiscoveryRole,
 				Version:                         "2.55.1", // fix Prometheus version 2 to avoid generating incompatible config
 				ScrapeProtocols:                 cfg.PrometheusCR.ScrapeProtocols,
+				ScrapeClasses:                   cfg.PrometheusCR.ScrapeClasses,
 			},
 			EvaluationInterval: monitoringv1.Duration("30s"),
 		},

cmd/otel-allocator/internal/watcher/promOperator_test.go

@@ -44,6 +44,7 @@ func TestLoadConfig(t *testing.T) {
 		name            string
 		serviceMonitors []*monitoringv1.ServiceMonitor
 		podMonitors     []*monitoringv1.PodMonitor
+		scrapeClasses   []*monitoringv1.ScrapeClass
 		scrapeConfigs   []*promv1alpha1.ScrapeConfig
 		probes          []*monitoringv1.Probe
 		want            *promconfig.Config
@@ -84,6 +85,7 @@ func TestLoadConfig(t *testing.T) {
 					},
 				},
 			},
+
 			cfg: allocatorconfig.Config{
 				PrometheusCR: allocatorconfig.PrometheusCRConfig{
 					ServiceMonitorSelector: &metav1.LabelSelector{},
@@ -919,6 +921,68 @@ func TestLoadConfig(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "pod selector with scrape class",
+			podMonitors: []*monitoringv1.PodMonitor{
+				{
+					ObjectMeta: metav1.ObjectMeta{
+						Name:      "simple",
+						Namespace: namespace,
+					},
+					Spec: monitoringv1.PodMonitorSpec{
+						JobLabel:        "test",
+						ScrapeClassName: ptr.To("attach-node-metadata"),
+						PodMetricsEndpoints: []monitoringv1.PodMetricsEndpoint{
+							{
+								Port: &portName,
+							},
+						},
+					},
+				},
+			},
+			cfg: allocatorconfig.Config{
+				PrometheusCR: allocatorconfig.PrometheusCRConfig{
+					PodMonitorSelector: &metav1.LabelSelector{},
+					ScrapeClasses: []monitoringv1.ScrapeClass{
+						{
+							Name: "attach-node-metadata",
+							AttachMetadata: &monitoringv1.AttachMetadata{
+								Node: ptr.To(true),
+							},
+						},
+					},
+				},
+			},
+			want: &promconfig.Config{
+				ScrapeConfigs: []*promconfig.ScrapeConfig{
+					{
+						JobName:         "podMonitor/test/simple/0",
+						ScrapeInterval:  model.Duration(30 * time.Second),
+						ScrapeProtocols: promconfig.DefaultScrapeProtocols,
+						ScrapeTimeout:   model.Duration(10 * time.Second),
+						HonorTimestamps: true,
+						HonorLabels:     false,
+						Scheme:          "http",
+						MetricsPath:     "/metrics",
+						ServiceDiscoveryConfigs: []discovery.Config{
+							&kubeDiscovery.SDConfig{
+								Role: "pod",
+								NamespaceDiscovery: kubeDiscovery.NamespaceDiscovery{
+									Names:               []string{namespace},
+									IncludeOwnNamespace: false,
+								},
+								HTTPClientConfig: config.DefaultHTTPClientConfig,
+								AttachMetadata: kubeDiscovery.AttachMetadataConfig{
+									Node: true, // Added by scrape-class!
+								},
+							},
+						},
+						HTTPClientConfig:  config.DefaultHTTPClientConfig,
+						EnableCompression: true,
+					},
+				},
+			},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -1244,6 +1308,7 @@ func getTestPrometheusCRWatcher(
 				ProbeNamespaceSelector:          cfg.PrometheusCR.ProbeNamespaceSelector,
 				ScrapeConfigSelector:            cfg.PrometheusCR.ScrapeConfigSelector,
 				ScrapeConfigNamespaceSelector:   cfg.PrometheusCR.ScrapeConfigNamespaceSelector,
+				ScrapeClasses:                   cfg.PrometheusCR.ScrapeClasses,
 				ServiceDiscoveryRole:            &serviceDiscoveryRole,
 				Version:                         "2.55.1",
 			},