WebSocketEngine增加forEachLocalWebSocket方法

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();