Merge pull request TooTallNate#712 from PhilipRoman/master

marci4 · web-flow · commit 41cf9727e0f2 · 2018-05-27T18:53:28.000+02:00
Added a broadcast method for ByteBuffers
diff --git a/src/main/java/org/java_websocket/server/WebSocketServer.java b/src/main/java/org/java_websocket/server/WebSocketServer.java
@@ -812,23 +812,42 @@ public void broadcast(byte[] data) {
 		broadcast( data, connections );
 	}
 
+	/**
+	 * Send a ByteBuffer to all connected endpoints
+	 * @param data the data to send to the endpoints
+	 */
+	public void broadcast(ByteBuffer data) {
+		broadcast(data, connections);
+	}
+
 	/**
 	 * Send a byte array to a specific collection of websocket connections
 	 * @param data the data to send to the endpoints
 	 * @param clients a collection of endpoints to whom the text has to be send
 	 */
 	public void broadcast(byte[] data, Collection<WebSocket> clients) {
+		if (data == null || clients == null) {
+			throw new IllegalArgumentException();
+		}
+		broadcast(ByteBuffer.wrap(data), clients);
+	}
+
+	/**
+	 * Send a ByteBuffer to a specific collection of websocket connections
+	 * @param data the data to send to the endpoints
+	 * @param clients a collection of endpoints to whom the text has to be send
+	 */
+	public void broadcast(ByteBuffer data, Collection<WebSocket> clients) {
 		if (data == null || clients == null) {
 			throw new IllegalArgumentException();
 		}
 		Map<Draft, List<Framedata>> draftFrames = new HashMap<Draft, List<Framedata>>();
-		ByteBuffer byteBufferData = ByteBuffer.wrap( data );
 		synchronized( clients ) {
 			for( WebSocket client : clients ) {
 				if( client != null ) {
 					Draft draft = client.getDraft();
 					if( !draftFrames.containsKey( draft ) ) {
-						List<Framedata> frames = draft.createFrames( byteBufferData, false );
+						List<Framedata> frames = draft.createFrames( data, false );
 						draftFrames.put( draft, frames );
 					}
 					try {
