From 842e93507c7a7f1c922917b8a6187b0d8f9390f1 Mon Sep 17 00:00:00 2001 From: Redkale <22250530@qq.com> Date: Sat, 1 Jul 2017 15:46:51 +0800 Subject: [PATCH] =?UTF-8?q?WebSocketEngine.broadcastMessage=20=E5=A2=9E?= =?UTF-8?q?=E5=8A=A0=20Predicate=20=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/org/redkale/net/http/WebSocketEngine.java | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/org/redkale/net/http/WebSocketEngine.java b/src/org/redkale/net/http/WebSocketEngine.java index 1c107f7ec..33c68363e 100644 --- a/src/org/redkale/net/http/WebSocketEngine.java +++ b/src/org/redkale/net/http/WebSocketEngine.java @@ -162,11 +162,13 @@ public final class WebSocketEngine { CompletableFuture future = null; if (single) { for (WebSocket websocket : websockets.values()) { + if (predicate != null && !predicate.test(websocket)) continue; future = future == null ? websocket.send(message, last) : future.thenCombine(websocket.send(message, last), (a, b) -> a | (Integer) b); } } else { for (List list : websockets2.values()) { for (WebSocket websocket : list) { + if (predicate != null && !predicate.test(websocket)) continue; future = future == null ? websocket.send(message, last) : future.thenCombine(websocket.send(message, last), (a, b) -> a | (Integer) b); } }