GH-653 Retain original 'lookupDestination' while adding 'reconsiledLookupDestination'

This is necessary to retain user provided information while also prpviding a way to reconsile the actual destination/function
This commit is contained in:
Oleg Zhurakousky
2021-02-22 14:08:02 +01:00
parent 0abce5a2b8
commit 3821e5745c
3 changed files with 79 additions and 2 deletions

View File

@@ -16,6 +16,7 @@
package org.springframework.cloud.function.rsocket;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
@@ -32,10 +33,14 @@ import org.springframework.cloud.function.context.config.RoutingFunction;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.Message;
import org.springframework.messaging.handler.DestinationPatternsMessageCondition;
import org.springframework.messaging.rsocket.RSocketRequester;
import org.springframework.util.MimeTypeUtils;
import org.springframework.util.SocketUtils;
import static org.assertj.core.api.Assertions.assertThat;
/**
*
* @author Oleg Zhurakousky
@@ -139,6 +144,35 @@ public class RSocketAutoConfigurationRoutingTests {
}
}
@Test
public void testRoutingWithDefinitionMessageFunction() {
int port = SocketUtils.findAvailableTcpPort();
try (
ConfigurableApplicationContext applicationContext =
new SpringApplicationBuilder(SampleFunctionConfiguration.class)
.web(WebApplicationType.NONE)
.run("--logging.level.org.springframework.cloud.function=DEBUG",
"--spring.cloud.function.definition=uppercase",
"--spring.cloud.function.routing-expression=headers.func_name",
"--spring.cloud.function.expected-content-type=text/plain",
"--spring.rsocket.server.port=" + port);
) {
RSocketRequester.Builder rsocketRequesterBuilder =
applicationContext.getBean(RSocketRequester.Builder.class);
rsocketRequesterBuilder.tcp("localhost", port)
.route("uppercase")
.metadata("{\"func_name\":\"uppercaseMessage\"}", MimeTypeUtils.APPLICATION_JSON)
.data("hello")
.retrieveMono(String.class)
.as(StepVerifier::create)
.expectNext("HELLO")
.expectComplete()
.verify();
}
}
@EnableAutoConfiguration
@Configuration
public static class SampleFunctionConfiguration {
@@ -150,6 +184,17 @@ public class RSocketAutoConfigurationRoutingTests {
return v -> v.toUpperCase();
}
@Bean
public Function<Message<String>, String> uppercaseMessage() {
return msg -> {
assertThat(msg.getHeaders()
.get(DestinationPatternsMessageCondition.LOOKUP_DESTINATION_HEADER)).toString().equals("uppercase");
assertThat(msg.getHeaders()
.get(FunctionRSocketMessageHandler.RECONSILED_LOOKUP_DESTINATION_HEADER)).toString().equals(RoutingFunction.FUNCTION_NAME);
return msg.getPayload().toUpperCase();
};
}
@Bean
public Function<String, String> concat() {
return v -> v + v;