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

Update STOMP MESSAGE frames with messageId

Browse Source
This commit is contained in:
Rossen Stoyanchev
2013-06-11 10:49:31 -04:00
parent d26b9d60e5
commit e7dde941b7
3 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
spring-websocket/src/main/java/org/springframework/web/messaging/PubSubHeaders.java
View File

@@ -17,6 +17,7 @@
package org.springframework.web.messaging;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -85,7 +86,7 @@ public class PubSubHeaders {
this.messageHeaders = readOnly ? messageHeaders : new HashMap<String, Object>(messageHeaders);
this.rawHeaders = this.messageHeaders.containsKey(RAW_HEADERS) ?
(Map<String, String>) messageHeaders.get(RAW_HEADERS) : new HashMap<String, String>();
(Map<String, String>) messageHeaders.get(RAW_HEADERS) : Collections.<String, String>emptyMap();
if (this.messageHeaders.get(MESSAGE_TYPE) == null) {
this.messageHeaders.put(MESSAGE_TYPE, MessageType.MESSAGE);

3
spring-websocket/src/main/java/org/springframework/web/messaging/service/AbstractMessageService.java
View File

@@ -75,7 +75,8 @@ public abstract class AbstractMessageService {
logger.trace("Processing notification: " + message);
}
MessageType messageType = (MessageType) message.getHeaders().get("messageType");
PubSubHeaders headers = new PubSubHeaders(message.getHeaders(), true);
MessageType messageType = headers.getMessageType();
if (messageType == null || messageType.equals(MessageType.OTHER)) {
processOther(message);
}

3
spring-websocket/src/main/java/org/springframework/web/messaging/stomp/StompHeaders.java
View File

@@ -224,6 +224,9 @@ public class StompHeaders extends PubSubHeaders {
if (subscriptionId != null) {
getRawHeaders().put(SUBSCRIPTION, subscriptionId);
}
if (StompCommand.MESSAGE.equals(getStompCommand()) && (getMessageId() == null)) {
getRawHeaders().put(MESSAGE_ID, getMessageHeaders().get(ID).toString());
}
}
}