Skip to content

Commit 1223ed3

Browse files
authored
Add a setter for otel headers (#1809)
Add a setter for otel headers
1 parent 7f68679 commit 1223ed3

File tree

2 files changed

+15
-0
lines changed

2 files changed

+15
-0
lines changed

internal/venice-client-common/src/main/java/com/linkedin/venice/stats/VeniceMetricsConfig.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,11 @@ public Builder setOtelAdditionalMetricsReader(MetricReader otelAdditionalMetrics
302302
return this;
303303
}
304304

305+
public Builder setOtelHeaders(Map<String, String> otelHeaders) {
306+
this.otelHeaders = otelHeaders;
307+
return this;
308+
}
309+
305310
/**
306311
* Extract and set otel configs
307312
*/

internal/venice-client-common/src/test/java/com/linkedin/venice/stats/VeniceMetricsConfigTest.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,4 +203,14 @@ public void testSetOtelCustomDimensionsMapWithInvalidValue() {
203203
.extractAndSetOtelConfigs(otelConfigs)
204204
.build();
205205
}
206+
207+
@Test
208+
public void testSetOtelHeaders() {
209+
Map<String, String> otelHeaders = new HashMap<>();
210+
otelHeaders.put("key1", "value1");
211+
212+
VeniceMetricsConfig config =
213+
new Builder().setServiceName("TestService").setMetricPrefix("TestPrefix").setOtelHeaders(otelHeaders).build();
214+
assertEquals(config.getOtelHeaders().get("key1"), "value1");
215+
}
206216
}

0 commit comments

Comments
 (0)