fix: Improve graceful shutdown

server/collaboration/tracing.js

@@ -2,7 +2,7 @@
 import Metrics from "../logging/metrics";
 
 export default class Tracing {
-  async onCreateDocument({
+  onCreateDocument({
     documentName,
     instance,
   }: {
@@ -17,11 +17,11 @@ export default class Tracing {
     );
   }
 
-  async onAuthenticationFailed({ documentName }: { documentName: string }) {
+  onAuthenticationFailed({ documentName }: { documentName: string }) {
     Metrics.increment("collaboration.authentication_failed", { documentName });
   }
 
-  async onConnect({
+  onConnect({
     documentName,
     instance,
   }: {
@@ -35,7 +35,7 @@ export default class Tracing {
     );
   }
 
-  async onDisconnect({
+  onDisconnect({
     documentName,
     instance,
   }: {
@@ -54,7 +54,12 @@ export default class Tracing {
     );
   }
 
-  async onChange({ documentName }: { documentName: string }) {
+  onChange({ documentName }: { documentName: string }) {
     Metrics.increment("collaboration.change", { documentName });
   }
+
+  onDestroy() {
+    Metrics.gaugePerInstance("collaboration.connections_count", 0);
+    Metrics.gaugePerInstance("collaboration.documents_count", 0);
+  }
 }

server/index.js

@@ -108,6 +108,8 @@ async function start(id: string, disconnect: () => void) {
 
   function shutdown() {
     Logger.info("lifecycle", "Stopping server");
+
+    server.emit("shutdown");
     server.stop(disconnect);
   }
 }

server/services/collaboration.js

@@ -36,4 +36,8 @@ export default function init(app: Koa, server: http.Server) {
   app.use(websocket());
   app.use(router.routes());
   app.use(router.allowedMethods());
+
+  server.on("shutdown", () => {
+    hocuspocus.destroy();
+  });
 }

server/services/websockets.js

@@ -23,6 +23,10 @@ export default function init(app: Koa, server: http.Server) {
     cookie: false,
   });
 
+  server.on("shutdown", () => {
+    Metrics.gaugePerInstance("websockets.count", 0);
+  });
+
   io.adapter(
     socketRedisAdapter({
       pubClient: client,