GH-585 Add initial support for passing/parsing Message headers

This woudl allow Message headers to be passed as JSON string via rsocket metadata

Resolves #585
This commit is contained in:
Oleg Zhurakousky
2020-09-01 13:21:58 +02:00
parent 5f586b7365
commit f2f1432912
2 changed files with 56 additions and 1 deletions

View File

@@ -31,12 +31,14 @@ import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.builder.SpringApplicationBuilder;
import org.springframework.boot.rsocket.context.RSocketServerBootstrap;
import org.springframework.boot.rsocket.server.RSocketServer;
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.core.env.ConfigurableEnvironment;
import org.springframework.messaging.rsocket.RSocketRequester;
import org.springframework.test.util.ReflectionTestUtils;
import org.springframework.util.MimeTypeUtils;
import org.springframework.util.SocketUtils;
/**
@@ -220,7 +222,6 @@ public class RSocketAutoConfigurationTests {
}
}
// @Disabled
@Test
public void testRequestReplyFunctionWithComposition() {
int portA = SocketUtils.findAvailableTcpPort();
@@ -367,6 +368,32 @@ public class RSocketAutoConfigurationTests {
}
}
@Test
public void testRoutingWithRoutingFunction() {
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.routing-expression=headers.function_definition",
"--spring.rsocket.server.port=" + port);
) {
RSocketRequester.Builder rsocketRequesterBuilder =
applicationContext.getBean(RSocketRequester.Builder.class);
rsocketRequesterBuilder.tcp("localhost", port)
.route(RoutingFunction.FUNCTION_NAME)
.metadata("{\"function_definition\":\"uppercase|concat\"}", MimeTypeUtils.APPLICATION_JSON)
.data("\"hello\"")
.retrieveMono(String.class)
.as(StepVerifier::create)
.expectNext("\"HELLOHELLO\"")
.expectComplete()
.verify();
}
}
@EnableAutoConfiguration
@Configuration