Issue #5122 - make number of active WS Sessions a managed attribute o…

…n SessionTracker Signed-off-by: Lachlan Roberts <lachlan@webtide.com>
jetty · Aug 6, 2020 · 8a3ff77 · 8a3ff77
1 parent be429f0
commit 8a3ff77
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/...ocket-client-impl/src/main/java/org/eclipse/jetty/websocket/jsr356/JsrSessionTracker.java b/...ocket-client-impl/src/main/java/org/eclipse/jetty/websocket/jsr356/JsrSessionTracker.java
@@ -18,16 +18,19 @@
 
 package org.eclipse.jetty.websocket.jsr356;
 
+import java.io.IOException;
 import java.util.Collections;
 import java.util.HashSet;
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 import javax.websocket.Session;
 
+import org.eclipse.jetty.util.annotation.ManagedAttribute;
 import org.eclipse.jetty.util.component.AbstractLifeCycle;
+import org.eclipse.jetty.util.component.Dumpable;
 import org.eclipse.jetty.util.component.LifeCycle;
 
-public class JsrSessionTracker extends AbstractLifeCycle implements JsrSessionListener
+public class JsrSessionTracker extends AbstractLifeCycle implements JsrSessionListener, Dumpable
 {
     private final Set<JsrSession> sessions = Collections.newSetFromMap(new ConcurrentHashMap<>());
 
@@ -57,4 +60,16 @@ protected void doStop() throws Exception
         }
         super.doStop();
     }
+
+    @ManagedAttribute("Total number of active WebSocket Sessions")
+    public int getNumSessions()
+    {
+        return sessions.size();
+    }
+
+    @Override
+    public void dump(Appendable out, String indent) throws IOException
+    {
+        Dumpable.dumpObjects(out, indent, this, sessions);
+    }
 }
diff --git a/...ket/websocket-common/src/main/java/org/eclipse/jetty/websocket/common/SessionTracker.java b/...ket/websocket-common/src/main/java/org/eclipse/jetty/websocket/common/SessionTracker.java
@@ -24,6 +24,7 @@
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 
+import org.eclipse.jetty.util.annotation.ManagedAttribute;
 import org.eclipse.jetty.util.component.AbstractLifeCycle;
 import org.eclipse.jetty.util.component.Dumpable;
 import org.eclipse.jetty.util.component.LifeCycle;
@@ -61,6 +62,12 @@ protected void doStop() throws Exception
         super.doStop();
     }
 
+    @ManagedAttribute("Total number of active WebSocket Sessions")
+    public int getNumSessions()
+    {
+        return sessions.size();
+    }
+
     @Override
     public void dump(Appendable out, String indent) throws IOException
     {