fix: Improve graceful shutdown

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,