Introduce typed pipeline (open-telemetry#2896)

* introduce more typing for pipelines * rock and a hard place * resolve tests * Fix tests * chlog * fix docs * thing * alphabet * remove chlog entry
ItielOlenick · Apr 30, 2024 · 325c814 · 325c814
1 parent 255092f
commit 325c814
Show file tree

Hide file tree

Showing 19 changed files with 324 additions and 32 deletions.
diff --git a/apis/v1beta1/config.go b/apis/v1beta1/config.go
@@ -27,6 +27,18 @@ import (
 	"gopkg.in/yaml.v3"
 )
 
+type ComponentType int
+
+const (
+	ComponentTypeReceiver ComponentType = iota
+	ComponentTypeExporter
+	ComponentTypeProcessor
+)
+
+func (c ComponentType) String() string {
+	return [...]string{"receiver", "exporter", "processor"}[c]
+}
+
 // AnyConfig represent parts of the config.
 type AnyConfig struct {
 	Object map[string]interface{} `json:"-" yaml:",inline"`
@@ -75,6 +87,37 @@ func (c *AnyConfig) MarshalJSON() ([]byte, error) {
 	return json.Marshal(c.Object)
 }
 
+// Pipeline is a struct of component type to a list of component IDs.
+type Pipeline struct {
+	Exporters  []string `json:"exporters" yaml:"exporters"`
+	Processors []string `json:"processors" yaml:"processors"`
+	Receivers  []string `json:"receivers" yaml:"receivers"`
+}
+
+// GetEnabledComponents constructs a list of enabled components by component type.
+func (c *Config) GetEnabledComponents() map[ComponentType]map[string]interface{} {
+	toReturn := map[ComponentType]map[string]interface{}{
+		ComponentTypeReceiver:  {},
+		ComponentTypeProcessor: {},
+		ComponentTypeExporter:  {},
+	}
+	for _, pipeline := range c.Service.Pipelines {
+		if pipeline == nil {
+			continue
+		}
+		for _, componentId := range pipeline.Receivers {
+			toReturn[ComponentTypeReceiver][componentId] = struct{}{}
+		}
+		for _, componentId := range pipeline.Exporters {
+			toReturn[ComponentTypeExporter][componentId] = struct{}{}
+		}
+		for _, componentId := range pipeline.Processors {
+			toReturn[ComponentTypeProcessor][componentId] = struct{}{}
+		}
+	}
+	return toReturn
+}
+
 // Config encapsulates collector config.
 type Config struct {
 	// +kubebuilder:pruning:PreserveUnknownFields
@@ -135,7 +178,7 @@ type Service struct {
 	// +kubebuilder:pruning:PreserveUnknownFields
 	Telemetry *AnyConfig `json:"telemetry,omitempty" yaml:"telemetry,omitempty"`
 	// +kubebuilder:pruning:PreserveUnknownFields
-	Pipelines AnyConfig `json:"pipelines" yaml:"pipelines"`
+	Pipelines map[string]*Pipeline `json:"pipelines" yaml:"pipelines"`
 }
 
 // MetricsPort gets the port number for the metrics endpoint from the collector config if it has been set.

diff --git a/apis/v1beta1/config_test.go b/apis/v1beta1/config_test.go
@@ -145,11 +145,11 @@ func TestConfigYaml(t *testing.T) {
 					"insights": "yeah!",
 				},
 			},
-			Pipelines: AnyConfig{
-				Object: map[string]interface{}{
-					"receivers":  []string{"otlp"},
-					"processors": []string{"modify_2000"},
-					"exporters":  []string{"otlp/exporter", "con"},
+			Pipelines: map[string]*Pipeline{
+				"traces": {
+					Receivers:  []string{"otlp"},
+					Processors: []string{"modify_2000"},
+					Exporters:  []string{"otlp/exporter", "con"},
 				},
 			},
 		},
@@ -173,13 +173,14 @@ service:
   telemetry:
     insights: yeah!
   pipelines:
-    exporters:
-      - otlp/exporter
-      - con
-    processors:
-      - modify_2000
-    receivers:
-      - otlp
+    traces:
+      exporters:
+        - otlp/exporter
+        - con
+      processors:
+        - modify_2000
+      receivers:
+        - otlp
 `
 
 	assert.Equal(t, expected, yamlCollector)
@@ -278,3 +279,109 @@ func TestConfigToMetricsPort(t *testing.T) {
 		})
 	}
 }
+
+func TestConfig_GetEnabledComponents(t *testing.T) {
+	tests := []struct {
+		name string
+		file string
+		want map[ComponentType]map[string]interface{}
+	}{
+
+		{
+			name: "connectors",
+			file: "testdata/otelcol-connectors.yaml",
+			want: map[ComponentType]map[string]interface{}{
+				ComponentTypeReceiver: {
+					"foo":   struct{}{},
+					"count": struct{}{},
+				},
+				ComponentTypeProcessor: {},
+				ComponentTypeExporter: {
+					"bar":   struct{}{},
+					"count": struct{}{},
+				},
+			},
+		},
+		{
+			name: "couchbase",
+			file: "testdata/otelcol-couchbase.yaml",
+			want: map[ComponentType]map[string]interface{}{
+				ComponentTypeReceiver: {
+					"prometheus/couchbase": struct{}{},
+				},
+				ComponentTypeProcessor: {
+					"filter/couchbase":           struct{}{},
+					"metricstransform/couchbase": struct{}{},
+					"transform/couchbase":        struct{}{},
+				},
+				ComponentTypeExporter: {
+					"prometheus": struct{}{},
+				},
+			},
+		},
+		{
+			name: "demo",
+			file: "testdata/otelcol-demo.yaml",
+			want: map[ComponentType]map[string]interface{}{
+				ComponentTypeReceiver: {
+					"otlp": struct{}{},
+				},
+				ComponentTypeProcessor: {
+					"batch": struct{}{},
+				},
+				ComponentTypeExporter: {
+					"debug":      struct{}{},
+					"zipkin":     struct{}{},
+					"otlp":       struct{}{},
+					"prometheus": struct{}{},
+				},
+			},
+		},
+		{
+			name: "extensions",
+			file: "testdata/otelcol-extensions.yaml",
+			want: map[ComponentType]map[string]interface{}{
+				ComponentTypeReceiver: {
+					"otlp": struct{}{},
+				},
+				ComponentTypeProcessor: {},
+				ComponentTypeExporter: {
+					"otlp/auth": struct{}{},
+				},
+			},
+		},
+		{
+			name: "filelog",
+			file: "testdata/otelcol-filelog.yaml",
+			want: map[ComponentType]map[string]interface{}{
+				ComponentTypeReceiver: {
+					"filelog": struct{}{},
+				},
+				ComponentTypeProcessor: {},
+				ComponentTypeExporter: {
+					"debug": struct{}{},
+				},
+			},
+		},
+		{
+			name: "null",
+			file: "testdata/otelcol-null-values.yaml",
+			want: map[ComponentType]map[string]interface{}{
+				ComponentTypeReceiver:  {},
+				ComponentTypeProcessor: {},
+				ComponentTypeExporter:  {},
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			collectorYaml, err := os.ReadFile(tt.file)
+			require.NoError(t, err)
+
+			c := &Config{}
+			err = go_yaml.Unmarshal(collectorYaml, c)
+			require.NoError(t, err)
+			assert.Equalf(t, tt.want, c.GetEnabledComponents(), "GetEnabledComponents()")
+		})
+	}
+}
diff --git a/apis/v1beta1/opentelemetrycollector_types.go b/apis/v1beta1/opentelemetrycollector_types.go
@@ -49,6 +49,7 @@ type OpenTelemetryCollector struct {
 	Status OpenTelemetryCollectorStatus `json:"status,omitempty"`
 }
 
+// Hub exists to allow for conversion.
 func (*OpenTelemetryCollector) Hub() {}
 
 //+kubebuilder:object:root=true

diff --git a/apis/v1beta1/testdata/otelcol-connectors.yaml b/apis/v1beta1/testdata/otelcol-connectors.yaml
@@ -17,7 +17,9 @@ service:
   pipelines:
     traces:
       receivers: [foo]
+      processors: []
       exporters: [count]
     metrics:
       receivers: [count]
+      processors: []
       exporters: [bar]
diff --git a/apis/v1beta1/testdata/otelcol-filelog.yaml b/apis/v1beta1/testdata/otelcol-filelog.yaml
@@ -81,4 +81,5 @@ service:
   pipelines:
     logs:
       receivers: [filelog]
+      processors: []
       exporters: [debug]
diff --git a/apis/v1beta1/zz_generated.deepcopy.go b/apis/v1beta1/zz_generated.deepcopy.go
diff --git a/bundle/manifests/opentelemetry.io_opentelemetrycollectors.yaml b/bundle/manifests/opentelemetry.io_opentelemetrycollectors.yaml
@@ -5522,6 +5522,25 @@ spec:
                           type: string
                         type: array
                       pipelines:
+                        additionalProperties:
+                          properties:
+                            exporters:
+                              items:
+                                type: string
+                              type: array
+                            processors:
+                              items:
+                                type: string
+                              type: array
+                            receivers:
+                              items:
+                                type: string
+                              type: array
+                          required:
+                          - exporters
+                          - processors
+                          - receivers
+                          type: object
                         type: object
                         x-kubernetes-preserve-unknown-fields: true
                       telemetry:

diff --git a/config/crd/bases/opentelemetry.io_opentelemetrycollectors.yaml b/config/crd/bases/opentelemetry.io_opentelemetrycollectors.yaml
@@ -5508,6 +5508,25 @@ spec:
                           type: string
                         type: array
                       pipelines:
+                        additionalProperties:
+                          properties:
+                            exporters:
+                              items:
+                                type: string
+                              type: array
+                            processors:
+                              items:
+                                type: string
+                              type: array
+                            receivers:
+                              items:
+                                type: string
+                              type: array
+                          required:
+                          - exporters
+                          - processors
+                          - receivers
+                          type: object
                         type: object
                         x-kubernetes-preserve-unknown-fields: true
                       telemetry: