Reintroduced MessageMethodArgumentResolver default constructor

Issue: SPR-14616
This commit is contained in:
Juergen Hoeller 2016-08-23 13:20:02 +02:00
parent 64bc0ca744
commit c4fff6db1b
2 changed files with 62 additions and 20 deletions

View File

@ -26,7 +26,6 @@ import org.springframework.messaging.converter.MessageConverter;
import org.springframework.messaging.converter.SmartMessageConverter; import org.springframework.messaging.converter.SmartMessageConverter;
import org.springframework.messaging.handler.invocation.HandlerMethodArgumentResolver; import org.springframework.messaging.handler.invocation.HandlerMethodArgumentResolver;
import org.springframework.messaging.support.MessageBuilder; import org.springframework.messaging.support.MessageBuilder;
import org.springframework.util.Assert;
import org.springframework.util.ClassUtils; import org.springframework.util.ClassUtils;
import org.springframework.util.StringUtils; import org.springframework.util.StringUtils;
@ -38,6 +37,7 @@ import org.springframework.util.StringUtils;
* *
* @author Rossen Stoyanchev * @author Rossen Stoyanchev
* @author Stephane Nicoll * @author Stephane Nicoll
* @author Juergen Hoeller
* @since 4.0 * @since 4.0
*/ */
public class MessageMethodArgumentResolver implements HandlerMethodArgumentResolver { public class MessageMethodArgumentResolver implements HandlerMethodArgumentResolver {
@ -46,12 +46,18 @@ public class MessageMethodArgumentResolver implements HandlerMethodArgumentResol
/** /**
* Create a new instance with the given {@link MessageConverter}. * Create a default resolver instance without message conversion.
* @param converter the MessageConverter to use (required) */
* @since 4.1 public MessageMethodArgumentResolver() {
this(null);
}
/**
* Create a resolver instance with the given {@link MessageConverter}.
* @param converter the MessageConverter to use (may be {@code null})
* @since 4.3
*/ */
public MessageMethodArgumentResolver(MessageConverter converter) { public MessageMethodArgumentResolver(MessageConverter converter) {
Assert.notNull(converter, "MessageConverter must not be null");
this.converter = converter; this.converter = converter;
} }
@ -63,7 +69,6 @@ public class MessageMethodArgumentResolver implements HandlerMethodArgumentResol
@Override @Override
public Object resolveArgument(MethodParameter parameter, Message<?> message) throws Exception { public Object resolveArgument(MethodParameter parameter, Message<?> message) throws Exception {
Class<?> targetMessageType = parameter.getParameterType(); Class<?> targetMessageType = parameter.getParameterType();
Class<?> targetPayloadType = getPayloadType(parameter); Class<?> targetPayloadType = getPayloadType(parameter);
@ -117,20 +122,20 @@ public class MessageMethodArgumentResolver implements HandlerMethodArgumentResol
} }
private Object convertPayload(Message<?> message, MethodParameter parameter, Class<?> targetPayloadType) { private Object convertPayload(Message<?> message, MethodParameter parameter, Class<?> targetPayloadType) {
Object result; Object result = null;
if (this.converter instanceof SmartMessageConverter) { if (this.converter instanceof SmartMessageConverter) {
SmartMessageConverter smartConverter = (SmartMessageConverter) this.converter; SmartMessageConverter smartConverter = (SmartMessageConverter) this.converter;
result = smartConverter.fromMessage(message, targetPayloadType, parameter); result = smartConverter.fromMessage(message, targetPayloadType, parameter);
} }
else { else if (this.converter != null) {
result = this.converter.fromMessage(message, targetPayloadType); result = this.converter.fromMessage(message, targetPayloadType);
} }
if (result == null) { if (result == null) {
String actual = ClassUtils.getQualifiedName(targetPayloadType); String actual = ClassUtils.getQualifiedName(targetPayloadType);
String expected = ClassUtils.getQualifiedName(message.getPayload().getClass()); String expected = ClassUtils.getQualifiedName(message.getPayload().getClass());
throw new MessageConversionException(message, "No converter found to convert payload " + throw new MessageConversionException(message, "No converter found to convert payload type [" +
"type [" + actual + "] to expected payload type [" + expected + "]."); actual + "] to expected payload type [" + expected + "]");
} }
return result; return result;
} }

View File

@ -33,14 +33,13 @@ import org.springframework.messaging.support.GenericMessage;
import org.springframework.messaging.support.MessageBuilder; import org.springframework.messaging.support.MessageBuilder;
import static org.junit.Assert.*; import static org.junit.Assert.*;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.*;
import static org.mockito.Mockito.when;
/** /**
* Unit tests for * Unit tests for {@link MessageMethodArgumentResolver}.
* {@link org.springframework.messaging.handler.annotation.support.MessageMethodArgumentResolver}.
* *
* @author Stephane Nicoll * @author Stephane Nicoll
* @author Juergen Hoeller
*/ */
public class MessageMethodArgumentResolverTests { public class MessageMethodArgumentResolverTests {
@ -56,10 +55,8 @@ public class MessageMethodArgumentResolverTests {
@Before @Before
public void setup() throws Exception { public void setup() throws Exception {
this.method = MessageMethodArgumentResolverTests.class.getDeclaredMethod("handle", this.method = MessageMethodArgumentResolverTests.class.getDeclaredMethod("handle",
Message.class, Message.class, Message.class, Message.class, Message.class, Message.class, Message.class, Message.class, ErrorMessage.class);
ErrorMessage.class);
this.converter = mock(MessageConverter.class); this.converter = mock(MessageConverter.class);
this.resolver = new MessageMethodArgumentResolver(this.converter); this.resolver = new MessageMethodArgumentResolver(this.converter);
@ -85,7 +82,7 @@ public class MessageMethodArgumentResolverTests {
} }
@Test @Test
public void resolveWithPayloadTypeSubClass() throws Exception { public void resolveWithPayloadTypeSubclass() throws Exception {
Message<Integer> message = MessageBuilder.withPayload(123).build(); Message<Integer> message = MessageBuilder.withPayload(123).build();
MethodParameter parameter = new MethodParameter(this.method, 2); MethodParameter parameter = new MethodParameter(this.method, 2);
@ -155,7 +152,7 @@ public class MessageMethodArgumentResolverTests {
} }
@Test @Test
public void resolveMessageSubClassMatch() throws Exception { public void resolveMessageSubclassMatch() throws Exception {
ErrorMessage message = new ErrorMessage(new UnsupportedOperationException()); ErrorMessage message = new ErrorMessage(new UnsupportedOperationException());
MethodParameter parameter = new MethodParameter(this.method, 4); MethodParameter parameter = new MethodParameter(this.method, 4);
@ -164,7 +161,7 @@ public class MessageMethodArgumentResolverTests {
} }
@Test @Test
public void resolveWithMessageSubClassAndPayloadWildcard() throws Exception { public void resolveWithMessageSubclassAndPayloadWildcard() throws Exception {
ErrorMessage message = new ErrorMessage(new UnsupportedOperationException()); ErrorMessage message = new ErrorMessage(new UnsupportedOperationException());
MethodParameter parameter = new MethodParameter(this.method, 0); MethodParameter parameter = new MethodParameter(this.method, 0);
@ -185,6 +182,46 @@ public class MessageMethodArgumentResolverTests {
assertSame(message, this.resolver.resolveArgument(parameter, message)); assertSame(message, this.resolver.resolveArgument(parameter, message));
} }
@Test
public void resolveWithPayloadTypeAsWildcardAndNoConverter() throws Exception {
this.resolver = new MessageMethodArgumentResolver();
Message<String> message = MessageBuilder.withPayload("test").build();
MethodParameter parameter = new MethodParameter(this.method, 0);
assertTrue(this.resolver.supportsParameter(parameter));
assertSame(message, this.resolver.resolveArgument(parameter, message));
}
@Test
public void resolveWithConversionNeededButNoConverter() throws Exception {
this.resolver = new MessageMethodArgumentResolver();
Message<String> message = MessageBuilder.withPayload("test").build();
MethodParameter parameter = new MethodParameter(this.method, 1);
assertTrue(this.resolver.supportsParameter(parameter));
thrown.expect(MessageConversionException.class);
thrown.expectMessage(Integer.class.getName());
thrown.expectMessage(String.class.getName());
this.resolver.resolveArgument(parameter, message);
}
@Test
public void resolveWithConversionEmptyPayloadButNoConverter() throws Exception {
this.resolver = new MessageMethodArgumentResolver();
Message<String> message = MessageBuilder.withPayload("").build();
MethodParameter parameter = new MethodParameter(this.method, 1);
assertTrue(this.resolver.supportsParameter(parameter));
thrown.expect(MessageConversionException.class);
thrown.expectMessage("the payload is empty");
thrown.expectMessage(Integer.class.getName());
thrown.expectMessage(String.class.getName());
this.resolver.resolveArgument(parameter, message);
}
@SuppressWarnings("unused") @SuppressWarnings("unused")
private void handle( private void handle(