From 13cf188e255dd51250fa940291c8fd1b10688982 Mon Sep 17 00:00:00 2001
From: Redkale <22250530@qq.com>
Date: Thu, 14 Dec 2017 19:10:49 +0800
Subject: [PATCH] =?UTF-8?q?WebSocketEngine=E5=A2=9E=E5=8A=A0forEachLocalWe?=
 =?UTF-8?q?bSocket=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/org/redkale/net/http/WebSocketEngine.java | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/org/redkale/net/http/WebSocketEngine.java b/src/org/redkale/net/http/WebSocketEngine.java
index 751a0cb06..232fcf048 100644
--- a/src/org/redkale/net/http/WebSocketEngine.java
+++ b/src/org/redkale/net/http/WebSocketEngine.java
@@ -10,7 +10,7 @@ import java.io.*;
 import java.util.*;
 import java.util.concurrent.*;
 import java.util.concurrent.atomic.*;
-import java.util.function.Predicate;
+import java.util.function.*;
 import java.util.logging.*;
 import java.util.stream.*;
 import org.redkale.convert.Convert;
@@ -323,6 +323,16 @@ public class WebSocketEngine {
         return list;
     }
 
+    @Comment("获取所有连接")
+    public void forEachLocalWebSocket(Consumer<WebSocket> consumer) {
+        if (consumer == null) return;
+        if (single) {
+            websockets.values().stream().forEach(consumer);
+        } else {
+            websockets2.values().forEach(x -> x.stream().forEach(consumer));
+        }
+    }
+
     @Comment("获取当前连接总数")
     public int getLocalWebSocketSize() {
         if (single) return websockets.size();