Update how protocol is set

open-telemetry · Dec 20, 2022 · e8c7252 · e8c7252
1 parent 2a70f7d
commit e8c7252
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 22 deletions.
diff --git a/pkg/instrumentation/podmutator_test.go b/pkg/instrumentation/podmutator_test.go
@@ -406,15 +406,15 @@ func TestMutatePod(t *testing.T) {
 							},
 							{
 								Name:  "OTEL_TRACES_EXPORTER",
-								Value: "otlp_proto_http",
+								Value: "otlp",
 							},
 							{
 								Name:  "OTEL_METRICS_EXPORTER",
-								Value: "none",
+								Value: "otlp",
 							},
 							{
 								Name:  "OTEL_EXPORTER_OTLP_ENDPOINT",
-								Value: "http://localhost:4317",
+								Value: "http://localhost:4318",
 							},
 						},
 					},
@@ -491,20 +491,28 @@ func TestMutatePod(t *testing.T) {
 								},
 								{
 									Name:  "OTEL_TRACES_EXPORTER",
-									Value: "otlp_proto_http",
+									Value: "otlp",
 								},
 								{
 									Name:  "OTEL_METRICS_EXPORTER",
-									Value: "none",
+									Value: "otlp",
 								},
 								{
 									Name:  "OTEL_EXPORTER_OTLP_ENDPOINT",
-									Value: "http://localhost:4317",
+									Value: "http://localhost:4318",
 								},
 								{
 									Name:  "PYTHONPATH",
 									Value: fmt.Sprintf("%s:%s", pythonPathPrefix, pythonPathSuffix),
 								},
+								{
+									Name:  "OTEL_EXPORTER_OTLP_TRACES_PROTOCOL",
+									Value: "http/protobuf",
+								},
+								{
+									Name:  "OTEL_EXPORTER_OTLP_METRICS_PROTOCOL",
+									Value: "http/protobuf",
+								},
 								{
 									Name:  "OTEL_EXPORTER_OTLP_TIMEOUT",
 									Value: "20",

diff --git a/pkg/instrumentation/python.go b/pkg/instrumentation/python.go
@@ -66,15 +66,15 @@ func injectPythonSDK(pythonSpec v1alpha1.Python, pod corev1.Pod, index int) (cor
 			Name:  envOtelTracesExporter,
 			Value: "otlp",
 		})
+	}
 
-		// Set OTEL_EXPORTER_OTLP_TRACES_PROTOCOL to http/protobuf if not set by user because it is what our autoinstrumentation supports.
-		idx = getIndexOfEnv(container.Env, envOtelExporterOTLPTracesProtocol)
-		if idx == -1 {
-			container.Env = append(container.Env, corev1.EnvVar{
-				Name:  envOtelExporterOTLPTracesProtocol,
-				Value: "http/protobuf",
-			})
-		}
+	// Set OTEL_EXPORTER_OTLP_TRACES_PROTOCOL to http/protobuf if not set by user because it is what our autoinstrumentation supports.
+	idx = getIndexOfEnv(container.Env, envOtelExporterOTLPTracesProtocol)
+	if idx == -1 {
+		container.Env = append(container.Env, corev1.EnvVar{
+			Name:  envOtelExporterOTLPTracesProtocol,
+			Value: "http/protobuf",
+		})
 	}
 
 	// Set OTEL_METRICS_EXPORTER to HTTP exporter if not set by user because it is what our autoinstrumentation supports.
@@ -84,15 +84,15 @@ func injectPythonSDK(pythonSpec v1alpha1.Python, pod corev1.Pod, index int) (cor
 			Name:  envOtelMetricsExporter,
 			Value: "otlp",
 		})
+	}
 
-		// Set OTEL_EXPORTER_OTLP_METRICS_PROTOCOL to http/protobuf if not set by user because it is what our autoinstrumentation supports.
-		idx = getIndexOfEnv(container.Env, envOtelExporterOTLPMetricsProtocol)
-		if idx == -1 {
-			container.Env = append(container.Env, corev1.EnvVar{
-				Name:  envOtelExporterOTLPMetricsProtocol,
-				Value: "http/protobuf",
-			})
-		}
+	// Set OTEL_EXPORTER_OTLP_METRICS_PROTOCOL to http/protobuf if not set by user because it is what our autoinstrumentation supports.
+	idx = getIndexOfEnv(container.Env, envOtelExporterOTLPMetricsProtocol)
+	if idx == -1 {
+		container.Env = append(container.Env, corev1.EnvVar{
+			Name:  envOtelExporterOTLPMetricsProtocol,
+			Value: "http/protobuf",
+		})
 	}
 
 	container.VolumeMounts = append(container.VolumeMounts, corev1.VolumeMount{

diff --git a/pkg/instrumentation/python_test.go b/pkg/instrumentation/python_test.go
@@ -228,6 +228,10 @@ func TestInjectPythonSDK(t *testing.T) {
 									Name:  "PYTHONPATH",
 									Value: fmt.Sprintf("%s:%s", pythonPathPrefix, pythonPathSuffix),
 								},
+								{
+									Name:  "OTEL_EXPORTER_OTLP_TRACES_PROTOCOL",
+									Value: "http/protobuf",
+								},
 								{
 									Name:  "OTEL_METRICS_EXPORTER",
 									Value: "otlp",
@@ -306,6 +310,10 @@ func TestInjectPythonSDK(t *testing.T) {
 									Name:  "OTEL_EXPORTER_OTLP_TRACES_PROTOCOL",
 									Value: "http/protobuf",
 								},
+								{
+									Name:  "OTEL_EXPORTER_OTLP_METRICS_PROTOCOL",
+									Value: "http/protobuf",
+								},
 							},
 						},
 					},