Skip to content

Commit 09b9a28

Browse files
committed
Polish
1 parent 8338767 commit 09b9a28

File tree

2 files changed

+20
-28
lines changed

2 files changed

+20
-28
lines changed

spring-boot-project/spring-boot-actuator-autoconfigure/src/main/java/org/springframework/boot/actuate/autoconfigure/tracing/BraveAutoConfiguration.java

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,6 @@
1616

1717
package org.springframework.boot.actuate.autoconfigure.tracing;
1818

19-
import java.util.ArrayList;
20-
import java.util.Arrays;
21-
import java.util.Collection;
22-
import java.util.Collections;
2319
import java.util.List;
2420

2521
import brave.CurrentSpanCustomizer;
@@ -38,27 +34,23 @@
3834
import brave.baggage.CorrelationScopeDecorator;
3935
import brave.context.slf4j.MDCScopeDecorator;
4036
import brave.handler.SpanHandler;
41-
import brave.propagation.B3Propagation;
4237
import brave.propagation.CurrentTraceContext;
4338
import brave.propagation.CurrentTraceContext.ScopeDecorator;
4439
import brave.propagation.CurrentTraceContextCustomizer;
4540
import brave.propagation.Propagation.Factory;
4641
import brave.propagation.ThreadLocalCurrentTraceContext;
4742
import brave.sampler.Sampler;
48-
import io.micrometer.tracing.BaggageManager;
4943
import io.micrometer.tracing.brave.bridge.BraveBaggageManager;
5044
import io.micrometer.tracing.brave.bridge.BraveCurrentTraceContext;
5145
import io.micrometer.tracing.brave.bridge.BravePropagator;
5246
import io.micrometer.tracing.brave.bridge.BraveSpanCustomizer;
5347
import io.micrometer.tracing.brave.bridge.BraveTracer;
5448
import io.micrometer.tracing.brave.bridge.CompositeSpanHandler;
55-
import io.micrometer.tracing.brave.bridge.W3CPropagation;
5649
import io.micrometer.tracing.exporter.SpanExportingPredicate;
5750
import io.micrometer.tracing.exporter.SpanFilter;
5851
import io.micrometer.tracing.exporter.SpanReporter;
5952

