root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Exposes STOMP frame stats 

Closes gh-23015
This commit is contained in:
Rossen Stoyanchev 2019-05-23 18:32:11 -04:00
parent e09c5fd9e5
commit 2294625cf0
3 changed files with 113 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
spring-messaging/src/main/java/org/springframework/messaging/simp/stomp/StompBrokerRelayMessageHandler.java
View File

@ -136,7 +136,7 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
@Nullable
private MessageHeaderInitializer headerInitializer;
private final Stats stats = new Stats();
private final DefaultStats stats = new DefaultStats();
private final Map<String, StompConnectionHandler> connectionHandlers = new ConcurrentHashMap<>();
@ -382,11 +382,21 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
/**
* Return a String describing internal state and counters.
* Effectively {@code toString()} on {@link #getStats() getStats()}.
*/
public String getStatsInfo() {
return this.stats.toString();
}
/**
* Return a structured object with internal state and counters.
* @since 5.2
*/
public Stats getStats() {
return this.stats;
}
/**
* Return the current count of TCP connection to the broker.
*/
@ -1007,7 +1017,35 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
}
private class Stats {
/**
* Contract for access to session counters.
* @since 5.2
*/
public interface Stats {
/**
* The number of connection handlers.
*/
int getTotalHandlers();
/**
* The number of CONNECT frames processed.
*/
int getTotalConnect();
/**
* The number of CONNECTED frames processed.
*/
int getTotalConnected();
/**
* The number of DISCONNECT frames processed.
*/
int getTotalDisconnect();
}
private class DefaultStats implements Stats {
private final AtomicInteger connect = new AtomicInteger();
@ -1015,6 +1053,7 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
private final AtomicInteger disconnect = new AtomicInteger();
public void incrementConnectCount() {
this.connect.incrementAndGet();
}
@ -1027,6 +1066,26 @@ public class StompBrokerRelayMessageHandler extends AbstractBrokerMessageHandler
this.disconnect.incrementAndGet();
}
@Override
public int getTotalHandlers() {
return connectionHandlers.size();
}
@Override
public int getTotalConnect() {
return this.connect.get();
}
@Override
public int getTotalConnected() {
return this.connected.get();
}
@Override
public int getTotalDisconnect() {
return this.disconnect.get();
}
public String toString() {
return (connectionHandlers.size() + " sessions, " + getTcpClientInfo() +
(isBrokerAvailable() ? " (available)" : " (not available)") +

52
spring-websocket/src/main/java/org/springframework/web/socket/messaging/StompSubProtocolHandler.java
View File

@ -114,7 +114,7 @@ public class StompSubProtocolHandler implements SubProtocolHandler, ApplicationE
@Nullable
private ApplicationEventPublisher eventPublisher;
private final Stats stats = new Stats();
private final DefaultStats stats = new DefaultStats();
/**
@ -203,11 +203,20 @@ public class StompSubProtocolHandler implements SubProtocolHandler, ApplicationE
/**
* Return a String describing internal state and counters.
* Effectively {@code toString()} on {@link #getStats() getStats()}.
*/
public String getStatsInfo() {
return this.stats.toString();
}
/**
* Return a structured object with internal state and counters.
* @since 5.2
*/
public Stats getStats() {
return this.stats;
}
/**
* Handle incoming WebSocket messages from clients.
@ -640,7 +649,30 @@ public class StompSubProtocolHandler implements SubProtocolHandler, ApplicationE
}
private static class Stats {
/**
* Contract for access to session counters.
* @since 5.2
*/
public interface Stats {
/**
* The number of CONNECT frames processed.
*/
int getTotalConnect();
/**
* The number of CONNECTED frames processed.
*/
int getTotalConnected();
/**
* The number of DISCONNECT frames processed.
*/
int getTotalDisconnect();
}
private static class DefaultStats implements Stats {
private final AtomicInteger connect = new AtomicInteger();
@ -648,6 +680,7 @@ public class StompSubProtocolHandler implements SubProtocolHandler, ApplicationE
private final AtomicInteger disconnect = new AtomicInteger();
public void incrementConnectCount() {
this.connect.incrementAndGet();
}
@ -660,6 +693,21 @@ public class StompSubProtocolHandler implements SubProtocolHandler, ApplicationE
this.disconnect.incrementAndGet();
}
@Override
public int getTotalConnect() {
return this.connect.get();
}
@Override
public int getTotalConnected() {
return this.connected.get();
}
@Override
public int getTotalDisconnect() {
return this.disconnect.get();
}
public String toString() {
return "processed CONNECT(" + this.connect.get() + ")-CONNECTED(" +
this.connected.get() + ")-DISCONNECT(" + this.disconnect.get() + ")";

3
spring-websocket/src/main/java/org/springframework/web/socket/messaging/SubProtocolWebSocketHandler.java
View File

@ -248,13 +248,14 @@ public class SubProtocolWebSocketHandler
/**
* Return a String describing internal state and counters.
* Effectively {@code toString()} on {@link #getStats() getStats()}.
*/
public String getStatsInfo() {
return this.stats.toString();
}
/**
* Return a {@link Stats} object that contains various session counters.
* Return a structured object with various session counters.
* @since 5.2
*/
public Stats getStats() {