Spring/spring-amqp
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add typed header getter method

Browse Source 
* Polishing - PR Comments
This commit is contained in:
Gary Russell
2019-04-16 13:14:04 -04:00
committed by Artem Bilan
parent d3ded64e7c
commit 529bb63d86
2 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
spring-amqp/src/main/java/org/springframework/amqp/core/MessageProperties.java
View File

@@ -136,6 +136,18 @@ public class MessageProperties implements Serializable {
this.headers.put(key, value);
}
/**
* Typed getter for a header.
* @param headerName the header name.
* @param <T> the type.
* @return the header value
* @since 2.2
*/
@SuppressWarnings("unchecked")
public <T> T getHeader(String headerName) {
return (T) this.headers.get(headerName);
}
public Map<String, Object> getHeaders() {
return this.headers;
}

11
spring-rabbit/src/test/java/org/springframework/amqp/rabbit/annotation/EnableRabbitIntegrationTests.java
View File

@@ -546,6 +546,8 @@ public class EnableRabbitIntegrationTests {
MessagePostProcessor messagePostProcessor = message -> {
message.getMessageProperties().setContentType("application/json");
message.getMessageProperties().setUserId("guest");
message.getMessageProperties().setHeader("stringHeader", "string");
message.getMessageProperties().setHeader("intHeader", 42);
return message;
};
returned = template.convertSendAndReceive("", "test.converted", "{ \"bar\" : \"baz\" }", messagePostProcessor);
@@ -581,6 +583,9 @@ public class EnableRabbitIntegrationTests {
messagePostProcessor);
assertThat(returned, instanceOf(byte[].class));
assertEquals("\"fooMessage\"", new String((byte[]) returned));
Foo2Service foo2service = ctx.getBean(Foo2Service.class);
assertThat(foo2service.stringHeader, equalTo("string"));
assertThat(foo2service.intHeader, equalTo(42));
returned = template.convertSendAndReceive("", "test.notconverted.channel", "{ \"bar\" : \"baz\" }",
messagePostProcessor);
@@ -1953,6 +1958,10 @@ public class EnableRabbitIntegrationTests {
public static class Foo2Service {
String stringHeader;
Integer intHeader;
@RabbitListener(queues = "test.converted")
public Foo2 foo2(Foo2 foo2) {
return foo2;
@@ -1990,6 +1999,8 @@ public class EnableRabbitIntegrationTests {
@RabbitListener(queues = "test.notconverted.message")
public String justMessage(Message message) {
this.stringHeader = message.getMessageProperties().getHeader("stringHeader");
this.intHeader = message.getMessageProperties().getHeader("intHeader");
return "foo" + message.getClass().getSimpleName();
}