feat: add Authorization header to OTEL config (CORE-2291)

monta-app · Aug 2, 2023 · 130acbc · 130acbc
1 parent d608f03
commit 130acbc
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/otelx/config.go b/otelx/config.go
@@ -20,9 +20,10 @@ type ZipkinConfig struct {
 }
 
 type OTLPConfig struct {
-	ServerURL string       `json:"server_url"`
-	Insecure  bool         `json:"insecure"`
-	Sampling  OTLPSampling `json:"sampling"`
+	ServerURL           string       `json:"server_url"`
+	Insecure            bool         `json:"insecure"`
+	Sampling            OTLPSampling `json:"sampling"`
+	AuthorizationHeader string       `json:"authorization_header"`
 }
 
 type JaegerSampling struct {

diff --git a/otelx/config.schema.json b/otelx/config.schema.json
@@ -129,6 +129,10 @@
                   "examples": [0.4]
                 }
               }
+            },
+            "authorization_header": {
+              "type": "string",
+              "examples": ["Bearer 2389s8fs9d8fus9f"]
             }
           }
         }

diff --git a/otelx/otlp.go b/otelx/otlp.go
@@ -27,6 +27,12 @@ func SetupOTLP(t *Tracer, tracerName string, c *Config) (trace.Tracer, error) {
 		clientOpts = append(clientOpts, otlptracehttp.WithInsecure())
 	}
 
+	if c.Providers.OTLP.AuthorizationHeader != "" {
+		clientOpts = append(clientOpts,
+			otlptracehttp.WithHeaders(map[string]string{"Authorization": c.Providers.OTLP.AuthorizationHeader}),
+		)
+	}
+
 	exp, err := otlptrace.New(
 		ctx, otlptracehttp.NewClient(clientOpts...),
 	)