MirrorMirror
/
Signal-Server
mirror of https://github.com/signalapp/Signal-Server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Keep a running count of the number of open websockets.

This commit is contained in:
Jon Chambers 2020-08-06 12:30:12 -04:00 committed by Jon Chambers
parent 3286c5e174
commit 38293efe75
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
service/src/main/java/org/whispersystems/textsecuregcm/websocket/AuthenticatedConnectListener.java
View File

@ -1,5 +1,6 @@
package org.whispersystems.textsecuregcm.websocket; package org.whispersystems.textsecuregcm.websocket;
import com.codahale.metrics.Counter;
import com.codahale.metrics.MetricRegistry; import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.SharedMetricRegistries; import com.codahale.metrics.SharedMetricRegistries;
import com.codahale.metrics.Timer; import com.codahale.metrics.Timer;
@ -29,6 +30,7 @@ public class AuthenticatedConnectListener implements WebSocketConnectListener {
private static final MetricRegistry metricRegistry = SharedMetricRegistries.getOrCreate(Constants.METRICS_NAME); private static final MetricRegistry metricRegistry = SharedMetricRegistries.getOrCreate(Constants.METRICS_NAME);
private static final Timer durationTimer = metricRegistry.timer(name(WebSocketConnection.class, "connected_duration" )); private static final Timer durationTimer = metricRegistry.timer(name(WebSocketConnection.class, "connected_duration" ));
private static final Timer unauthenticatedDurationTimer = metricRegistry.timer(name(WebSocketConnection.class, "unauthenticated_connection_duration")); private static final Timer unauthenticatedDurationTimer = metricRegistry.timer(name(WebSocketConnection.class, "unauthenticated_connection_duration"));
private static final Counter openWebsocketCounter = metricRegistry.counter(name(WebSocketConnection.class, "open_websockets"));
private final PushSender pushSender; private final PushSender pushSender;
private final ReceiptSender receiptSender; private final ReceiptSender receiptSender;
@ -64,6 +66,7 @@ public class AuthenticatedConnectListener implements WebSocketConnectListener {
.setContent(ByteString.copyFrom(connectionId.getBytes())) .setContent(ByteString.copyFrom(connectionId.getBytes()))
.build(); .build();
openWebsocketCounter.inc();
RedisOperation.unchecked(() -> apnFallbackManager.cancel(account, device)); RedisOperation.unchecked(() -> apnFallbackManager.cancel(account, device));
pubSubManager.publish(address, connectMessage); pubSubManager.publish(address, connectMessage);
pubSubManager.subscribe(address, connection); pubSubManager.subscribe(address, connection);
@ -71,6 +74,7 @@ public class AuthenticatedConnectListener implements WebSocketConnectListener {
context.addListener(new WebSocketSessionContext.WebSocketEventListener() { context.addListener(new WebSocketSessionContext.WebSocketEventListener() {
@Override @Override
public void onWebSocketClose(WebSocketSessionContext context, int statusCode, String reason) { public void onWebSocketClose(WebSocketSessionContext context, int statusCode, String reason) {
openWebsocketCounter.dec();
pubSubManager.unsubscribe(address, connection); pubSubManager.unsubscribe(address, connection);
timer.stop(); timer.stop();
} }