Add user-name header to CONNECTED frame

Author: rstoyanchev

spring-messaging/src/main/java/org/springframework/messaging/simp/stomp/StompWebSocketHandler.java

@@ -43,6 +43,13 @@
  */
 public class StompWebSocketHandler extends TextWebSocketHandlerAdapter implements MessageHandler {
 
+	/**
+	 * The name of the header set on the CONNECTED frame indicating the name of the user
+	 * connected authenticated on the WebSocket session.
+	 */
+	public static final String CONNECTED_USER_HEADER = "user-name";
+
+
 	private static final byte[] EMPTY_PAYLOAD = new byte[0];
 
 	private static Log logger = LogFactory.getLog(StompWebSocketHandler.class);
@@ -138,6 +145,10 @@ else if (acceptVersions.isEmpty()) {
 		}
 		connectedHeaders.setHeartbeat(0,0); // TODO
 
+		if (session.getPrincipal() != null) {
+			connectedHeaders.setNativeHeader(CONNECTED_USER_HEADER, session.getPrincipal().getName());
+		}
+
 		// TODO: security
 
 		Message<?> connectedMessage = MessageBuilder.withPayload(EMPTY_PAYLOAD).copyHeaders(

spring-messaging/src/main/java/org/springframework/messaging/support/NativeMessageHeaderAccessor.java

@@ -118,7 +118,7 @@ else if (this.originalNativeHeaders != null) {
 		return null;
 	}
 
-	protected String getFirstNativeHeader(String headerName) {
+	public String getFirstNativeHeader(String headerName) {
 		List<String> values = getNativeHeader(headerName);
 		return CollectionUtils.isEmpty(values) ? null : values.get(0);
 	}
@@ -133,7 +133,7 @@ protected void putNativeHeader(String name, List<String> value) {
 		}
 	}
 
-	protected void setNativeHeader(String name, String value) {
+	public void setNativeHeader(String name, String value) {
 		this.nativeHeaders.set(name, value);
 	}