|
29 | 29 | import brave.baggage.BaggagePropagation.FactoryBuilder;
|
30 | 30 | import brave.baggage.BaggagePropagationConfig;
|
31 | 31 | import brave.baggage.BaggagePropagationCustomizer;
|
32 |
| -import brave.baggage.CorrelationScopeConfig; |
| 32 | +import brave.baggage.CorrelationScopeConfig.SingleCorrelationField; |
33 | 33 | import brave.baggage.CorrelationScopeCustomizer;
|
34 | 34 | import brave.baggage.CorrelationScopeDecorator;
|
35 | 35 | import brave.context.slf4j.MDCScopeDecorator;
|
|
51 | 51 | import io.micrometer.tracing.exporter.SpanReporter;
|
52 | 52 |
|
53 | 53 | import org.springframework.beans.factory.ObjectProvider;
|
| 54 | +import org.springframework.boot.actuate.autoconfigure.tracing.TracingProperties.Baggage.Correlation; |
54 | 55 | import org.springframework.boot.autoconfigure.AutoConfiguration;
|
55 | 56 | import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
|
56 | 57 | import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
@@ -225,11 +226,13 @@ CorrelationScopeDecorator.Builder mdcCorrelationScopeDecoratorBuilder(
|
225 | 226 | matchIfMissing = true)
|
226 | 227 | CorrelationScopeCustomizer correlationFieldsCorrelationScopeCustomizer() {
|
227 | 228 | return (builder) -> {
|
228 |
| - List<String> correlationFields = this.tracingProperties.getBaggage().getCorrelation().getFields(); |
229 |
| - for (String field : correlationFields) { |
230 |
| - builder.add(CorrelationScopeConfig.SingleCorrelationField.newBuilder(BaggageField.create(field)) |
| 229 | + Correlation correlationProperties = this.tracingProperties.getBaggage().getCorrelation(); |
| 230 | + for (String field : correlationProperties.getFields()) { |
| 231 | + BaggageField baggageField = BaggageField.create(field); |
| 232 | + SingleCorrelationField correlationField = SingleCorrelationField.newBuilder(baggageField) |
231 | 233 | .flushOnUpdate()
|
232 |
| - .build()); |
| 234 | + .build(); |
| 235 | + builder.add(correlationField); |
233 | 236 | }
|
234 | 237 | };
|
235 | 238 | }
|
|
0 commit comments