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

GH-884 Add initial support for BiFunction

Browse Source 
Resolves #884
This commit is contained in:
Oleg Zhurakousky
2022-06-14 09:08:31 +02:00
parent 78dea0aad7
commit 8789c7ebd7
4 changed files with 65 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
spring-cloud-function-context/src/test/java/org/springframework/cloud/function/context/catalog/BeanFactoryAwareFunctionRegistryTests.java
View File

@@ -34,6 +34,7 @@ import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.BiFunction;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
@@ -185,18 +186,19 @@ public class BeanFactoryAwareFunctionRegistryTests {
assertThat(((FunctionInvocationWrapper) function).isComposed()).isTrue();
}
@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testBiFunction() {
FunctionCatalog catalog = this.configureCatalog();
Function biFunction = catalog.lookup("biFuncUpperCase");
assertThat(biFunction.apply("hello")).isEqualTo("HELLO");
}
@Test
public void testImperativeFunction() {
FunctionCatalog catalog = this.configureCatalog();
// Function<String, String> asIs = catalog.lookup("uppercase");
// assertThat(asIs.apply("uppercase")).isEqualTo("UPPERCASE");
//
// Function<Flux<String>, Flux<String>> asFlux = catalog.lookup("uppercase");
// List<String> result = asFlux.apply(Flux.just("uppercaseFlux", "uppercaseFlux2")).collectList().block();
// assertThat(result.get(0)).isEqualTo("UPPERCASEFLUX");
// assertThat(result.get(1)).isEqualTo("UPPERCASEFLUX2");
Function<Flux<Message<byte[]>>, Flux<Message<byte[]>>> messageFlux = catalog.lookup("uppercase", "application/json");
Message<byte[]> message1 = MessageBuilder.withPayload("\"uppercaseFlux\"".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "application/json").build();
Message<byte[]> message2 = MessageBuilder.withPayload("\"uppercaseFlux2\"".getBytes()).setHeader(MessageHeaders.CONTENT_TYPE, "application/json").build();
@@ -1052,6 +1054,13 @@ public class BeanFactoryAwareFunctionRegistryTests {
return () -> "one";
}
@Bean
public BiFunction<String, Map, String> biFuncUpperCase() {
return (p, h) -> {
return p.toUpperCase();
};
}
@Bean
public Function<Map<String, Object>, Person> maptopojo() {
return map -> {