SpringCloud/spring-cloud-function
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

GH-654 Remove FunctionProperties injection from MessageRoutingCallbackRSocketTests

Browse Source
This commit is contained in:
Oleg Zhurakousky
2021-02-24 13:31:31 +01:00
parent 410d494467
commit 1f43bbce28
3 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/MessageRoutingCallback.java
View File

@@ -37,8 +37,7 @@ public interface MessageRoutingCallback {
* Determines the name of the function definition to route incoming {@link Message}.
*
* @param message instance of incoming {@link Message}
* @param functionProperties instance of {@link FunctionProperties}
* @return the name of the route-to function definition
*/
String functionDefinition(Message<?> message, FunctionProperties functionProperties);
String functionDefinition(Message<?> message);
}

2
spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/RoutingFunction.java
View File

@@ -174,7 +174,7 @@ public class RoutingFunction implements Function<Object, Object> {
private FunctionInvocationWrapper functionFromCallback(Object input) {
if (input instanceof Message) {
String functionDefinition = this.routingCallback.functionDefinition((Message<?>) input, this.functionProperties);
String functionDefinition = this.routingCallback.functionDefinition((Message<?>) input);
if (StringUtils.hasText(functionDefinition)) {
return this.functionFromDefinition(functionDefinition);
}

3
spring-cloud-function-rsocket/src/test/java/org/springframework/cloud/function/rsocket/MessageRoutingCallbackRSocketTests.java
View File

@@ -25,7 +25,6 @@ import reactor.test.StepVerifier;
import org.springframework.boot.WebApplicationType;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.builder.SpringApplicationBuilder;
import org.springframework.cloud.function.context.FunctionProperties;
import org.springframework.cloud.function.context.MessageRoutingCallback;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
@@ -110,7 +109,7 @@ public class MessageRoutingCallbackRSocketTests {
public MessageRoutingCallback customRouter() {
return new MessageRoutingCallback() {
@Override
public String functionDefinition(Message<?> message, FunctionProperties functionProperties) {
public String functionDefinition(Message<?> message) {
return (String) message.getHeaders().get("func_name");
}
};