diff --git a/src/com/wentch/redkale/net/http/WebSocket.java b/src/com/wentch/redkale/net/http/WebSocket.java index 5fddad621..c6e1499aa 100644 --- a/src/com/wentch/redkale/net/http/WebSocket.java +++ b/src/com/wentch/redkale/net/http/WebSocket.java @@ -114,6 +114,10 @@ public abstract class WebSocket { protected final WebSocketGroup getWebSocketGroup(long groupid) { return engine.getWebSocketGroup(groupid); } + + protected final Collection getWebSocketGroups() { + return engine.getWebSocketGroups(); + } //------------------------------------------------------------------- /** diff --git a/src/com/wentch/redkale/net/http/WebSocketEngine.java b/src/com/wentch/redkale/net/http/WebSocketEngine.java index 3d8ebea35..b86de7009 100644 --- a/src/com/wentch/redkale/net/http/WebSocketEngine.java +++ b/src/com/wentch/redkale/net/http/WebSocketEngine.java @@ -49,6 +49,10 @@ public final class WebSocketEngine { if (group.isEmpty()) containers.remove(socket.groupid); } + Collection getWebSocketGroups() { + return containers.values(); + } + public WebSocketGroup getWebSocketGroup(Serializable groupid) { return containers.get(groupid); }