File tree Expand file tree Collapse file tree 2 files changed +15
-0
lines changed
internal/venice-client-common/src
main/java/com/linkedin/venice/stats
test/java/com/linkedin/venice/stats Expand file tree Collapse file tree 2 files changed +15
-0
lines changed Original file line number Diff line number Diff line change @@ -302,6 +302,11 @@ public Builder setOtelAdditionalMetricsReader(MetricReader otelAdditionalMetrics
302
302
return this ;
303
303
}
304
304
305
+ public Builder setOtelHeaders (Map <String , String > otelHeaders ) {
306
+ this .otelHeaders = otelHeaders ;
307
+ return this ;
308
+ }
309
+
305
310
/**
306
311
* Extract and set otel configs
307
312
*/
Original file line number Diff line number Diff line change @@ -203,4 +203,14 @@ public void testSetOtelCustomDimensionsMapWithInvalidValue() {
203
203
.extractAndSetOtelConfigs (otelConfigs )
204
204
.build ();
205
205
}
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
+ }
206
216
}
You can’t perform that action at this time.
0 commit comments