6053
import org.springframework.beans.factory.ObjectProvider;
61-
import org.springframework.boot.actuate.autoconfigure.tracing.TracingProperties.Propagation.PropagationType;
6254
import org.springframework.boot.autoconfigure.AutoConfiguration;
6355
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
6456
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
@@ -194,7 +186,8 @@ static class BraveBaggageConfiguration {
194186
@ConditionalOnMissingBean
195187
BaggagePropagation.FactoryBuilder propagationFactoryBuilder(
196188
ObjectProvider<BaggagePropagationCustomizer> baggagePropagationCustomizers) {
197-
Factory delegate = CompositePropagationFactory.create(BRAVE_BAGGAGE_MANAGER, this.tracingProperties.getPropagation().getType());
189+
Factory delegate = CompositePropagationFactory.create(BRAVE_BAGGAGE_MANAGER,
190+
this.tracingProperties.getPropagation().getType());
198191
FactoryBuilder builder = BaggagePropagation.newFactoryBuilder(delegate);
199192
baggagePropagationCustomizers.orderedStream().forEach((customizer) -> customizer.customize(builder));
200193
return builder;

spring-boot-project/spring-boot-actuator-autoconfigure/src/main/java/org/springframework/boot/actuate/autoconfigure/tracing/CompositePropagationFactory.java

Lines changed: 18 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -57,9 +57,9 @@ class CompositePropagationFactory extends Propagation.Factory implements Propaga
5757
this.injectors = stream(this.injectorFactories).map(Factory::get).toList();
5858
this.extractors = stream(extractorFactories).map(Factory::get).toList();
5959
this.supportsJoin = Stream.concat(stream(this.injectorFactories), stream(this.extractorFactories))
60-
.allMatch(Factory::supportsJoin);
60+
.allMatch(Factory::supportsJoin);
6161
this.requires128BitTraceId = Stream.concat(stream(this.injectorFactories), stream(this.extractorFactories))
62-
.anyMatch(Factory::requires128BitTraceId);
62+
.anyMatch(Factory::requires128BitTraceId);
6363
}
6464

6565
Iterable<Factory> getInjectorFactories() {
@@ -69,9 +69,9 @@ Iterable<Factory> getInjectorFactories() {
6969
@Override
7070
public List<String> keys() {
7171
return Stream.concat(this.injectors.stream(), this.extractors.stream())
72-
.flatMap((entry) -> entry.keys().stream())
73-
.distinct()
74-
.toList();
72+
.flatMap((entry) -> entry.keys().stream())
73+
.distinct()
74+
.toList();
7575
}
7676

7777
@Override
@@ -130,23 +130,25 @@ public TraceContext decorate(TraceContext context) {
130130
}
131131

132132
/**
133-
* Creates a new {@link CompositePropagationFactory}, which uses the given {@code injectionTypes} for injection and all supported
134-
* types for extraction.
135-
*
133+
* Creates a new {@link CompositePropagationFactory}, which uses the given
134+
* {@code injectionTypes} for injection and all supported types for extraction.
136135
* @param baggageManager the baggage manager to use, or {@code null}
137136
* @param injectionTypes the propagation types for injection
138137
* @return the {@link CompositePropagationFactory}
139138
*/
140-
static CompositePropagationFactory create(BaggageManager baggageManager, Iterable<TracingProperties.Propagation.PropagationType> injectionTypes) {
141-
List<Factory> injectors = stream(injectionTypes).map((injector) -> factoryForType(baggageManager, injector)).toList();
142-
List<Factory> extractors = Arrays.stream(TracingProperties.Propagation.PropagationType.values()).map((extractor) -> factoryForType(baggageManager, extractor)).toList();
139+
static CompositePropagationFactory create(BaggageManager baggageManager,
140+
Iterable<TracingProperties.Propagation.PropagationType> injectionTypes) {
141+
List<Factory> injectors = stream(injectionTypes).map((injection) -> factoryForType(baggageManager, injection))
142+
.toList();
143+
List<Factory> extractors = Arrays.stream(TracingProperties.Propagation.PropagationType.values())
144+
.map((extraction) -> factoryForType(baggageManager, extraction))
145+
.toList();
143146
return new CompositePropagationFactory(injectors, extractors);
144147
}
145148

146149
/**
147-
* Creates a new {@link CompositePropagationFactory}, which uses the given {@code injectionTypes} for injection and all supported
148-
* types for extraction.
149-
*
150+
* Creates a new {@link CompositePropagationFactory}, which uses the given
151+
* {@code injectionTypes} for injection and all supported types for extraction.
150152
* @param injectionTypes the propagation types for injection
151153
* @return the {@link CompositePropagationFactory}
152154
*/
@@ -158,7 +160,8 @@ private static <T> Stream<T> stream(Iterable<T> iterable) {
158160
return StreamSupport.stream(iterable.spliterator(), false);
159161
}
160162

161-
private static Factory factoryForType(BaggageManager baggageManager, TracingProperties.Propagation.PropagationType type) {
163+
private static Factory factoryForType(BaggageManager baggageManager,
164+
TracingProperties.Propagation.PropagationType type) {
162165
return switch (type) {
163166
case B3 -> b3Single();
164167
case B3_MULTI -> b3Multi();
@@ -168,7 +171,6 @@ private static Factory factoryForType(BaggageManager baggageManager, TracingProp
168171

169172
/**
170173
* Creates a new B3 propagation factory using a single B3 header.
171-
*
172174
* @return the B3 propagation factory
173175
*/
174176
private static Factory b3Single() {
@@ -177,7 +179,6 @@ private static Factory b3Single() {
177179

178180
/**
179181
* Creates a new B3 propagation factory using multiple B3 headers.
180-
*
181182
* @return the B3 propagation factory
182183
*/
183184
private static Factory b3Multi() {
@@ -186,7 +187,6 @@ private static Factory b3Multi() {
186187

187188
/**
188189
* Creates a new W3C propagation factory.
189-
*
190190
* @param baggageManager baggage manager to use, or {@code null}
191191
* @return the W3C propagation factory
192192
*/
@@ -195,5 +195,4 @@ private static W3CPropagation w3c(BaggageManager baggageManager) {
195195
: new W3CPropagation();
196196
}
197197

198-
199198
}

0 commit comments

Comments
 (0)