GH-873 Fix JsonNode conversion

Resolves #873
This commit is contained in:
Oleg Zhurakousky
2022-05-30 14:38:34 +02:00
parent ca428e7ad6
commit 94bd64456e
2 changed files with 36 additions and 1 deletions

View File

@@ -31,6 +31,7 @@ import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Stream;
import com.fasterxml.jackson.databind.JsonNode;
import net.jodah.typetools.TypeResolver;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
@@ -52,6 +53,8 @@ import org.springframework.util.CollectionUtils;
import org.springframework.util.ReflectionUtils;
import org.springframework.util.StringUtils;
/**
* Set of utility operations to interrogate function definitions.
*
@@ -82,13 +85,17 @@ public final class FunctionTypeUtils {
}
type = getGenericType(type);
Class<?> rawType = type instanceof ParameterizedType ? getRawType(type) : (Class<?>) type;
return Collection.class.isAssignableFrom(rawType);
return Collection.class.isAssignableFrom(rawType) || JsonNode.class.isAssignableFrom(rawType);
}
public static boolean isTypeArray(Type type) {
return getRawType(type).isArray();
}
public static boolean isJsonNode(Type type) {
return getRawType(type).isArray();
}
/**
* A convenience method identical to {@link #getImmediateGenericType(Type, int)}
* for cases when provided 'type' is {@link Publisher} or {@link Message}.

View File

@@ -39,6 +39,7 @@ import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collectors;
import com.fasterxml.jackson.databind.JsonNode;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Disabled;
@@ -104,6 +105,15 @@ public class BeanFactoryAwareFunctionRegistryTests {
System.clearProperty("spring.cloud.function.definition");
}
@Test
public void testJsonNodeAsInput() throws Exception {
FunctionCatalog catalog = this.configureCatalog(JsonNodeConfiguration.class);
Function<Message<String>, Message<byte[]>> f = catalog.lookup("messageAsJsonNode", "application/json");
Message<String> m = MessageBuilder.withPayload("[{\"name\":\"bob\"}, {\"name\":\"bob\"}]").setHeader(MessageHeaders.CONTENT_TYPE, "application/json").build();
assertThat(new String(f.apply(m).getPayload())).isEqualTo("[{\"name\":\"bob\"},{\"name\":\"bob\"}]");
f = catalog.lookup("asJsonNode", "application/json");
assertThat(new String(f.apply(m).getPayload())).isEqualTo("[{\"name\":\"bob\"},{\"name\":\"bob\"}]");
}
@SuppressWarnings({ "rawtypes" })
@Test
@@ -782,6 +792,24 @@ public class BeanFactoryAwareFunctionRegistryTests {
}
}
@EnableAutoConfiguration
@Configuration
public static class JsonNodeConfiguration {
@Bean
public Function<Message<JsonNode>, String> messageAsJsonNode() {
return v -> {
return v.getPayload().toString();
};
}
@Bean
public Function<JsonNode, String> asJsonNode() {
return v -> {
return v.toString();
};
}
}
@EnableAutoConfiguration
public static class EmptyConfiguration {