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

RSocket - Add Cbor test, fixed consumer test and minor cleanup

Browse Source 
polish
This commit is contained in:
Oleg Zhurakousky
2021-03-16 18:50:23 +01:00
parent 85515c12f9
commit 835f3cc723
5 changed files with 111 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
spring-cloud-function-rsocket/src/main/java/org/springframework/cloud/function/rsocket/FunctionRSocketMessageHandler.java
View File

@@ -264,23 +264,24 @@ class FunctionRSocketMessageHandler extends RSocketMessageHandler {
return MessageBuilder.withPayload(structure).build();
}
}
else {
return MessageBuilder.withPayload(structure).build();
}
}
return MessageBuilder.withPayload(bytePayload).copyHeadersIfAbsent(message.getHeaders()).build();
});
return MessageBuilder.createMessage(argument, message.getHeaders());
}
else {
else { // delegate to the existing argument resolvers
for (HandlerMethodArgumentResolver handlerMethodArgumentResolver : this.resolvers) {
if (handlerMethodArgumentResolver.supportsParameter(parameter)) {
Publisher<?> arg = handlerMethodArgumentResolver.resolveArgument(parameter, message);
return MessageBuilder.withPayload(arg).copyHeadersIfAbsent(message.getHeaders()).build();
}
}
return message;
}
}
}
protected static final class FunctionRSocketPayloadReturnValueHandler extends RSocketPayloadReturnValueHandler {

2
spring-cloud-function-rsocket/src/main/java/org/springframework/cloud/function/rsocket/MessageAwareJsonDecoder.java
View File

@@ -55,7 +55,7 @@ class MessageAwareJsonDecoder extends AbstractDecoder<Object> {
@Override
public boolean canDecode(ResolvableType elementType, @Nullable MimeType mimeType) {
return mimeType.isCompatibleWith(MimeTypeUtils.APPLICATION_JSON);
return mimeType != null && mimeType.isCompatibleWith(MimeTypeUtils.APPLICATION_JSON);
}
@SuppressWarnings("unchecked")

2
spring-cloud-function-rsocket/src/main/java/org/springframework/cloud/function/rsocket/MessageAwareJsonEncoder.java
View File

@@ -66,7 +66,7 @@ class MessageAwareJsonEncoder extends AbstractEncoder<Object> {
@Override
public boolean canEncode(ResolvableType elementType, MimeType mimeType) {
boolean canEncode = mimeType.isCompatibleWith(MimeTypeUtils.APPLICATION_JSON);
boolean canEncode = mimeType != null && mimeType.isCompatibleWith(MimeTypeUtils.APPLICATION_JSON);
if (canEncode && this.isClient) {
canEncode = (FunctionTypeUtils.isMessage(elementType.getType())
|| Map.class.isAssignableFrom(FunctionTypeUtils.getRawType(elementType.getType())));

87
spring-cloud-function-rsocket/src/main/java/org/springframework/cloud/function/rsocket/RSocketListenerFunction.java
View File

@@ -46,56 +46,51 @@ class RSocketListenerFunction implements Function<Object, Publisher<?>> {
private final FunctionInvocationWrapper targetFunction;
RSocketListenerFunction(FunctionInvocationWrapper targetFunction) {
this.targetFunction = targetFunction;
}
/*
* We need to maintain the input typeless to ensure that no encoder/decoders will attempt any conversion.
* That said it will always be Message<Publisher<Object>>
*/
@SuppressWarnings("unchecked")
@Override
public Publisher<?> apply(Object input) {
Assert.isTrue(this.targetFunction != null, "Failed to discover target function. \n"
Assert.isTrue(targetFunction != null, "Failed to discover target function. \n"
+ "To fix it you should either provide 'spring.cloud.function.definition' property "
+ "or if you are using RSocketRequester provide valid function definition via 'route' "
+ "operator (e.g., requester.route(\"echo\"))");
// if (input instanceof Message) {
Message<Publisher<Object>> inputMessage = (Message<Publisher<Object>>) input;
FrameType frameType = RSocketFrameTypeMessageCondition.getFrameType(inputMessage);
switch (frameType) {
case REQUEST_FNF:
return handle(inputMessage);
case REQUEST_RESPONSE:
case REQUEST_STREAM:
case REQUEST_CHANNEL:
return handleAndReply(inputMessage);
default:
throw new UnsupportedOperationException();
}
// }
// throw new UnsupportedOperationException("Expecting input to be of type Message<Publisher<Object>>");
this.targetFunction = targetFunction;
}
@SuppressWarnings("unchecked")
@Override
public Publisher<?> apply(Object input) {
/*
* We need to maintain the input typeless to ensure that no encoder/decoders will attempt any conversion.
* That said it will always be Message<Publisher<Object>>
*/
Message<Publisher<Object>> inputMessage = (Message<Publisher<Object>>) input;
FrameType frameType = RSocketFrameTypeMessageCondition.getFrameType(inputMessage);
switch (frameType) {
case REQUEST_FNF:
return handle(inputMessage);
case REQUEST_RESPONSE:
case REQUEST_STREAM:
case REQUEST_CHANNEL:
return handleAndReply(inputMessage);
default:
throw new UnsupportedOperationException();
}
}
@SuppressWarnings({ "unchecked", "rawtypes" })
private Mono<Void> handle(Message<Publisher<Object>> messageToProcess) {
if (this.targetFunction.isRoutingFunction()) {
Flux<?> dataFlux = Flux.from(messageToProcess.getPayload())
.map((payload) -> {
return MessageBuilder.createMessage(payload, messageToProcess.getHeaders());
});
.map(payload -> MessageBuilder.createMessage(payload, messageToProcess.getHeaders()));
return dataFlux.doOnNext(this.targetFunction).then();
}
else if (this.targetFunction.isConsumer()) {
Flux<?> dataFlux =
Flux.from(messageToProcess.getPayload())
.map((payload) -> MessageBuilder.createMessage(payload, messageToProcess.getHeaders()));
if (FunctionTypeUtils.isPublisher(this.targetFunction.getInputType())) {
dataFlux = dataFlux.transform((Function) this.targetFunction);
}
else {
dataFlux = dataFlux.doOnNext(this.targetFunction);
}
Flux<?> dataFlux = Flux.from(messageToProcess.getPayload())
.map(payload -> this.buildReceivedMessage(payload, messageToProcess.getHeaders()));
dataFlux = FunctionTypeUtils.isPublisher(this.targetFunction.getInputType())
? dataFlux.transform((Function) this.targetFunction)
: dataFlux.doOnNext(this.targetFunction);
return dataFlux.then();
}
else {
@@ -105,13 +100,9 @@ class RSocketListenerFunction implements Function<Object, Publisher<?>> {
@SuppressWarnings({ "unchecked", "rawtypes" })
private Flux<?> handleAndReply(Message<Publisher<Object>> messageToProcess) {
Flux<?> dataFlux =
Flux.from(messageToProcess.getPayload())
.map((payload) -> {
return payload instanceof Message
? MessageBuilder.fromMessage((Message<?>) payload).copyHeadersIfAbsent(messageToProcess.getHeaders()).build()
: MessageBuilder.withPayload(payload).copyHeadersIfAbsent(messageToProcess.getHeaders()).build();
});
Flux<?> dataFlux = Flux.from(messageToProcess.getPayload())
.map(payload -> this.buildReceivedMessage(payload, messageToProcess.getHeaders()));
if (this.targetFunction.getInputType() != null && FunctionTypeUtils.isPublisher(this.targetFunction.getInputType())) {
dataFlux = dataFlux.transform((Function) this.targetFunction);
}
@@ -132,6 +123,12 @@ class RSocketListenerFunction implements Function<Object, Publisher<?>> {
return dataFlux;
}
private Message<?> buildReceivedMessage(Object mayBeMessage, MessageHeaders messageHeaders) {
return mayBeMessage instanceof Message
? MessageBuilder.fromMessage((Message<?>) mayBeMessage).copyHeadersIfAbsent(messageHeaders).build()
: MessageBuilder.withPayload(mayBeMessage).copyHeadersIfAbsent(messageHeaders).build();
}
/*
* This will ensure that unless CT is application/json for which we provide Message aware encoder/decoder
* the payload is extracted since no other available encoders/decoders understand Message.