org.springframework.cloud.sleuth.autoconfig.brave.instrument.web.BraveHttpConfiguration Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-cloud-sleuth-autoconfigure Show documentation
Show all versions of spring-cloud-sleuth-autoconfigure Show documentation
Spring Cloud Sleuth AutoConfigure
/*
* Copyright 2013-2021 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.cloud.sleuth.autoconfig.brave.instrument.web;
import java.util.List;
import java.util.regex.Pattern;
import javax.validation.constraints.NotNull;
import brave.Tracer;
import brave.Tracing;
import brave.http.HttpRequest;
import brave.http.HttpTracing;
import brave.http.HttpTracingCustomizer;
import brave.propagation.CurrentTraceContext;
import brave.sampler.SamplerFunction;
import brave.sampler.SamplerFunctions;
import reactor.util.context.Context;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.cloud.sleuth.autoconfig.instrument.web.ConditionalOnSleuthHttp;
import org.springframework.cloud.sleuth.autoconfig.instrument.web.SleuthHttpProperties;
import org.springframework.cloud.sleuth.autoconfig.instrument.web.SleuthWebProperties;
import org.springframework.cloud.sleuth.brave.bridge.BraveHttpRequestParser;
import org.springframework.cloud.sleuth.brave.bridge.BraveHttpResponseParser;
import org.springframework.cloud.sleuth.brave.bridge.BraveSamplerFunction;
import org.springframework.cloud.sleuth.brave.instrument.web.BraveSpanFromContextRetriever;
import org.springframework.cloud.sleuth.brave.instrument.web.CompositeHttpSampler;
import org.springframework.cloud.sleuth.brave.instrument.web.SkipPatternHttpClientSampler;
import org.springframework.cloud.sleuth.brave.instrument.web.SkipPatternHttpServerSampler;
import org.springframework.cloud.sleuth.http.HttpRequestParser;
import org.springframework.cloud.sleuth.http.HttpResponseParser;
import org.springframework.cloud.sleuth.instrument.web.HttpClientRequestParser;
import org.springframework.cloud.sleuth.instrument.web.HttpClientResponseParser;
import org.springframework.cloud.sleuth.instrument.web.HttpClientSampler;
import org.springframework.cloud.sleuth.instrument.web.HttpServerRequestParser;
import org.springframework.cloud.sleuth.instrument.web.HttpServerResponseParser;
import org.springframework.cloud.sleuth.instrument.web.HttpServerSampler;
import org.springframework.cloud.sleuth.instrument.web.SkipPatternProvider;
import org.springframework.cloud.sleuth.instrument.web.SpanFromContextRetriever;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.lang.Nullable;
/**
* {@link org.springframework.boot.autoconfigure.EnableAutoConfiguration
* Auto-configuration} related to HTTP based communication.
*
* @author Marcin Grzejszczak
* @since 2.0.0
*/
@Configuration(proxyBeanMethods = false)
@ConditionalOnSleuthHttp
@ConditionalOnClass(HttpTracing.class)
@EnableConfigurationProperties({ SleuthWebProperties.class, SleuthHttpProperties.class })
@Import(BraveHttpBridgeConfiguration.class)
public class BraveHttpConfiguration {
@Bean
@ConditionalOnMissingBean
// NOTE: stable bean name as might be used outside sleuth
HttpTracing httpTracing(Tracing tracing, @Nullable SkipPatternProvider provider,
@Nullable brave.http.HttpClientParser clientParser, @Nullable brave.http.HttpServerParser serverParser,
BeanFactory beanFactory, @Nullable List httpTracingCustomizers) {
HttpTracing.Builder builder = httpTracingBuilder(tracing, provider, beanFactory);
brave.http.HttpRequestParser httpClientRequestParser = httpRequestParser(beanFactory,
HttpClientRequestParser.NAME);
brave.http.HttpResponseParser httpClientResponseParser = httpResponseParser(beanFactory,
HttpClientResponseParser.NAME);
brave.http.HttpRequestParser httpServerRequestParser = httpRequestParser(beanFactory,
HttpServerRequestParser.NAME);
brave.http.HttpResponseParser httpServerResponseParser = httpResponseParser(beanFactory,
HttpServerResponseParser.NAME);
if (httpClientRequestParser != null || httpClientResponseParser != null) {
if (httpClientRequestParser != null) {
builder.clientRequestParser(httpClientRequestParser);
}
if (httpClientResponseParser != null) {
builder.clientResponseParser(httpClientResponseParser);
}
}
else if (clientParser != null) { // consider deprecated last
builder.clientParser(clientParser);
}
if (httpServerRequestParser != null || httpServerResponseParser != null) {
if (httpServerRequestParser != null) {
builder.serverRequestParser(httpServerRequestParser);
}
if (httpServerResponseParser != null) {
builder.serverResponseParser(httpServerResponseParser);
}
}
else if (serverParser != null) { // consider deprecated last
builder.serverParser(serverParser);
}
if (httpTracingCustomizers != null) {
for (HttpTracingCustomizer customizer : httpTracingCustomizers) {
customizer.customize(builder);
}
}
return builder.build();
}
private brave.http.HttpRequestParser httpRequestParser(BeanFactory beanFactory, String name) {
return beanFactory.containsBean(name) ? toBraveHttpRequestParser(beanFactory, name) : null;
}
private brave.http.HttpResponseParser httpResponseParser(BeanFactory beanFactory, String name) {
return beanFactory.containsBean(name) ? toBraveHttpResponseParser(beanFactory, name) : null;
}
@NotNull
private HttpTracing.Builder httpTracingBuilder(Tracing tracing, @Nullable SkipPatternProvider provider,
BeanFactory beanFactory) {
SamplerFunction httpClientSampler = toBraveSampler(beanFactory, HttpClientSampler.NAME);
SamplerFunction httpServerSampler = httpServerSampler(beanFactory);
SamplerFunction combinedSampler = combineUserProvidedSamplerWithSkipPatternSampler(
httpServerSampler, provider);
return HttpTracing.newBuilder(tracing).clientSampler(httpClientSampler).serverSampler(combinedSampler);
}
@Nullable
private SamplerFunction httpServerSampler(BeanFactory beanFactory) {
return beanFactory.containsBean(HttpServerSampler.NAME) ? toBraveSampler(beanFactory, HttpServerSampler.NAME)
: null;
}
private brave.http.HttpRequestParser toBraveHttpRequestParser(BeanFactory beanFactory, String beanName) {
Object bean = beanFactory.getBean(beanName);
brave.http.HttpRequestParser parser = bean instanceof brave.http.HttpRequestParser
? (brave.http.HttpRequestParser) bean
: bean instanceof HttpRequestParser ? BraveHttpRequestParser.toBrave((HttpRequestParser) bean) : null;
return returnOrThrow(bean, parser, beanName, brave.http.HttpRequestParser.class, HttpRequestParser.class);
}
private brave.http.HttpResponseParser toBraveHttpResponseParser(BeanFactory beanFactory, String beanName) {
Object bean = beanFactory.getBean(beanName);
brave.http.HttpResponseParser parser = bean instanceof brave.http.HttpResponseParser
? (brave.http.HttpResponseParser) bean : bean instanceof HttpResponseParser
? BraveHttpResponseParser.toBrave((HttpResponseParser) bean) : null;
return returnOrThrow(bean, parser, beanName, brave.http.HttpResponseParser.class, HttpResponseParser.class);
}
private SamplerFunction toBraveSampler(BeanFactory beanFactory, String beanName) {
Object bean = beanFactory.getBean(beanName);
SamplerFunction braveSampler = bean instanceof SamplerFunction
? (SamplerFunction) bean
: bean instanceof org.springframework.cloud.sleuth.SamplerFunction ? BraveSamplerFunction.toHttpBrave(
(org.springframework.cloud.sleuth.SamplerFunction) bean)
: null;
return returnOrThrow(bean, braveSampler, beanName, SamplerFunction.class,
org.springframework.cloud.sleuth.SamplerFunction.class);
}
@NotNull
private T returnOrThrow(Object bean, T convertedBean, String name, Class brave, Class sleuth) {
if (convertedBean == null) {
throw new IllegalStateException(
"Bean with name [" + name + "] is of type [" + bean.getClass() + "] and only ["
+ brave.getCanonicalName() + "] and [" + sleuth.getCanonicalName() + "] are supported");
}
return convertedBean;
}
private SamplerFunction combineUserProvidedSamplerWithSkipPatternSampler(
@Nullable SamplerFunction serverSampler, @Nullable SkipPatternProvider provider) {
SamplerFunction skipPatternSampler = provider != null
? new SkipPatternHttpServerSampler(provider) : null;
if (serverSampler == null && skipPatternSampler == null) {
return SamplerFunctions.deferDecision();
}
else if (serverSampler == null) {
return skipPatternSampler;
}
else if (skipPatternSampler == null) {
return serverSampler;
}
return new CompositeHttpSampler(skipPatternSampler, serverSampler);
}
@Bean
@ConditionalOnMissingBean(name = HttpClientSampler.NAME)
SamplerFunction sleuthHttpClientSampler(SleuthWebProperties sleuthWebProperties) {
String skipPattern = sleuthWebProperties.getClient().getSkipPattern();
if (skipPattern == null) {
return SamplerFunctions.deferDecision();
}
return new SkipPatternHttpClientSampler(Pattern.compile(skipPattern));
}
@Configuration(proxyBeanMethods = false)
@ConditionalOnClass(Context.class)
static class BraveWebFilterConfiguration {
@Bean
SpanFromContextRetriever braveSpanFromContextRetriever(CurrentTraceContext currentTraceContext, Tracer tracer) {
return new BraveSpanFromContextRetriever(currentTraceContext, tracer);
}
}
}