From 1fb1d17f52ad45f98e8909ccfa9b07b638654ac4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=9C=B0=E5=B9=B3=E7=BA=BF?= <22250530@qq.com>
Date: Sat, 27 Jun 2015 10:27:01 +0800
Subject: [PATCH]

---
 src/com/wentch/redkale/net/http/WebSocket.java       | 4 ++++
 src/com/wentch/redkale/net/http/WebSocketEngine.java | 4 ++++
 2 files changed, 8 insertions(+)

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<WebSocketGroup> 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<WebSocketGroup> getWebSocketGroups() {
+        return containers.values();
+    }
+
     public WebSocketGroup getWebSocketGroup(Serializable groupid) {
         return containers.get(groupid);
     }