Ci neuron

receiver/awscontainerinsightreceiver/config.go

@@ -61,7 +61,7 @@ type Config struct {
 	// EnableGpuMetric toggles GPU monitoring where metrics are scraped from vendor specific sources
 	EnableGpuMetric bool `mapstructure:"gpu_metrics"`
 
-	// EnableNeuronMetric disables Neuron monitoring where metrics are scraped from vendor specific sources
-	// The default value is true meaning Neuron metrics get collected out of the box unless it's disabled
+	// EnableNeuronMetric toggles Neuron monitoring where metrics are scraped from neuron monitor
+	// The default value is false.
 	EnableNeuronMetric bool `mapstructure:"neuron_metrics"`
 }

receiver/awscontainerinsightreceiver/factory.go

@@ -42,6 +42,9 @@ const (
 
 	// Don't enable EKS control plane metrics by default
 	defaultEnableControlPlaneMetrics = false
+
+	// Don't enable Neuron metrics by default
+	defaultEnableNeuronMetrics = false
 )
 
 // NewFactory creates a factory for AWS container insight receiver
@@ -64,6 +67,7 @@ func createDefaultConfig() component.Config {
 		ClusterName:                 defaultClusterName,
 		LeaderLockName:              defaultLeaderLockName,
 		EnableControlPlaneMetrics:   defaultEnableControlPlaneMetrics,
+		EnableNeuronMetric:          defaultEnableNeuronMetrics,
 	}
 }
 

receiver/awscontainerinsightreceiver/internal/neuron/neuron_monitor_scraper_config.go

@@ -0,0 +1,100 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package nueron
+
+import (
+	"time"
+
+	"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awscontainerinsightreceiver/internal/prometheusscraper"
+	"github.com/prometheus/common/model"
+	"github.com/prometheus/prometheus/config"
+	"github.com/prometheus/prometheus/discovery"
+	"github.com/prometheus/prometheus/discovery/kubernetes"
+	"github.com/prometheus/prometheus/model/relabel"
+)
+
+const (
+	caFile             = "/var/run/secrets/kubernetes.io/serviceaccount/ca.crt"
+	collectionInterval = 60 * time.Second
+	jobName            = "containerInsightsNeuronMonitorScraper"
+)
+
+func GetNueronScrapeConfig(opts prometheusscraper.SimplePromethuesScraperOpts) *config.ScrapeConfig {
+
+	return &config.ScrapeConfig{
+		ScrapeInterval: model.Duration(collectionInterval),
+		ScrapeTimeout:  model.Duration(collectionInterval),
+		JobName:        jobName,
+		Scheme:         "http",
+		MetricsPath:    "/metrics",
+		ServiceDiscoveryConfigs: discovery.Configs{
+			&kubernetes.SDConfig{
+				Role: kubernetes.RoleService,
+				NamespaceDiscovery: kubernetes.NamespaceDiscovery{
+					IncludeOwnNamespace: true,
+				},
+				Selectors: []kubernetes.SelectorConfig{
+					{
+						Role:  kubernetes.RoleService,
+						Label: "k8s-app=neuron-monitor-service",
+					},
+				},
+				AttachMetadata: kubernetes.AttachMetadataConfig{
+					Node: true,
+				},
+			},
+		},
+		RelabelConfigs: []*relabel.Config{
+			{
+				SourceLabels: model.LabelNames{"__address__"},
+				Regex:        relabel.MustNewRegexp("([^:]+)(?::\\d+)?"),
+				Replacement:  "${1}:8000",
+				TargetLabel:  "__address__",
+				Action:       relabel.Replace,
+			},
+		},
+		MetricRelabelConfigs: GetNueronMetricRelabelConfigs(opts),
+	}
+}
+
+func GetNueronMetricRelabelConfigs(opts prometheusscraper.SimplePromethuesScraperOpts) []*relabel.Config {
+
+	return []*relabel.Config{
+		{
+			SourceLabels: model.LabelNames{"__name__"},
+			Regex:        relabel.MustNewRegexp("neuron.*|system_.*|execution_.*"),
+			Action:       relabel.Keep,
+		},
+		{
+			SourceLabels: model.LabelNames{"instance_name"},
+			TargetLabel:  "NodeName",
+			Regex:        relabel.MustNewRegexp("(.*)"),
+			Replacement:  "${1}",
+			Action:       relabel.Replace,
+		},
+		{
+			SourceLabels: model.LabelNames{"instance_id"},
+			TargetLabel:  "InstanceId",
+			Regex:        relabel.MustNewRegexp("(.*)"),
+			Replacement:  "${1}",
+			Action:       relabel.Replace,
+		},
+		{
+			SourceLabels: model.LabelNames{"neuroncore"},
+			TargetLabel:  "DeviceId",
+			Regex:        relabel.MustNewRegexp("(.*)"),
+			Replacement:  "${1}",
+			Action:       relabel.Replace,
+		},
+		// hacky way to inject static values (clusterName) to label set without additional processor
+		// relabel looks up an existing label then creates another label with given key (TargetLabel) and value (static)
+		{
+			SourceLabels: model.LabelNames{"instance_id"},
+			TargetLabel:  "ClusterName",
+			Regex:        relabel.MustNewRegexp("(.*)"),
+			Replacement:  opts.HostInfoProvider.GetClusterName(),
+			Action:       relabel.Replace,
+		},
+	}
+}