Minor tweaks to WebSocketMessage

- payload is now required
- byte data is obtained and stored only once (allowing multiple calls)
- minor formatting polish

spring-websocket/src/main/java/org/springframework/websocket/BinaryMessage.java

@@ -19,6 +19,8 @@ import java.io.ByteArrayInputStream;
 import java.io.InputStream;
 import java.nio.ByteBuffer;
 
+import org.springframework.util.Assert;
+
 
 /**
  * Represents a binary WebSocket message.
@@ -28,7 +30,7 @@ import java.nio.ByteBuffer;
  */
 public final class BinaryMessage extends WebSocketMessage<ByteBuffer> {
 
-	private final byte[] bytes;
+	private byte[] bytes;
 
 	private final boolean last;
 
@@ -48,8 +50,19 @@ public final class BinaryMessage extends WebSocketMessage<ByteBuffer> {
 	}
 
 	public BinaryMessage(byte[] payload, boolean isLast) {
-		super((payload != null) ? ByteBuffer.wrap(payload) : null);
+		this(payload, 0, (payload == null ? 0 : payload.length), isLast);
+	}
+
+	public BinaryMessage(byte[] payload, int offset, int len) {
+		this(payload, offset, len, true);
+	}
+
+	public BinaryMessage(byte[] payload, int offset, int len, boolean isLast) {
+		super(payload != null ? ByteBuffer.wrap(payload, offset, len) : null);
+		if(payload != null && offset == 0 && len == payload.length) {
+			// FIXME better if a message always needs a payload?
 			this.bytes = payload;
+		}
 		this.last = isLast;
 	}
 
@@ -58,18 +71,17 @@ public final class BinaryMessage extends WebSocketMessage<ByteBuffer> {
 	}
 
 	public byte[] getByteArray() {
-		if (this.bytes != null) {
+		if(this.bytes == null && getPayload() != null) {
+			this.bytes = getRemainingBytes(getPayload());
+		}
 		return this.bytes;
 	}
-		else if (getPayload() != null){
+
+	private byte[] getRemainingBytes(ByteBuffer payload) {
 		byte[] result = new byte[getPayload().remaining()];
 		getPayload().get(result);
 		return result;
 	}
-		else {
-			return null;
-		}
-	}
 
 	public InputStream getInputStream() {
 		byte[] array = getByteArray();

spring-websocket/src/main/java/org/springframework/websocket/TextMessage.java

@@ -18,7 +18,6 @@ package org.springframework.websocket;
 import java.io.Reader;
 import java.io.StringReader;
 
-
 /**
  * Represents a text WebSocket message.
  *
@@ -27,7 +26,6 @@ import java.io.StringReader;
  */
 public final class TextMessage extends WebSocketMessage<String> {
 
-
 	public TextMessage(String payload) {
 		super(payload);
 	}

spring-websocket/src/main/java/org/springframework/websocket/WebSocketMessage.java

@@ -15,6 +15,7 @@
  */
 package org.springframework.websocket;
 
+import org.springframework.util.Assert;
 import org.springframework.util.ObjectUtils;
 
 
@@ -29,6 +30,7 @@ public abstract class WebSocketMessage<T> {
 
 
 	WebSocketMessage(T payload) {
+		Assert.notNull(payload, "Payload must not be null");
 		this.payload = payload;
 	}