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