adjust server global active connection counter to consider multiplexing

packages/server/src/Hocuspocus.ts

@@ -274,10 +274,17 @@ export class Hocuspocus {
    * Get the total number of active connections
    */
   getConnectionsCount(): number {
-    return Array.from(this.documents.values()).reduce((acc, document) => {
-      acc += document.getConnectionsCount()
-      return acc
+    const uniqueSocketIds = new Set<string>()
+    const totalDirectConnections = Array.from(this.documents.values()).reduce((acc, document) => {
+      // Accumulate unique socket IDs
+      document.getConnections().forEach(({ socketId }) => {
+        uniqueSocketIds.add(socketId)
+      })
+      // Accumulate direct connections
+      return acc + document.directConnectionsCount
     }, 0)
+    // Return the sum of unique socket IDs and direct connections
+    return uniqueSocketIds.size + totalDirectConnections
   }
 
   /**

tests/server/getConnectionsCount.ts

@@ -1,6 +1,6 @@
 import test from 'ava'
 import { retryableAssertion } from '../utils/retryableAssertion.js'
-import { newHocuspocus, newHocuspocusProvider } from '../utils/index.js'
+import { newHocuspocus, newHocuspocusProvider, newHocuspocusProviderWebsocket } from '../utils/index.js'
 
 test('returns 0 connections when there’s no one connected', async t => {
   await new Promise(async resolve => {
@@ -87,3 +87,27 @@ test('adds and removes connections properly', async t => {
     tt.is(server.getConnectionsCount(), 0)
   })
 })
+
+test('multiplexed connections counts properly', async t => {
+  const server = await newHocuspocus()
+  const socket = newHocuspocusProviderWebsocket(server)
+
+  const providers = [
+    newHocuspocusProvider(server, { name: 'mux-1' }, {}, socket),
+    newHocuspocusProvider(server, { name: 'mux-2' }, {}, socket),
+    newHocuspocusProvider(server, { name: 'mux-3' }, {}, socket),
+    newHocuspocusProvider(server),
+    newHocuspocusProvider(server),
+
+  ]
+
+  await retryableAssertion(t, tt => {
+    tt.is(server.getConnectionsCount(), 3)
+  })
+
+  providers.forEach(provider => { provider.disconnect(); provider.configuration.websocketProvider.disconnect() })
+
+  await retryableAssertion(t, tt => {
+    tt.is(server.getConnectionsCount(), 0)
+  })
+})

tests/utils/newHocuspocusProvider.ts

@@ -1,6 +1,6 @@
 import {
   HocuspocusProvider,
-  HocuspocusProviderConfiguration, HocuspocusProviderWebsocketConfiguration,
+  HocuspocusProviderConfiguration, HocuspocusProviderWebsocket, HocuspocusProviderWebsocketConfiguration,
 } from '@hocuspocus/provider'
 import { Hocuspocus } from '@hocuspocus/server'
 import { newHocuspocusProviderWebsocket } from './newHocuspocusProviderWebsocket.js'
@@ -9,9 +9,10 @@ export const newHocuspocusProvider = (
   server: Hocuspocus,
   options: Partial<HocuspocusProviderConfiguration> = {},
   websocketOptions: Partial<HocuspocusProviderWebsocketConfiguration> = {},
+  websocketProvider?: HocuspocusProviderWebsocket,
 ): HocuspocusProvider => {
   return new HocuspocusProvider({
-    websocketProvider: newHocuspocusProviderWebsocket(server, websocketOptions),
+    websocketProvider: websocketProvider ?? newHocuspocusProviderWebsocket(server, websocketOptions),
     // Just use a generic document name for all tests.
     name: 'hocuspocus-test',
     // There is no need to share data with other browser tabs in the testing environment.