commit
f6e208c647
@ -1,43 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2012-2023 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.boot.autoconfigure.reactor;
|
|
||||||
|
|
||||||
import reactor.core.publisher.Hooks;
|
|
||||||
|
|
||||||
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
|
|
||||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
|
||||||
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
|
||||||
import org.springframework.context.annotation.Configuration;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* {@link EnableAutoConfiguration Auto-configuration} for Reactor.
|
|
||||||
*
|
|
||||||
* @author Brian Clozel
|
|
||||||
* @since 3.0.2
|
|
||||||
*/
|
|
||||||
@Configuration(proxyBeanMethods = false)
|
|
||||||
@ConditionalOnClass(Hooks.class)
|
|
||||||
@EnableConfigurationProperties(ReactorProperties.class)
|
|
||||||
public class ReactorAutoConfiguration {
|
|
||||||
|
|
||||||
public ReactorAutoConfiguration(ReactorProperties properties) {
|
|
||||||
if (properties.getContextPropagation() == ReactorProperties.ContextPropagationMode.AUTO) {
|
|
||||||
Hooks.enableAutomaticContextPropagation();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
@ -1,57 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2012-2023 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.boot.autoconfigure.reactor;
|
|
||||||
|
|
||||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Configuration properties for Reactor.
|
|
||||||
*
|
|
||||||
* @author Brian Clozel
|
|
||||||
* @since 3.0.3
|
|
||||||
*/
|
|
||||||
@ConfigurationProperties(prefix = "spring.reactor")
|
|
||||||
public class ReactorProperties {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Context Propagation support mode for Reactor operators.
|
|
||||||
*/
|
|
||||||
private ContextPropagationMode contextPropagation = ContextPropagationMode.AUTO;
|
|
||||||
|
|
||||||
public ContextPropagationMode getContextPropagation() {
|
|
||||||
return this.contextPropagation;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setContextPropagation(ContextPropagationMode contextPropagation) {
|
|
||||||
this.contextPropagation = contextPropagation;
|
|
||||||
}
|
|
||||||
|
|
||||||
public enum ContextPropagationMode {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Context Propagation is applied to all Reactor operators.
|
|
||||||
*/
|
|
||||||
AUTO,
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Context Propagation is only applied to "tap" and "handle" Reactor operators.
|
|
||||||
*/
|
|
||||||
LIMITED
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
@ -1,20 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2012-2023 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Auto-configuration for Reactor.
|
|
||||||
*/
|
|
||||||
package org.springframework.boot.autoconfigure.reactor;
|
|
@ -1,64 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2012-2023 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.boot.autoconfigure.reactor;
|
|
||||||
|
|
||||||
import java.util.concurrent.atomic.AtomicReference;
|
|
||||||
|
|
||||||
import io.micrometer.context.ContextRegistry;
|
|
||||||
import org.junit.jupiter.api.BeforeAll;
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import reactor.core.publisher.Mono;
|
|
||||||
import reactor.util.context.Context;
|
|
||||||
|
|
||||||
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
|
||||||
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Tests for {@link ReactorAutoConfiguration}.
|
|
||||||
*
|
|
||||||
* @author Brian Clozel
|
|
||||||
*/
|
|
||||||
class ReactorAutoConfigurationTests {
|
|
||||||
|
|
||||||
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
|
||||||
.withConfiguration(AutoConfigurations.of(ReactorAutoConfiguration.class));
|
|
||||||
|
|
||||||
private static final String THREADLOCAL_KEY = "ReactorAutoConfigurationTests";
|
|
||||||
|
|
||||||
private static final ThreadLocal<String> THREADLOCAL_VALUE = ThreadLocal.withInitial(() -> "failure");
|
|
||||||
|
|
||||||
@BeforeAll
|
|
||||||
static void initializeThreadLocalAccessors() {
|
|
||||||
ContextRegistry globalRegistry = ContextRegistry.getInstance();
|
|
||||||
globalRegistry.registerThreadLocalAccessor(THREADLOCAL_KEY, THREADLOCAL_VALUE);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void shouldConfigureAutomaticContextPropagation() {
|
|
||||||
AtomicReference<String> threadLocalValue = new AtomicReference<>();
|
|
||||||
this.contextRunner.run((applicationContext) -> {
|
|
||||||
Mono.just("test")
|
|
||||||
.doOnNext((element) -> threadLocalValue.set(THREADLOCAL_VALUE.get()))
|
|
||||||
.contextWrite(Context.of(THREADLOCAL_KEY, "success"))
|
|
||||||
.block();
|
|
||||||
assertThat(threadLocalValue.get()).isEqualTo("success");
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
Loading…
Reference in New Issue