@webiny/api-websockets-standalone 0.0.0-unstable.7c2def32a7 → 0.0.0-unstable.9609e4e0c9

package/adapter/NodeWsAdapter.d.ts

@@ -5,10 +5,15 @@ import { WebSocket } from "ws";
 import { WebsocketsServerAdapter } from "./abstractions.js";
 declare class NodeWsAdapterImpl implements WebsocketsServerAdapter.Interface<WebSocket> {
     private wss;
+    private pingTimer;
+    private readonly answered;
+    private pingInterval;
+    static withPingInterval(pingInterval: number): NodeWsAdapterImpl;
     private getWss;
     start(_server: HttpServer): void;
     handleUpgrade(request: IncomingMessage, socket: Duplex, head: Buffer): void;
     stop(): Promise<void>;
+    private pingAll;
     onConnection(cb: (socket: WebSocket, request: import("node:http").IncomingMessage) => void): void;
     onMessage(socket: WebSocket, cb: (data: Buffer) => void): void;
     onClose(socket: WebSocket, cb: (code: number, reason: Buffer) => void): void;

package/adapter/NodeWsAdapter.js

@@ -3,6 +3,7 @@ import { WebsocketsServerAdapter } from "./abstractions.js";
 const MAX_PAYLOAD_BYTES = 65536;
 const SERVICE_RESTART = 1012;
 const CLOSE_WAIT_MS = 1000;
+const PING_INTERVAL_MS = 30000;
 const closeForRestart = (client)=>new Promise((resolve)=>{
         if (client.readyState === WebSocket.CLOSED) return void resolve();
         const timer = setTimeout(()=>{
@@ -15,15 +16,31 @@ const closeForRestart = (client)=>new Promise((resolve)=>{
         client.close(SERVICE_RESTART, "Server restarting");
     });
 class NodeWsAdapterImpl {
+    static withPingInterval(pingInterval) {
+        const adapter = new NodeWsAdapterImpl();
+        adapter.pingInterval = pingInterval;
+        return adapter;
+    }
     getWss() {
         if (!this.wss) throw new Error("WebSocketServer has not been started. Call start() first.");
         return this.wss;
     }
     start(_server) {
-        this.wss = new WebSocketServer({
+        if (this.wss) return;
+        const wss = new WebSocketServer({
             noServer: true,
             maxPayload: MAX_PAYLOAD_BYTES
         });
+        wss.on("connection", (socket)=>{
+            this.answered.add(socket);
+            socket.on("pong", ()=>{
+                this.answered.add(socket);
+            });
+        });
+        this.wss = wss;
+        this.pingTimer = setInterval(()=>{
+            this.pingAll(wss);
+        }, this.pingInterval);
     }
     handleUpgrade(request, socket, head) {
         this.getWss().handleUpgrade(request, socket, head, (ws)=>{
@@ -34,6 +51,7 @@ class NodeWsAdapterImpl {
         const wss = this.wss;
         if (!wss) return;
         this.wss = void 0;
+        clearInterval(this.pingTimer);
         const clients = [
             ...wss.clients
         ];
@@ -46,6 +64,16 @@ class NodeWsAdapterImpl {
             });
         });
     }
+    pingAll(wss) {
+        for (const socket of wss.clients){
+            if (!this.answered.has(socket)) {
+                socket.terminate();
+                continue;
+            }
+            this.answered.delete(socket);
+            if (socket.readyState === WebSocket.OPEN) socket.ping();
+        }
+    }
     onConnection(cb) {
         this.getWss().on("connection", cb);
     }
@@ -69,6 +97,10 @@ class NodeWsAdapterImpl {
     close(socket, code, reason) {
         socket.close(code, reason);
     }
+    constructor(){
+        this.answered = new WeakSet();
+        this.pingInterval = PING_INTERVAL_MS;
+    }
 }
 const NodeWsAdapter = WebsocketsServerAdapter.createImplementation({
     implementation: NodeWsAdapterImpl,

package/adapter/NodeWsAdapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"adapter/NodeWsAdapter.js","sources":["../../src/adapter/NodeWsAdapter.ts"],"sourcesContent":["import type { Server as HttpServer } from \"node:http\";\nimport type { IncomingMessage } from \"node:http\";\nimport type { Duplex } from \"node:stream\";\nimport { WebSocketServer } from \"ws\";\nimport { WebSocket } from \"ws\";\nimport { WebsocketsServerAdapter } from \"./abstractions.js\";\n\n/*\n * Clients only send small JSON envelopes (a ping carrying a token). The `ws` default is 100 MiB,\n * which any client, authenticated or not, could make the server buffer per message.\n */\nconst MAX_PAYLOAD_BYTES = 64 * 1024;\n\n/*\n * Sent to every client when the server stops. 1012 means \"service restart\": the client reconnects,\n * and behind a load balancer it reaches a process that is still up.\n */\nconst SERVICE_RESTART = 1012;\n\n// How long a client gets to answer the close before its socket is cut.\nconst CLOSE_WAIT_MS = 1000;\n\n/*\n * Closes the socket with `SERVICE_RESTART` and resolves once it is closed. A client that has not\n * answered within `CLOSE_WAIT_MS` (a frozen tab, a dead network) is cut.\n */\nconst closeForRestart = (client: WebSocket): Promise<void> => {\n    return new Promise(resolve => {\n        if (client.readyState === WebSocket.CLOSED) {\n            resolve();\n            return;\n        }\n        const timer = setTimeout(() => {\n            client.terminate();\n        }, CLOSE_WAIT_MS);\n        client.once(\"close\", () => {\n            clearTimeout(timer);\n            resolve();\n        });\n        client.close(SERVICE_RESTART, \"Server restarting\");\n    });\n};\n\nclass NodeWsAdapterImpl implements WebsocketsServerAdapter.Interface<WebSocket> {\n    private wss: WebSocketServer | undefined;\n\n    private getWss(): WebSocketServer {\n        if (!this.wss) {\n            throw new Error(\"WebSocketServer has not been started. Call start() first.\");\n        }\n        return this.wss;\n    }\n\n    /* The server param is kept for interface compliance but not used —\n       the HTTP upgrade event is handled externally via handleUpgrade(). */\n    public start(_server: HttpServer): void {\n        this.wss = new WebSocketServer({ noServer: true, maxPayload: MAX_PAYLOAD_BYTES });\n    }\n\n    public handleUpgrade(request: IncomingMessage, socket: Duplex, head: Buffer): void {\n        this.getWss().handleUpgrade(request, socket, head, ws => {\n            this.getWss().emit(\"connection\", ws, request);\n        });\n    }\n\n    // Safe to call again: a stopped adapter has nothing left to close.\n    public async stop(): Promise<void> {\n        const wss = this.wss;\n        if (!wss) {\n            return;\n        }\n        this.wss = undefined;\n\n        const clients = [...wss.clients];\n        const closing = clients.map(closeForRestart);\n        await Promise.all(closing);\n\n        await new Promise<void>((resolve, reject) => {\n            wss.close(err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    public onConnection(\n        cb: (socket: WebSocket, request: import(\"node:http\").IncomingMessage) => void\n    ): void {\n        this.getWss().on(\"connection\", cb);\n    }\n\n    public onMessage(socket: WebSocket, cb: (data: Buffer) => void): void {\n        socket.on(\"message\", cb);\n    }\n\n    public onClose(socket: WebSocket, cb: (code: number, reason: Buffer) => void): void {\n        socket.on(\"close\", cb);\n    }\n\n    public onError(socket: WebSocket, cb: (error: Error) => void): void {\n        socket.on(\"error\", cb);\n    }\n\n    public send(socket: WebSocket, data: string): Promise<void> {\n        return new Promise((resolve, reject) => {\n            socket.send(data, err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    public close(socket: WebSocket, code?: number, reason?: string): void {\n        socket.close(code, reason);\n    }\n}\n\nexport { NodeWsAdapterImpl };\n\nexport const NodeWsAdapter = WebsocketsServerAdapter.createImplementation({\n    implementation: NodeWsAdapterImpl,\n    dependencies: []\n});\n"],"names":["MAX_PAYLOAD_BYTES","SERVICE_RESTART","CLOSE_WAIT_MS","closeForRestart","client","Promise","resolve","WebSocket","timer","setTimeout","clearTimeout","NodeWsAdapterImpl","Error","_server","WebSocketServer","request","socket","head","ws","wss","undefined","clients","closing","reject","err","cb","data","code","reason","NodeWsAdapter","WebsocketsServerAdapter"],"mappings":";;AAWA,MAAMA,oBAAoB;AAM1B,MAAMC,kBAAkB;AAGxB,MAAMC,gBAAgB;AAMtB,MAAMC,kBAAkB,CAACC,SACd,IAAIC,QAAQC,CAAAA;QACf,IAAIF,OAAO,UAAU,KAAKG,UAAU,MAAM,EAAE,YACxCD;QAGJ,MAAME,QAAQC,WAAW;YACrBL,OAAO,SAAS;QACpB,GAAGF;QACHE,OAAO,IAAI,CAAC,SAAS;YACjBM,aAAaF;YACbF;QACJ;QACAF,OAAO,KAAK,CAACH,iBAAiB;IAClC;AAGJ,MAAMU;IAGM,SAA0B;QAC9B,IAAI,CAAC,IAAI,CAAC,GAAG,EACT,MAAM,IAAIC,MAAM;QAEpB,OAAO,IAAI,CAAC,GAAG;IACnB;IAIO,MAAMC,OAAmB,EAAQ;QACpC,IAAI,CAAC,GAAG,GAAG,IAAIC,gBAAgB;YAAE,UAAU;YAAM,YAAYd;QAAkB;IACnF;IAEO,cAAce,OAAwB,EAAEC,MAAc,EAAEC,IAAY,EAAQ;QAC/E,IAAI,CAAC,MAAM,GAAG,aAAa,CAACF,SAASC,QAAQC,MAAMC,CAAAA;YAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,cAAcA,IAAIH;QACzC;IACJ;IAGA,MAAa,OAAsB;QAC/B,MAAMI,MAAM,IAAI,CAAC,GAAG;QACpB,IAAI,CAACA,KACD;QAEJ,IAAI,CAAC,GAAG,GAAGC;QAEX,MAAMC,UAAU;eAAIF,IAAI,OAAO;SAAC;QAChC,MAAMG,UAAUD,QAAQ,GAAG,CAAClB;QAC5B,MAAME,QAAQ,GAAG,CAACiB;QAElB,MAAM,IAAIjB,QAAc,CAACC,SAASiB;YAC9BJ,IAAI,KAAK,CAACK,CAAAA;gBACN,IAAIA,KAAK,YACLD,OAAOC;gBAGXlB;YACJ;QACJ;IACJ;IAEO,aACHmB,EAA6E,EACzE;QACJ,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,cAAcA;IACnC;IAEO,UAAUT,MAAiB,EAAES,EAA0B,EAAQ;QAClET,OAAO,EAAE,CAAC,WAAWS;IACzB;IAEO,QAAQT,MAAiB,EAAES,EAA0C,EAAQ;QAChFT,OAAO,EAAE,CAAC,SAASS;IACvB;IAEO,QAAQT,MAAiB,EAAES,EAA0B,EAAQ;QAChET,OAAO,EAAE,CAAC,SAASS;IACvB;IAEO,KAAKT,MAAiB,EAAEU,IAAY,EAAiB;QACxD,OAAO,IAAIrB,QAAQ,CAACC,SAASiB;YACzBP,OAAO,IAAI,CAACU,MAAMF,CAAAA;gBACd,IAAIA,KAAK,YACLD,OAAOC;gBAGXlB;YACJ;QACJ;IACJ;IAEO,MAAMU,MAAiB,EAAEW,IAAa,EAAEC,MAAe,EAAQ;QAClEZ,OAAO,KAAK,CAACW,MAAMC;IACvB;AACJ;AAIO,MAAMC,gBAAgBC,wBAAwB,oBAAoB,CAAC;IACtE,gBAAgBnB;IAChB,cAAc,EAAE;AACpB"}
+{"version":3,"file":"adapter/NodeWsAdapter.js","sources":["../../src/adapter/NodeWsAdapter.ts"],"sourcesContent":["import type { Server as HttpServer } from \"node:http\";\nimport type { IncomingMessage } from \"node:http\";\nimport type { Duplex } from \"node:stream\";\nimport { WebSocketServer } from \"ws\";\nimport { WebSocket } from \"ws\";\nimport { WebsocketsServerAdapter } from \"./abstractions.js\";\n\n/*\n * Clients only send small JSON envelopes (a ping carrying a token). The `ws` default is 100 MiB,\n * which any client, authenticated or not, could make the server buffer per message.\n */\nconst MAX_PAYLOAD_BYTES = 64 * 1024;\n\n/*\n * Sent to every client when the server stops. 1012 means \"service restart\": the client reconnects,\n * and behind a load balancer it reaches a process that is still up.\n */\nconst SERVICE_RESTART = 1012;\n\n// How long a client gets to answer the close before its socket is cut.\nconst CLOSE_WAIT_MS = 1000;\n\n/*\n * How often every open socket gets a WebSocket ping frame. Proxies and load balancers close a\n * connection that carries nothing for a while (60 seconds by default on nginx and AWS ALB), and the\n * admin's own ping comes only every 5 minutes. Browsers answer ping frames by themselves; no client\n * code is involved, and nothing is written to the registry.\n */\nconst PING_INTERVAL_MS = 30_000;\n\n/*\n * Closes the socket with `SERVICE_RESTART` and resolves once it is closed. A client that has not\n * answered within `CLOSE_WAIT_MS` (a frozen tab, a dead network) is cut.\n */\nconst closeForRestart = (client: WebSocket): Promise<void> => {\n    return new Promise(resolve => {\n        if (client.readyState === WebSocket.CLOSED) {\n            resolve();\n            return;\n        }\n        const timer = setTimeout(() => {\n            client.terminate();\n        }, CLOSE_WAIT_MS);\n        client.once(\"close\", () => {\n            clearTimeout(timer);\n            resolve();\n        });\n        client.close(SERVICE_RESTART, \"Server restarting\");\n    });\n};\n\nclass NodeWsAdapterImpl implements WebsocketsServerAdapter.Interface<WebSocket> {\n    private wss: WebSocketServer | undefined;\n    private pingTimer: ReturnType<typeof setInterval> | undefined;\n    // Sockets that answered the last ping, or connected since it was sent.\n    private readonly answered = new WeakSet<WebSocket>();\n\n    private pingInterval = PING_INTERVAL_MS;\n\n    /*\n     * An adapter that pings more often, for tests. The constructor takes nothing: DI builds this class\n     * with no dependencies.\n     */\n    public static withPingInterval(pingInterval: number): NodeWsAdapterImpl {\n        const adapter = new NodeWsAdapterImpl();\n        adapter.pingInterval = pingInterval;\n        return adapter;\n    }\n\n    private getWss(): WebSocketServer {\n        if (!this.wss) {\n            throw new Error(\"WebSocketServer has not been started. Call start() first.\");\n        }\n        return this.wss;\n    }\n\n    /* The server param is kept for interface compliance but not used —\n       the HTTP upgrade event is handled externally via handleUpgrade(). */\n    public start(_server: HttpServer): void {\n        // Already started: a second server and timer would be left running, unreachable by stop().\n        if (this.wss) {\n            return;\n        }\n        const wss = new WebSocketServer({ noServer: true, maxPayload: MAX_PAYLOAD_BYTES });\n        wss.on(\"connection\", socket => {\n            this.answered.add(socket);\n            socket.on(\"pong\", () => {\n                this.answered.add(socket);\n            });\n        });\n        this.wss = wss;\n        this.pingTimer = setInterval(() => {\n            this.pingAll(wss);\n        }, this.pingInterval);\n    }\n\n    public handleUpgrade(request: IncomingMessage, socket: Duplex, head: Buffer): void {\n        this.getWss().handleUpgrade(request, socket, head, ws => {\n            this.getWss().emit(\"connection\", ws, request);\n        });\n    }\n\n    // Safe to call again: a stopped adapter has nothing left to close.\n    public async stop(): Promise<void> {\n        const wss = this.wss;\n        if (!wss) {\n            return;\n        }\n        this.wss = undefined;\n        clearInterval(this.pingTimer);\n\n        const clients = [...wss.clients];\n        const closing = clients.map(closeForRestart);\n        await Promise.all(closing);\n\n        await new Promise<void>((resolve, reject) => {\n            wss.close(err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    /*\n     * Pings every open socket. One that has not answered the previous ping within a whole interval is\n     * gone without having closed (a laptop asleep, a dropped network), so it is cut; its close\n     * handler then removes the connection.\n     */\n    private pingAll(wss: WebSocketServer): void {\n        for (const socket of wss.clients) {\n            if (!this.answered.has(socket)) {\n                socket.terminate();\n                continue;\n            }\n            this.answered.delete(socket);\n            if (socket.readyState === WebSocket.OPEN) {\n                socket.ping();\n            }\n        }\n    }\n\n    public onConnection(\n        cb: (socket: WebSocket, request: import(\"node:http\").IncomingMessage) => void\n    ): void {\n        this.getWss().on(\"connection\", cb);\n    }\n\n    public onMessage(socket: WebSocket, cb: (data: Buffer) => void): void {\n        socket.on(\"message\", cb);\n    }\n\n    public onClose(socket: WebSocket, cb: (code: number, reason: Buffer) => void): void {\n        socket.on(\"close\", cb);\n    }\n\n    public onError(socket: WebSocket, cb: (error: Error) => void): void {\n        socket.on(\"error\", cb);\n    }\n\n    public send(socket: WebSocket, data: string): Promise<void> {\n        return new Promise((resolve, reject) => {\n            socket.send(data, err => {\n                if (err) {\n                    reject(err);\n                    return;\n                }\n                resolve();\n            });\n        });\n    }\n\n    public close(socket: WebSocket, code?: number, reason?: string): void {\n        socket.close(code, reason);\n    }\n}\n\nexport { NodeWsAdapterImpl };\n\nexport const NodeWsAdapter = WebsocketsServerAdapter.createImplementation({\n    implementation: NodeWsAdapterImpl,\n    dependencies: []\n});\n"],"names":["MAX_PAYLOAD_BYTES","SERVICE_RESTART","CLOSE_WAIT_MS","PING_INTERVAL_MS","closeForRestart","client","Promise","resolve","WebSocket","timer","setTimeout","clearTimeout","NodeWsAdapterImpl","pingInterval","adapter","Error","_server","wss","WebSocketServer","socket","setInterval","request","head","ws","undefined","clearInterval","clients","closing","reject","err","cb","data","code","reason","WeakSet","NodeWsAdapter","WebsocketsServerAdapter"],"mappings":";;AAWA,MAAMA,oBAAoB;AAM1B,MAAMC,kBAAkB;AAGxB,MAAMC,gBAAgB;AAQtB,MAAMC,mBAAmB;AAMzB,MAAMC,kBAAkB,CAACC,SACd,IAAIC,QAAQC,CAAAA;QACf,IAAIF,OAAO,UAAU,KAAKG,UAAU,MAAM,EAAE,YACxCD;QAGJ,MAAME,QAAQC,WAAW;YACrBL,OAAO,SAAS;QACpB,GAAGH;QACHG,OAAO,IAAI,CAAC,SAAS;YACjBM,aAAaF;YACbF;QACJ;QACAF,OAAO,KAAK,CAACJ,iBAAiB;IAClC;AAGJ,MAAMW;IAYF,OAAc,iBAAiBC,YAAoB,EAAqB;QACpE,MAAMC,UAAU,IAAIF;QACpBE,QAAQ,YAAY,GAAGD;QACvB,OAAOC;IACX;IAEQ,SAA0B;QAC9B,IAAI,CAAC,IAAI,CAAC,GAAG,EACT,MAAM,IAAIC,MAAM;QAEpB,OAAO,IAAI,CAAC,GAAG;IACnB;IAIO,MAAMC,OAAmB,EAAQ;QAEpC,IAAI,IAAI,CAAC,GAAG,EACR;QAEJ,MAAMC,MAAM,IAAIC,gBAAgB;YAAE,UAAU;YAAM,YAAYlB;QAAkB;QAChFiB,IAAI,EAAE,CAAC,cAAcE,CAAAA;YACjB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA;YAClBA,OAAO,EAAE,CAAC,QAAQ;gBACd,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA;YACtB;QACJ;QACA,IAAI,CAAC,GAAG,GAAGF;QACX,IAAI,CAAC,SAAS,GAAGG,YAAY;YACzB,IAAI,CAAC,OAAO,CAACH;QACjB,GAAG,IAAI,CAAC,YAAY;IACxB;IAEO,cAAcI,OAAwB,EAAEF,MAAc,EAAEG,IAAY,EAAQ;QAC/E,IAAI,CAAC,MAAM,GAAG,aAAa,CAACD,SAASF,QAAQG,MAAMC,CAAAA;YAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,cAAcA,IAAIF;QACzC;IACJ;IAGA,MAAa,OAAsB;QAC/B,MAAMJ,MAAM,IAAI,CAAC,GAAG;QACpB,IAAI,CAACA,KACD;QAEJ,IAAI,CAAC,GAAG,GAAGO;QACXC,cAAc,IAAI,CAAC,SAAS;QAE5B,MAAMC,UAAU;eAAIT,IAAI,OAAO;SAAC;QAChC,MAAMU,UAAUD,QAAQ,GAAG,CAACtB;QAC5B,MAAME,QAAQ,GAAG,CAACqB;QAElB,MAAM,IAAIrB,QAAc,CAACC,SAASqB;YAC9BX,IAAI,KAAK,CAACY,CAAAA;gBACN,IAAIA,KAAK,YACLD,OAAOC;gBAGXtB;YACJ;QACJ;IACJ;IAOQ,QAAQU,GAAoB,EAAQ;QACxC,KAAK,MAAME,UAAUF,IAAI,OAAO,CAAE;YAC9B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACE,SAAS;gBAC5BA,OAAO,SAAS;gBAChB;YACJ;YACA,IAAI,CAAC,QAAQ,CAAC,MAAM,CAACA;YACrB,IAAIA,OAAO,UAAU,KAAKX,UAAU,IAAI,EACpCW,OAAO,IAAI;QAEnB;IACJ;IAEO,aACHW,EAA6E,EACzE;QACJ,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,cAAcA;IACnC;IAEO,UAAUX,MAAiB,EAAEW,EAA0B,EAAQ;QAClEX,OAAO,EAAE,CAAC,WAAWW;IACzB;IAEO,QAAQX,MAAiB,EAAEW,EAA0C,EAAQ;QAChFX,OAAO,EAAE,CAAC,SAASW;IACvB;IAEO,QAAQX,MAAiB,EAAEW,EAA0B,EAAQ;QAChEX,OAAO,EAAE,CAAC,SAASW;IACvB;IAEO,KAAKX,MAAiB,EAAEY,IAAY,EAAiB;QACxD,OAAO,IAAIzB,QAAQ,CAACC,SAASqB;YACzBT,OAAO,IAAI,CAACY,MAAMF,CAAAA;gBACd,IAAIA,KAAK,YACLD,OAAOC;gBAGXtB;YACJ;QACJ;IACJ;IAEO,MAAMY,MAAiB,EAAEa,IAAa,EAAEC,MAAe,EAAQ;QAClEd,OAAO,KAAK,CAACa,MAAMC;IACvB;;aAzHiB,QAAQ,GAAG,IAAIC;aAExB,YAAY,GAAG/B;;AAwH3B;AAIO,MAAMgC,gBAAgBC,wBAAwB,oBAAoB,CAAC;IACtE,gBAAgBxB;IAChB,cAAc,EAAE;AACpB"}

package/bus/InProcessMessageBus.d.ts

@@ -0,0 +1,11 @@
+import { WebsocketsMessageBus } from "./abstractions.js";
+declare class InProcessMessageBusImpl implements WebsocketsMessageBus.Interface {
+    private readonly handlers;
+    publish(owner: string, message: WebsocketsMessageBus.Message): Promise<void>;
+    subscribe(owner: string, handler: WebsocketsMessageBus.Handler): Promise<void>;
+    close(): Promise<void>;
+}
+export { InProcessMessageBusImpl };
+export declare const InProcessMessageBus: typeof InProcessMessageBusImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("./abstractions.js").IWebsocketsMessageBus>;
+};

package/bus/InProcessMessageBus.js

@@ -0,0 +1,24 @@
+import { WebsocketsMessageBus } from "./abstractions.js";
+class InProcessMessageBusImpl {
+    async publish(owner, message) {
+        const handler = this.handlers.get(owner);
+        if (!handler) return;
+        await handler(message);
+    }
+    async subscribe(owner, handler) {
+        this.handlers.set(owner, handler);
+    }
+    async close() {
+        this.handlers.clear();
+    }
+    constructor(){
+        this.handlers = new Map();
+    }
+}
+const InProcessMessageBus = WebsocketsMessageBus.createImplementation({
+    implementation: InProcessMessageBusImpl,
+    dependencies: []
+});
+export { InProcessMessageBus, InProcessMessageBusImpl };
+
+//# sourceMappingURL=InProcessMessageBus.js.map

package/bus/InProcessMessageBus.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bus/InProcessMessageBus.js","sources":["../../src/bus/InProcessMessageBus.ts"],"sourcesContent":["import { WebsocketsMessageBus } from \"./abstractions.js\";\n\n/*\n * The bus for a single process, and the default. Every socket is in this process, so the only\n * listener it can reach is its own. Rows another process wrote, such as this process's predecessor\n * before a restart, name an owner nobody listens for here; those messages are dropped and the rows\n * are left to the heartbeat's stale cleanup.\n */\nclass InProcessMessageBusImpl implements WebsocketsMessageBus.Interface {\n    private readonly handlers = new Map<string, WebsocketsMessageBus.Handler>();\n\n    public async publish(owner: string, message: WebsocketsMessageBus.Message): Promise<void> {\n        const handler = this.handlers.get(owner);\n        if (!handler) {\n            return;\n        }\n        await handler(message);\n    }\n\n    public async subscribe(owner: string, handler: WebsocketsMessageBus.Handler): Promise<void> {\n        this.handlers.set(owner, handler);\n    }\n\n    public async close(): Promise<void> {\n        this.handlers.clear();\n    }\n}\n\nexport { InProcessMessageBusImpl };\n\nexport const InProcessMessageBus = WebsocketsMessageBus.createImplementation({\n    implementation: InProcessMessageBusImpl,\n    dependencies: []\n});\n"],"names":["InProcessMessageBusImpl","owner","message","handler","Map","InProcessMessageBus","WebsocketsMessageBus"],"mappings":";AAQA,MAAMA;IAGF,MAAa,QAAQC,KAAa,EAAEC,OAAqC,EAAiB;QACtF,MAAMC,UAAU,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACF;QAClC,IAAI,CAACE,SACD;QAEJ,MAAMA,QAAQD;IAClB;IAEA,MAAa,UAAUD,KAAa,EAAEE,OAAqC,EAAiB;QACxF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACF,OAAOE;IAC7B;IAEA,MAAa,QAAuB;QAChC,IAAI,CAAC,QAAQ,CAAC,KAAK;IACvB;;aAhBiB,QAAQ,GAAG,IAAIC;;AAiBpC;AAIO,MAAMC,sBAAsBC,qBAAqB,oBAAoB,CAAC;IACzE,gBAAgBN;IAChB,cAAc,EAAE;AACpB"}

package/bus/abstractions.d.ts

@@ -0,0 +1,30 @@
+import type { GenericRecord } from "@webiny/api/types.js";
+export interface IWebsocketsBusSendMessage {
+    action: "send";
+    connectionIds: string[];
+    data: GenericRecord;
+}
+export interface IWebsocketsBusDisconnectMessage {
+    action: "disconnect";
+    connectionIds: string[];
+}
+export type IWebsocketsBusMessage = IWebsocketsBusSendMessage | IWebsocketsBusDisconnectMessage;
+export type IWebsocketsBusHandler = (message: IWebsocketsBusMessage) => Promise<void>;
+/**
+ * Carries a send or a disconnect to the process that holds the sockets. Each process listens on its
+ * own channel, named after its `ProcessOwner` id, and other processes publish to it.
+ *
+ * Delivery is at most once and nothing is queued: a message for a process that is not listening is
+ * dropped, as a push to a reconnecting client is on AWS.
+ */
+export interface IWebsocketsMessageBus {
+    publish(owner: string, message: IWebsocketsBusMessage): Promise<void>;
+    subscribe(owner: string, handler: IWebsocketsBusHandler): Promise<void>;
+    close(): Promise<void>;
+}
+export declare const WebsocketsMessageBus: import("@webiny/di").Abstraction<IWebsocketsMessageBus>;
+export declare namespace WebsocketsMessageBus {
+    type Interface = IWebsocketsMessageBus;
+    type Message = IWebsocketsBusMessage;
+    type Handler = IWebsocketsBusHandler;
+}

package/bus/abstractions.js

@@ -0,0 +1,5 @@
+import { createAbstraction } from "@webiny/feature/api";
+const WebsocketsMessageBus = createAbstraction("Websockets/MessageBus");
+export { WebsocketsMessageBus };
+
+//# sourceMappingURL=abstractions.js.map

package/bus/abstractions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bus/abstractions.js","sources":["../../src/bus/abstractions.ts"],"sourcesContent":["import type { GenericRecord } from \"@webiny/api/types.js\";\nimport { createAbstraction } from \"@webiny/feature/api\";\n\nexport interface IWebsocketsBusSendMessage {\n    action: \"send\";\n    connectionIds: string[];\n    data: GenericRecord;\n}\n\nexport interface IWebsocketsBusDisconnectMessage {\n    action: \"disconnect\";\n    connectionIds: string[];\n}\n\nexport type IWebsocketsBusMessage = IWebsocketsBusSendMessage | IWebsocketsBusDisconnectMessage;\n\nexport type IWebsocketsBusHandler = (message: IWebsocketsBusMessage) => Promise<void>;\n\n/**\n * Carries a send or a disconnect to the process that holds the sockets. Each process listens on its\n * own channel, named after its `ProcessOwner` id, and other processes publish to it.\n *\n * Delivery is at most once and nothing is queued: a message for a process that is not listening is\n * dropped, as a push to a reconnecting client is on AWS.\n */\nexport interface IWebsocketsMessageBus {\n    publish(owner: string, message: IWebsocketsBusMessage): Promise<void>;\n    subscribe(owner: string, handler: IWebsocketsBusHandler): Promise<void>;\n    close(): Promise<void>;\n}\n\nexport const WebsocketsMessageBus =\n    createAbstraction<IWebsocketsMessageBus>(\"Websockets/MessageBus\");\n\nexport namespace WebsocketsMessageBus {\n    export type Interface = IWebsocketsMessageBus;\n    export type Message = IWebsocketsBusMessage;\n    export type Handler = IWebsocketsBusHandler;\n}\n"],"names":["WebsocketsMessageBus","createAbstraction"],"mappings":";AA+BO,MAAMA,uBACTC,kBAAyC"}

package/bus/createBusMessageHandler.d.ts

@@ -0,0 +1,10 @@
+import type { Logger } from "@webiny/api-core/features/logger/abstractions.js";
+import type { WebsocketsConnectionManager } from "../connectionManager/abstractions.js";
+import type { WebsocketsServerAdapter } from "../adapter/abstractions.js";
+import type { WebsocketsMessageBus } from "./abstractions.js";
+export interface ICreateBusMessageHandlerParams {
+    connectionManager: WebsocketsConnectionManager.Interface<unknown>;
+    adapter: WebsocketsServerAdapter.Interface<unknown>;
+    logger: Logger.Interface;
+}
+export declare const createBusMessageHandler: (params: ICreateBusMessageHandlerParams) => WebsocketsMessageBus.Handler;

package/bus/createBusMessageHandler.js

@@ -0,0 +1,20 @@
+import { sendToLocalSockets } from "../delivery/sendToLocalSockets.js";
+import { disconnectLocalSockets } from "../delivery/disconnectLocalSockets.js";
+const createBusMessageHandler = (params)=>{
+    const { connectionManager } = params;
+    return async (message)=>{
+        let missing;
+        missing = "send" === message.action ? await sendToLocalSockets({
+            ...params,
+            connectionIds: message.connectionIds,
+            data: message.data
+        }) : await disconnectLocalSockets({
+            ...params,
+            connectionIds: message.connectionIds
+        });
+        for (const connectionId of missing)await connectionManager.remove(connectionId);
+    };
+};
+export { createBusMessageHandler };
+
+//# sourceMappingURL=createBusMessageHandler.js.map

package/bus/createBusMessageHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bus/createBusMessageHandler.js","sources":["../../src/bus/createBusMessageHandler.ts"],"sourcesContent":["import type { Logger } from \"@webiny/api-core/features/logger/abstractions.js\";\nimport type { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport type { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\nimport type { WebsocketsMessageBus } from \"./abstractions.js\";\nimport { sendToLocalSockets } from \"~/delivery/sendToLocalSockets.js\";\nimport { disconnectLocalSockets } from \"~/delivery/disconnectLocalSockets.js\";\n\nexport interface ICreateBusMessageHandlerParams {\n    connectionManager: WebsocketsConnectionManager.Interface<unknown>;\n    adapter: WebsocketsServerAdapter.Interface<unknown>;\n    logger: Logger.Interface;\n}\n\n/*\n * Handles what other processes publish to this one: sends and disconnects for sockets this process\n * holds. Messages arrive here only for connections this process owns, so one with no socket left is\n * gone, and its connection is removed.\n */\nexport const createBusMessageHandler = (\n    params: ICreateBusMessageHandlerParams\n): WebsocketsMessageBus.Handler => {\n    const { connectionManager } = params;\n    return async message => {\n        let missing: string[];\n        if (message.action === \"send\") {\n            missing = await sendToLocalSockets({\n                ...params,\n                connectionIds: message.connectionIds,\n                data: message.data\n            });\n        } else {\n            missing = await disconnectLocalSockets({\n                ...params,\n                connectionIds: message.connectionIds\n            });\n        }\n        for (const connectionId of missing) {\n            await connectionManager.remove(connectionId);\n        }\n    };\n};\n"],"names":["createBusMessageHandler","params","connectionManager","message","missing","sendToLocalSockets","disconnectLocalSockets","connectionId"],"mappings":";;AAkBO,MAAMA,0BAA0B,CACnCC;IAEA,MAAM,EAAEC,iBAAiB,EAAE,GAAGD;IAC9B,OAAO,OAAME;QACT,IAAIC;QAEAA,UADAD,AAAmB,WAAnBA,QAAQ,MAAM,GACJ,MAAME,mBAAmB;YAC/B,GAAGJ,MAAM;YACT,eAAeE,QAAQ,aAAa;YACpC,MAAMA,QAAQ,IAAI;QACtB,KAEU,MAAMG,uBAAuB;YACnC,GAAGL,MAAM;YACT,eAAeE,QAAQ,aAAa;QACxC;QAEJ,KAAK,MAAMI,gBAAgBH,QACvB,MAAMF,kBAAkB,MAAM,CAACK;IAEvC;AACJ"}

package/connectionManager/ServerConnectionManager.d.ts

@@ -9,7 +9,7 @@ export declare class ServerConnectionManagerImpl implements WebsocketsConnection
     constructor(registry: ConnectionRegistry.Interface, owner: ProcessOwner.Interface);
     add(params: WebsocketsConnectionManager.AddParams<unknown>): Promise<void>;
     remove(connectionId: string): Promise<void>;
-    removeOwnedHere(connectionIds: string[]): Promise<void>;
+    groupByOwner(connectionIds: string[]): Promise<Map<string, string[]>>;
     getSocket(connectionId: string): unknown | undefined;
     getMetadata(connectionId: string): WebsocketsConnectionManager.ConnectionMetadata | undefined;
     updateLastSeen(connectionId: string): Promise<void>;

package/connectionManager/ServerConnectionManager.js

@@ -38,11 +38,17 @@ class ServerConnectionManagerImpl {
             throw err;
         }
     }
-    async removeOwnedHere(connectionIds) {
-        if (0 === connectionIds.length) return;
+    async groupByOwner(connectionIds) {
+        const byOwner = new Map();
+        if (0 === connectionIds.length) return byOwner;
         const connections = await this.registry.listViaConnections(connectionIds);
-        const owned = connections.filter((connection)=>connection.owner === this.owner.id);
-        for (const connection of owned)await this.remove(connection.connectionId);
+        for (const connection of connections){
+            if (!connection.owner) continue;
+            const group = byOwner.get(connection.owner) ?? [];
+            group.push(connection.connectionId);
+            byOwner.set(connection.owner, group);
+        }
+        return byOwner;
     }
     getSocket(connectionId) {
         return this.sockets.get(connectionId);

package/connectionManager/ServerConnectionManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"connectionManager/ServerConnectionManager.js","sources":["../../src/connectionManager/ServerConnectionManager.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { ProcessOwner } from \"@webiny/event-handler-standalone\";\nimport { WebsocketsConnectionManager } from \"./abstractions.js\";\n\nexport class ServerConnectionManagerImpl implements WebsocketsConnectionManager.Interface<unknown> {\n    private readonly sockets = new Map<string, unknown>();\n    private readonly metadata = new Map<string, WebsocketsConnectionManager.ConnectionMetadata>();\n\n    public constructor(\n        private readonly registry: ConnectionRegistry.Interface,\n        private readonly owner: ProcessOwner.Interface\n    ) {}\n\n    public async add(params: WebsocketsConnectionManager.AddParams<unknown>): Promise<void> {\n        this.sockets.set(params.connectionId, params.socket);\n        this.metadata.set(params.connectionId, {\n            connectionId: params.connectionId,\n            endpoint: params.endpoint,\n            connectedAt: params.connectedAt,\n            host: params.host,\n            headers: params.headers\n        });\n\n        // Save the connection to the shared registry. This is what lets the server later push a\n        // message to a specific user — e.g. when a background task finishes and calls\n        // SendToIdentity. That flow looks connections up in the registry by identity id to get their\n        // connection ids, then comes back here to fetch each live socket by connection id and send.\n        // The in-memory map above only answers \"socket for this connection id?\" — it can't be\n        // searched by identity, so without this registry row a targeted send finds nobody. The\n        // registry stores `connectedOn` as an ISO string (it compares connections against a recency\n        // cutoff), so we convert the epoch-ms `connectedAt`.\n        await this.registry.register({\n            connectionId: params.connectionId,\n            identity: params.identity,\n            tenant: params.tenant,\n            endpoint: params.endpoint,\n            connectedOn: new Date(params.connectedAt).toISOString(),\n            owner: this.owner.id\n        });\n    }\n\n    public async remove(connectionId: string): Promise<void> {\n        this.sockets.delete(connectionId);\n        this.metadata.delete(connectionId);\n        try {\n            await this.registry.unregister({ connectionId });\n        } catch (err) {\n            if ((err as any).code === \"CONNECTION_NOT_FOUND\") {\n                return;\n            }\n            throw err;\n        }\n    }\n\n    /*\n     * For connections this process has no socket for. Only the process that holds a socket can tell\n     * that it is gone, so a connection is removed only when this process is its owner. Another\n     * process's connections are left alone: with several processes, a socket missing here is\n     * usually open there. Rows without an owner are left to the heartbeat's stale cleanup.\n     */\n    public async removeOwnedHere(connectionIds: string[]): Promise<void> {\n        if (connectionIds.length === 0) {\n            return;\n        }\n        const connections = await this.registry.listViaConnections(connectionIds);\n        const owned = connections.filter(connection => connection.owner === this.owner.id);\n        for (const connection of owned) {\n            await this.remove(connection.connectionId);\n        }\n    }\n\n    public getSocket(connectionId: string): unknown | undefined {\n        return this.sockets.get(connectionId);\n    }\n\n    public getMetadata(\n        connectionId: string\n    ): WebsocketsConnectionManager.ConnectionMetadata | undefined {\n        return this.metadata.get(connectionId);\n    }\n\n    public async updateLastSeen(connectionId: string): Promise<void> {\n        await this.registry.updateLastSeen(connectionId);\n    }\n\n    public async cleanup(maxAge: number): Promise<string[]> {\n        const olderThan = new Date(Date.now() - maxAge);\n        const stale = await this.registry.listStale(olderThan);\n        const removed: string[] = [];\n        for (const entry of stale) {\n            await this.remove(entry.connectionId);\n            removed.push(entry.connectionId);\n        }\n        return removed;\n    }\n\n    public getActiveConnectionIds(): string[] {\n        return Array.from(this.sockets.keys());\n    }\n}\n\nexport const ServerConnectionManager = WebsocketsConnectionManager.createImplementation({\n    implementation: ServerConnectionManagerImpl,\n    dependencies: [ConnectionRegistry, ProcessOwner]\n});\n"],"names":["ServerConnectionManagerImpl","registry","owner","Map","params","Date","connectionId","err","connectionIds","connections","owned","connection","maxAge","olderThan","stale","removed","entry","Array","ServerConnectionManager","WebsocketsConnectionManager","ConnectionRegistry","ProcessOwner"],"mappings":";;;AAIO,MAAMA;IAIT,YACqBC,QAAsC,EACtCC,KAA6B,CAChD;aAFmBD,QAAQ,GAARA;aACAC,KAAK,GAALA;aALJ,OAAO,GAAG,IAAIC;aACd,QAAQ,GAAG,IAAIA;IAK7B;IAEH,MAAa,IAAIC,MAAsD,EAAiB;QACpF,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA,OAAO,YAAY,EAAEA,OAAO,MAAM;QACnD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA,OAAO,YAAY,EAAE;YACnC,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,aAAaA,OAAO,WAAW;YAC/B,MAAMA,OAAO,IAAI;YACjB,SAASA,OAAO,OAAO;QAC3B;QAUA,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACzB,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,QAAQA,OAAO,MAAM;YACrB,UAAUA,OAAO,QAAQ;YACzB,aAAa,IAAIC,KAAKD,OAAO,WAAW,EAAE,WAAW;YACrD,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE;QACxB;IACJ;IAEA,MAAa,OAAOE,YAAoB,EAAiB;QACrD,IAAI,CAAC,OAAO,CAAC,MAAM,CAACA;QACpB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAACA;QACrB,IAAI;YACA,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAAEA;YAAa;QAClD,EAAE,OAAOC,KAAK;YACV,IAAKA,AAAqB,2BAArBA,IAAY,IAAI,EACjB;YAEJ,MAAMA;QACV;IACJ;IAQA,MAAa,gBAAgBC,aAAuB,EAAiB;QACjE,IAAIA,AAAyB,MAAzBA,cAAc,MAAM,EACpB;QAEJ,MAAMC,cAAc,MAAM,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAACD;QAC3D,MAAME,QAAQD,YAAY,MAAM,CAACE,CAAAA,aAAcA,WAAW,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,EAAE;QACjF,KAAK,MAAMA,cAAcD,MACrB,MAAM,IAAI,CAAC,MAAM,CAACC,WAAW,YAAY;IAEjD;IAEO,UAAUL,YAAoB,EAAuB;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA;IAC5B;IAEO,YACHA,YAAoB,EACsC;QAC1D,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA;IAC7B;IAEA,MAAa,eAAeA,YAAoB,EAAiB;QAC7D,MAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,CAACA;IACvC;IAEA,MAAa,QAAQM,MAAc,EAAqB;QACpD,MAAMC,YAAY,IAAIR,KAAKA,KAAK,GAAG,KAAKO;QACxC,MAAME,QAAQ,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAACD;QAC5C,MAAME,UAAoB,EAAE;QAC5B,KAAK,MAAMC,SAASF,MAAO;YACvB,MAAM,IAAI,CAAC,MAAM,CAACE,MAAM,YAAY;YACpCD,QAAQ,IAAI,CAACC,MAAM,YAAY;QACnC;QACA,OAAOD;IACX;IAEO,yBAAmC;QACtC,OAAOE,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI;IACvC;AACJ;AAEO,MAAMC,0BAA0BC,4BAA4B,oBAAoB,CAAC;IACpF,gBAAgBnB;IAChB,cAAc;QAACoB;QAAoBC;KAAa;AACpD"}
+{"version":3,"file":"connectionManager/ServerConnectionManager.js","sources":["../../src/connectionManager/ServerConnectionManager.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { ProcessOwner } from \"@webiny/event-handler-standalone\";\nimport { WebsocketsConnectionManager } from \"./abstractions.js\";\n\nexport class ServerConnectionManagerImpl implements WebsocketsConnectionManager.Interface<unknown> {\n    private readonly sockets = new Map<string, unknown>();\n    private readonly metadata = new Map<string, WebsocketsConnectionManager.ConnectionMetadata>();\n\n    public constructor(\n        private readonly registry: ConnectionRegistry.Interface,\n        private readonly owner: ProcessOwner.Interface\n    ) {}\n\n    public async add(params: WebsocketsConnectionManager.AddParams<unknown>): Promise<void> {\n        this.sockets.set(params.connectionId, params.socket);\n        this.metadata.set(params.connectionId, {\n            connectionId: params.connectionId,\n            endpoint: params.endpoint,\n            connectedAt: params.connectedAt,\n            host: params.host,\n            headers: params.headers\n        });\n\n        // Save the connection to the shared registry. This is what lets the server later push a\n        // message to a specific user — e.g. when a background task finishes and calls\n        // SendToIdentity. That flow looks connections up in the registry by identity id to get their\n        // connection ids, then comes back here to fetch each live socket by connection id and send.\n        // The in-memory map above only answers \"socket for this connection id?\" — it can't be\n        // searched by identity, so without this registry row a targeted send finds nobody. The\n        // registry stores `connectedOn` as an ISO string (it compares connections against a recency\n        // cutoff), so we convert the epoch-ms `connectedAt`.\n        await this.registry.register({\n            connectionId: params.connectionId,\n            identity: params.identity,\n            tenant: params.tenant,\n            endpoint: params.endpoint,\n            connectedOn: new Date(params.connectedAt).toISOString(),\n            owner: this.owner.id\n        });\n    }\n\n    public async remove(connectionId: string): Promise<void> {\n        this.sockets.delete(connectionId);\n        this.metadata.delete(connectionId);\n        try {\n            await this.registry.unregister({ connectionId });\n        } catch (err) {\n            if ((err as any).code === \"CONNECTION_NOT_FOUND\") {\n                return;\n            }\n            throw err;\n        }\n    }\n\n    /*\n     * The process holding each of these connections, for connections this process has no socket for.\n     * Rows without an owner are left out: nobody can say who holds them, so they wait for the\n     * heartbeat's stale cleanup.\n     */\n    public async groupByOwner(connectionIds: string[]): Promise<Map<string, string[]>> {\n        const byOwner = new Map<string, string[]>();\n        if (connectionIds.length === 0) {\n            return byOwner;\n        }\n        const connections = await this.registry.listViaConnections(connectionIds);\n        for (const connection of connections) {\n            if (!connection.owner) {\n                continue;\n            }\n            const group = byOwner.get(connection.owner) ?? [];\n            group.push(connection.connectionId);\n            byOwner.set(connection.owner, group);\n        }\n        return byOwner;\n    }\n\n    public getSocket(connectionId: string): unknown | undefined {\n        return this.sockets.get(connectionId);\n    }\n\n    public getMetadata(\n        connectionId: string\n    ): WebsocketsConnectionManager.ConnectionMetadata | undefined {\n        return this.metadata.get(connectionId);\n    }\n\n    public async updateLastSeen(connectionId: string): Promise<void> {\n        await this.registry.updateLastSeen(connectionId);\n    }\n\n    public async cleanup(maxAge: number): Promise<string[]> {\n        const olderThan = new Date(Date.now() - maxAge);\n        const stale = await this.registry.listStale(olderThan);\n        const removed: string[] = [];\n        for (const entry of stale) {\n            await this.remove(entry.connectionId);\n            removed.push(entry.connectionId);\n        }\n        return removed;\n    }\n\n    public getActiveConnectionIds(): string[] {\n        return Array.from(this.sockets.keys());\n    }\n}\n\nexport const ServerConnectionManager = WebsocketsConnectionManager.createImplementation({\n    implementation: ServerConnectionManagerImpl,\n    dependencies: [ConnectionRegistry, ProcessOwner]\n});\n"],"names":["ServerConnectionManagerImpl","registry","owner","Map","params","Date","connectionId","err","connectionIds","byOwner","connections","connection","group","maxAge","olderThan","stale","removed","entry","Array","ServerConnectionManager","WebsocketsConnectionManager","ConnectionRegistry","ProcessOwner"],"mappings":";;;AAIO,MAAMA;IAIT,YACqBC,QAAsC,EACtCC,KAA6B,CAChD;aAFmBD,QAAQ,GAARA;aACAC,KAAK,GAALA;aALJ,OAAO,GAAG,IAAIC;aACd,QAAQ,GAAG,IAAIA;IAK7B;IAEH,MAAa,IAAIC,MAAsD,EAAiB;QACpF,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA,OAAO,YAAY,EAAEA,OAAO,MAAM;QACnD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA,OAAO,YAAY,EAAE;YACnC,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,aAAaA,OAAO,WAAW;YAC/B,MAAMA,OAAO,IAAI;YACjB,SAASA,OAAO,OAAO;QAC3B;QAUA,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACzB,cAAcA,OAAO,YAAY;YACjC,UAAUA,OAAO,QAAQ;YACzB,QAAQA,OAAO,MAAM;YACrB,UAAUA,OAAO,QAAQ;YACzB,aAAa,IAAIC,KAAKD,OAAO,WAAW,EAAE,WAAW;YACrD,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE;QACxB;IACJ;IAEA,MAAa,OAAOE,YAAoB,EAAiB;QACrD,IAAI,CAAC,OAAO,CAAC,MAAM,CAACA;QACpB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAACA;QACrB,IAAI;YACA,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAAEA;YAAa;QAClD,EAAE,OAAOC,KAAK;YACV,IAAKA,AAAqB,2BAArBA,IAAY,IAAI,EACjB;YAEJ,MAAMA;QACV;IACJ;IAOA,MAAa,aAAaC,aAAuB,EAAkC;QAC/E,MAAMC,UAAU,IAAIN;QACpB,IAAIK,AAAyB,MAAzBA,cAAc,MAAM,EACpB,OAAOC;QAEX,MAAMC,cAAc,MAAM,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAACF;QAC3D,KAAK,MAAMG,cAAcD,YAAa;YAClC,IAAI,CAACC,WAAW,KAAK,EACjB;YAEJ,MAAMC,QAAQH,QAAQ,GAAG,CAACE,WAAW,KAAK,KAAK,EAAE;YACjDC,MAAM,IAAI,CAACD,WAAW,YAAY;YAClCF,QAAQ,GAAG,CAACE,WAAW,KAAK,EAAEC;QAClC;QACA,OAAOH;IACX;IAEO,UAAUH,YAAoB,EAAuB;QACxD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAACA;IAC5B;IAEO,YACHA,YAAoB,EACsC;QAC1D,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACA;IAC7B;IAEA,MAAa,eAAeA,YAAoB,EAAiB;QAC7D,MAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,CAACA;IACvC;IAEA,MAAa,QAAQO,MAAc,EAAqB;QACpD,MAAMC,YAAY,IAAIT,KAAKA,KAAK,GAAG,KAAKQ;QACxC,MAAME,QAAQ,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAACD;QAC5C,MAAME,UAAoB,EAAE;QAC5B,KAAK,MAAMC,SAASF,MAAO;YACvB,MAAM,IAAI,CAAC,MAAM,CAACE,MAAM,YAAY;YACpCD,QAAQ,IAAI,CAACC,MAAM,YAAY;QACnC;QACA,OAAOD;IACX;IAEO,yBAAmC;QACtC,OAAOE,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI;IACvC;AACJ;AAEO,MAAMC,0BAA0BC,4BAA4B,oBAAoB,CAAC;IACpF,gBAAgBpB;IAChB,cAAc;QAACqB;QAAoBC;KAAa;AACpD"}

package/connectionManager/abstractions.d.ts

@@ -19,7 +19,7 @@ export interface IWebsocketsConnectionManagerMetadata {
 export interface IWebsocketsConnectionManager<TSocket> {
     add(params: IWebsocketsConnectionManagerAddParams<TSocket>): Promise<void>;
     remove(connectionId: string): Promise<void>;
-    removeOwnedHere(connectionIds: string[]): Promise<void>;
+    groupByOwner(connectionIds: string[]): Promise<Map<string, string[]>>;
     getSocket(connectionId: string): TSocket | undefined;
     getMetadata(connectionId: string): IWebsocketsConnectionManagerMetadata | undefined;
     updateLastSeen(connectionId: string): Promise<void>;

package/connectionManager/abstractions.js.map

@@ -1 +1 @@
-{"version":3,"file":"connectionManager/abstractions.js","sources":["../../src/connectionManager/abstractions.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { createAbstraction } from \"@webiny/feature/api\";\n\nexport interface IWebsocketsConnectionManagerAddParams<TSocket> {\n    connectionId: string;\n    socket: TSocket;\n    endpoint: string;\n    identity: ConnectionRegistry.Identity;\n    tenant: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManagerMetadata {\n    connectionId: string;\n    endpoint: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManager<TSocket> {\n    add(params: IWebsocketsConnectionManagerAddParams<TSocket>): Promise<void>;\n    remove(connectionId: string): Promise<void>;\n    removeOwnedHere(connectionIds: string[]): Promise<void>;\n    getSocket(connectionId: string): TSocket | undefined;\n    getMetadata(connectionId: string): IWebsocketsConnectionManagerMetadata | undefined;\n    updateLastSeen(connectionId: string): Promise<void>;\n    cleanup(maxAge: number): Promise<string[]>;\n    getActiveConnectionIds(): string[];\n}\n\nexport const WebsocketsConnectionManager = createAbstraction<IWebsocketsConnectionManager<unknown>>(\n    \"WebsocketsConnectionManager\"\n);\n\nexport namespace WebsocketsConnectionManager {\n    export type Interface<TSocket> = IWebsocketsConnectionManager<TSocket>;\n    export type AddParams<TSocket> = IWebsocketsConnectionManagerAddParams<TSocket>;\n    export type ConnectionMetadata = IWebsocketsConnectionManagerMetadata;\n}\n"],"names":["WebsocketsConnectionManager","createAbstraction"],"mappings":";AAiCO,MAAMA,8BAA8BC,kBACvC"}
+{"version":3,"file":"connectionManager/abstractions.js","sources":["../../src/connectionManager/abstractions.ts"],"sourcesContent":["import { ConnectionRegistry } from \"@webiny/api-websockets/exports/api.js\";\nimport { createAbstraction } from \"@webiny/feature/api\";\n\nexport interface IWebsocketsConnectionManagerAddParams<TSocket> {\n    connectionId: string;\n    socket: TSocket;\n    endpoint: string;\n    identity: ConnectionRegistry.Identity;\n    tenant: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManagerMetadata {\n    connectionId: string;\n    endpoint: string;\n    connectedAt: number;\n    host: string;\n    headers: Record<string, string>;\n}\n\nexport interface IWebsocketsConnectionManager<TSocket> {\n    add(params: IWebsocketsConnectionManagerAddParams<TSocket>): Promise<void>;\n    remove(connectionId: string): Promise<void>;\n    groupByOwner(connectionIds: string[]): Promise<Map<string, string[]>>;\n    getSocket(connectionId: string): TSocket | undefined;\n    getMetadata(connectionId: string): IWebsocketsConnectionManagerMetadata | undefined;\n    updateLastSeen(connectionId: string): Promise<void>;\n    cleanup(maxAge: number): Promise<string[]>;\n    getActiveConnectionIds(): string[];\n}\n\nexport const WebsocketsConnectionManager = createAbstraction<IWebsocketsConnectionManager<unknown>>(\n    \"WebsocketsConnectionManager\"\n);\n\nexport namespace WebsocketsConnectionManager {\n    export type Interface<TSocket> = IWebsocketsConnectionManager<TSocket>;\n    export type AddParams<TSocket> = IWebsocketsConnectionManagerAddParams<TSocket>;\n    export type ConnectionMetadata = IWebsocketsConnectionManagerMetadata;\n}\n"],"names":["WebsocketsConnectionManager","createAbstraction"],"mappings":";AAiCO,MAAMA,8BAA8BC,kBACvC"}

package/delivery/disconnectLocalSockets.d.ts

@@ -0,0 +1,10 @@
+import type { Logger } from "@webiny/api-core/features/logger/abstractions.js";
+import type { WebsocketsConnectionManager } from "../connectionManager/abstractions.js";
+import type { WebsocketsServerAdapter } from "../adapter/abstractions.js";
+export interface IDisconnectLocalSocketsParams {
+    connectionManager: WebsocketsConnectionManager.Interface<unknown>;
+    adapter: WebsocketsServerAdapter.Interface<unknown>;
+    logger: Logger.Interface;
+    connectionIds: string[];
+}
+export declare const disconnectLocalSockets: (params: IDisconnectLocalSocketsParams) => Promise<string[]>;

package/delivery/disconnectLocalSockets.js

@@ -0,0 +1,24 @@
+const disconnectLocalSockets = async (params)=>{
+    const { connectionManager, adapter, logger, connectionIds } = params;
+    const missing = [];
+    for (const connectionId of connectionIds){
+        const socket = connectionManager.getSocket(connectionId);
+        if (void 0 === socket) {
+            missing.push(connectionId);
+            continue;
+        }
+        try {
+            adapter.close(socket);
+            await connectionManager.remove(connectionId);
+        } catch (error) {
+            logger.error({
+                error,
+                connectionId
+            }, "[websockets] could not disconnect a connection");
+        }
+    }
+    return missing;
+};
+export { disconnectLocalSockets };
+
+//# sourceMappingURL=disconnectLocalSockets.js.map

package/delivery/disconnectLocalSockets.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"delivery/disconnectLocalSockets.js","sources":["../../src/delivery/disconnectLocalSockets.ts"],"sourcesContent":["import type { Logger } from \"@webiny/api-core/features/logger/abstractions.js\";\nimport type { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport type { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\n\nexport interface IDisconnectLocalSocketsParams {\n    connectionManager: WebsocketsConnectionManager.Interface<unknown>;\n    adapter: WebsocketsServerAdapter.Interface<unknown>;\n    logger: Logger.Interface;\n    connectionIds: string[];\n}\n\n/*\n * Closes each socket this process holds, removes its connection, and returns the ids it holds no\n * socket for. A failed close is logged and the others go on.\n */\nexport const disconnectLocalSockets = async (\n    params: IDisconnectLocalSocketsParams\n): Promise<string[]> => {\n    const { connectionManager, adapter, logger, connectionIds } = params;\n    const missing: string[] = [];\n    for (const connectionId of connectionIds) {\n        const socket = connectionManager.getSocket(connectionId);\n        if (socket === undefined) {\n            missing.push(connectionId);\n            continue;\n        }\n        try {\n            adapter.close(socket);\n            await connectionManager.remove(connectionId);\n        } catch (error) {\n            logger.error({ error, connectionId }, \"[websockets] could not disconnect a connection\");\n        }\n    }\n    return missing;\n};\n"],"names":["disconnectLocalSockets","params","connectionManager","adapter","logger","connectionIds","missing","connectionId","socket","undefined","error"],"mappings":"AAeO,MAAMA,yBAAyB,OAClCC;IAEA,MAAM,EAAEC,iBAAiB,EAAEC,OAAO,EAAEC,MAAM,EAAEC,aAAa,EAAE,GAAGJ;IAC9D,MAAMK,UAAoB,EAAE;IAC5B,KAAK,MAAMC,gBAAgBF,cAAe;QACtC,MAAMG,SAASN,kBAAkB,SAAS,CAACK;QAC3C,IAAIC,AAAWC,WAAXD,QAAsB;YACtBF,QAAQ,IAAI,CAACC;YACb;QACJ;QACA,IAAI;YACAJ,QAAQ,KAAK,CAACK;YACd,MAAMN,kBAAkB,MAAM,CAACK;QACnC,EAAE,OAAOG,OAAO;YACZN,OAAO,KAAK,CAAC;gBAAEM;gBAAOH;YAAa,GAAG;QAC1C;IACJ;IACA,OAAOD;AACX"}

package/delivery/sendToLocalSockets.d.ts

@@ -0,0 +1,12 @@
+import type { GenericRecord } from "@webiny/api/types.js";
+import type { Logger } from "@webiny/api-core/features/logger/abstractions.js";
+import type { WebsocketsConnectionManager } from "../connectionManager/abstractions.js";
+import type { WebsocketsServerAdapter } from "../adapter/abstractions.js";
+export interface ISendToLocalSocketsParams {
+    connectionManager: WebsocketsConnectionManager.Interface<unknown>;
+    adapter: WebsocketsServerAdapter.Interface<unknown>;
+    logger: Logger.Interface;
+    connectionIds: string[];
+    data: GenericRecord;
+}
+export declare const sendToLocalSockets: (params: ISendToLocalSocketsParams) => Promise<string[]>;

package/delivery/sendToLocalSockets.js

@@ -0,0 +1,24 @@
+const sendToLocalSockets = async (params)=>{
+    const { connectionManager, adapter, logger, connectionIds, data } = params;
+    const payload = JSON.stringify(data);
+    const missing = [];
+    for (const connectionId of connectionIds){
+        const socket = connectionManager.getSocket(connectionId);
+        if (void 0 === socket) {
+            missing.push(connectionId);
+            continue;
+        }
+        try {
+            await adapter.send(socket, payload);
+        } catch (error) {
+            logger.error({
+                error,
+                connectionId
+            }, "[websockets] could not send to a connection");
+        }
+    }
+    return missing;
+};
+export { sendToLocalSockets };
+
+//# sourceMappingURL=sendToLocalSockets.js.map

package/delivery/sendToLocalSockets.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"delivery/sendToLocalSockets.js","sources":["../../src/delivery/sendToLocalSockets.ts"],"sourcesContent":["import type { GenericRecord } from \"@webiny/api/types.js\";\nimport type { Logger } from \"@webiny/api-core/features/logger/abstractions.js\";\nimport type { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport type { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\n\nexport interface ISendToLocalSocketsParams {\n    connectionManager: WebsocketsConnectionManager.Interface<unknown>;\n    adapter: WebsocketsServerAdapter.Interface<unknown>;\n    logger: Logger.Interface;\n    connectionIds: string[];\n    data: GenericRecord;\n}\n\n/*\n * Sends `data` on each socket this process holds and returns the ids it holds no socket for. A failed\n * send is logged and the others go on.\n */\nexport const sendToLocalSockets = async (params: ISendToLocalSocketsParams): Promise<string[]> => {\n    const { connectionManager, adapter, logger, connectionIds, data } = params;\n    const payload = JSON.stringify(data);\n    const missing: string[] = [];\n    for (const connectionId of connectionIds) {\n        const socket = connectionManager.getSocket(connectionId);\n        if (socket === undefined) {\n            missing.push(connectionId);\n            continue;\n        }\n        try {\n            await adapter.send(socket, payload);\n        } catch (error) {\n            logger.error({ error, connectionId }, \"[websockets] could not send to a connection\");\n        }\n    }\n    return missing;\n};\n"],"names":["sendToLocalSockets","params","connectionManager","adapter","logger","connectionIds","data","payload","JSON","missing","connectionId","socket","undefined","error"],"mappings":"AAiBO,MAAMA,qBAAqB,OAAOC;IACrC,MAAM,EAAEC,iBAAiB,EAAEC,OAAO,EAAEC,MAAM,EAAEC,aAAa,EAAEC,IAAI,EAAE,GAAGL;IACpE,MAAMM,UAAUC,KAAK,SAAS,CAACF;IAC/B,MAAMG,UAAoB,EAAE;IAC5B,KAAK,MAAMC,gBAAgBL,cAAe;QACtC,MAAMM,SAAST,kBAAkB,SAAS,CAACQ;QAC3C,IAAIC,AAAWC,WAAXD,QAAsB;YACtBF,QAAQ,IAAI,CAACC;YACb;QACJ;QACA,IAAI;YACA,MAAMP,QAAQ,IAAI,CAACQ,QAAQJ;QAC/B,EAAE,OAAOM,OAAO;YACZT,OAAO,KAAK,CAAC;gBAAES;gBAAOH;YAAa,GAAG;QAC1C;IACJ;IACA,OAAOD;AACX"}

package/index.d.ts

@@ -3,7 +3,10 @@ export type { IWebsocketsServer, WebsocketsConnectionAuthenticator } from "./ser
 export * from "./adapter/abstractions.js";
 export * from "./upgradeHandler/abstractions.js";
 export * from "./connectionManager/abstractions.js";
+export { WebsocketsMessageBus } from "./bus/abstractions.js";
+export { createBusMessageHandler } from "./bus/createBusMessageHandler.js";
 export { ServerConnectionManager } from "./connectionManager/ServerConnectionManager.js";
 export { NodeWsAdapter } from "./adapter/NodeWsAdapter.js";
 export { ServerWebsocketsTransport } from "./transport/ServerWebsocketsTransport.js";
+export { InProcessMessageBus } from "./bus/InProcessMessageBus.js";
 export { WebsocketsStandaloneFeature } from "./WebsocketsStandaloneFeature.js";

package/index.js

@@ -2,7 +2,10 @@ export * from "./adapter/abstractions.js";
 export * from "./upgradeHandler/abstractions.js";
 export * from "./connectionManager/abstractions.js";
 export { attachWebsocketsServer, createWebsocketsServer } from "./server/WebsocketsServer.js";
+export { WebsocketsMessageBus } from "./bus/abstractions.js";
+export { createBusMessageHandler } from "./bus/createBusMessageHandler.js";
 export { ServerConnectionManager } from "./connectionManager/ServerConnectionManager.js";
 export { NodeWsAdapter } from "./adapter/NodeWsAdapter.js";
 export { ServerWebsocketsTransport } from "./transport/ServerWebsocketsTransport.js";
+export { InProcessMessageBus } from "./bus/InProcessMessageBus.js";
 export { WebsocketsStandaloneFeature } from "./WebsocketsStandaloneFeature.js";

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@webiny/api-websockets-standalone",
-	"version": "0.0.0-unstable.7c2def32a7",
+	"version": "0.0.0-unstable.9609e4e0c9",
 	"type": "module",
 	"repository": {
 		"type": "git",
@@ -16,21 +16,22 @@
 	},
 	"license": "MIT",
 	"dependencies": {
-		"@webiny/api": "0.0.0-unstable.7c2def32a7",
-		"@webiny/api-websockets": "0.0.0-unstable.7c2def32a7",
-		"@webiny/event-handler-standalone": "0.0.0-unstable.7c2def32a7",
-		"@webiny/feature": "0.0.0-unstable.7c2def32a7",
-		"@webiny/plugins": "0.0.0-unstable.7c2def32a7",
-		"@webiny/utils": "0.0.0-unstable.7c2def32a7",
-		"ws": "8.21.3",
-		"zod": "4.4.3"
+		"@webiny/api": "0.0.0-unstable.9609e4e0c9",
+		"@webiny/api-core": "0.0.0-unstable.9609e4e0c9",
+		"@webiny/api-websockets": "0.0.0-unstable.9609e4e0c9",
+		"@webiny/event-handler-standalone": "0.0.0-unstable.9609e4e0c9",
+		"@webiny/feature": "0.0.0-unstable.9609e4e0c9",
+		"@webiny/plugins": "0.0.0-unstable.9609e4e0c9",
+		"@webiny/utils": "0.0.0-unstable.9609e4e0c9",
+		"ws": "8.22.0",
+		"zod": "4.6.7"
 	},
 	"devDependencies": {
-		"@types/ws": "8.18.1",
-		"@webiny/build-tools": "0.0.0-unstable.7c2def32a7",
+		"@types/ws": "8.18.2",
+		"@webiny/build-tools": "0.0.0-unstable.9609e4e0c9",
 		"rimraf": "6.1.3",
 		"typescript": "7.0.2",
-		"vitest": "4.1.11"
+		"vitest": "5.0.5"
 	},
 	"publishConfig": {
 		"access": "public"

package/transport/ServerWebsocketsTransport.d.ts

@@ -1,13 +1,20 @@
 import { WebsocketsTransport } from "@webiny/api-websockets/exports/api.js";
 import type { GenericRecord } from "@webiny/api/types.js";
+import { Logger } from "@webiny/api-core/features/logger/abstractions.js";
+import { ProcessOwner } from "@webiny/event-handler-standalone";
 import { WebsocketsConnectionManager } from "../connectionManager/abstractions.js";
 import { WebsocketsServerAdapter } from "../adapter/abstractions.js";
+import { WebsocketsMessageBus } from "../bus/abstractions.js";
 export declare class ServerWebsocketsTransportImpl implements WebsocketsTransport.Interface {
     private readonly connectionManager;
     private readonly adapter;
-    constructor(connectionManager: WebsocketsConnectionManager.Interface<unknown>, adapter: WebsocketsServerAdapter.Interface<unknown>);
+    private readonly bus;
+    private readonly owner;
+    private readonly logger;
+    constructor(connectionManager: WebsocketsConnectionManager.Interface<unknown>, adapter: WebsocketsServerAdapter.Interface<unknown>, bus: WebsocketsMessageBus.Interface, owner: ProcessOwner.Interface, logger: Logger.Interface);
     send<T extends GenericRecord = GenericRecord>(connections: WebsocketsTransport.SendConnection[], data: WebsocketsTransport.SendData<T>): Promise<void>;
     disconnect(connections: WebsocketsTransport.DisconnectConnection[]): Promise<void>;
+    private forward;
 }
 export declare const ServerWebsocketsTransport: typeof ServerWebsocketsTransportImpl & {
     __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-websockets/transport").IWebsocketsTransport>;

package/transport/ServerWebsocketsTransport.js

@@ -1,54 +1,75 @@
 import { WebsocketsTransport } from "@webiny/api-websockets/exports/api.js";
+import { Logger } from "@webiny/api-core/features/logger/abstractions.js";
+import { ProcessOwner } from "@webiny/event-handler-standalone";
 import { WebsocketsConnectionManager } from "../connectionManager/abstractions.js";
 import { WebsocketsServerAdapter } from "../adapter/abstractions.js";
+import { WebsocketsMessageBus } from "../bus/abstractions.js";
+import { sendToLocalSockets } from "../delivery/sendToLocalSockets.js";
+import { disconnectLocalSockets } from "../delivery/disconnectLocalSockets.js";
 class ServerWebsocketsTransportImpl {
-    constructor(connectionManager, adapter){
+    constructor(connectionManager, adapter, bus, owner, logger){
         this.connectionManager = connectionManager;
         this.adapter = adapter;
+        this.bus = bus;
+        this.owner = owner;
+        this.logger = logger;
     }
     async send(connections, data) {
-        const missing = [];
-        for (const connection of connections){
-            const { connectionId } = connection;
-            const socket = this.connectionManager.getSocket(connectionId);
-            if (void 0 === socket) {
-                missing.push(connectionId);
-                continue;
-            }
-            try {
-                await this.adapter.send(socket, JSON.stringify(data));
-            } catch (ex) {
-                console.error(`Failed to send message to connection "${connectionId}". Check logs for more information.`);
-                console.log(ex);
-            }
-        }
-        await this.connectionManager.removeOwnedHere(missing);
+        const connectionIds = connections.map((connection)=>connection.connectionId);
+        const missing = await sendToLocalSockets({
+            connectionManager: this.connectionManager,
+            adapter: this.adapter,
+            logger: this.logger,
+            connectionIds,
+            data
+        });
+        await this.forward(missing, (ids)=>({
+                action: "send",
+                connectionIds: ids,
+                data
+            }));
     }
     async disconnect(connections) {
-        const missing = [];
-        for (const connection of connections){
-            const { connectionId } = connection;
-            const socket = this.connectionManager.getSocket(connectionId);
-            if (void 0 === socket) {
-                missing.push(connectionId);
+        const connectionIds = connections.map((connection)=>connection.connectionId);
+        const missing = await disconnectLocalSockets({
+            connectionManager: this.connectionManager,
+            adapter: this.adapter,
+            logger: this.logger,
+            connectionIds
+        });
+        await this.forward(missing, (ids)=>({
+                action: "disconnect",
+                connectionIds: ids
+            }));
+    }
+    async forward(missing, toMessage) {
+        const byOwner = await this.connectionManager.groupByOwner(missing);
+        for (const [owner, connectionIds] of byOwner){
+            if (owner === this.owner.id) {
+                for (const connectionId of connectionIds)await this.connectionManager.remove(connectionId);
                 continue;
             }
+            const message = toMessage(connectionIds);
             try {
-                this.adapter.close(socket);
-                await this.connectionManager.remove(connectionId);
-            } catch (ex) {
-                console.error(`Failed to disconnect connection "${connectionId}". Check logs for more information.`);
-                console.log(ex);
+                await this.bus.publish(owner, message);
+            } catch (error) {
+                this.logger.error({
+                    error,
+                    owner,
+                    connectionIds
+                }, "[websockets] could not publish to the process holding these connections");
             }
         }
-        await this.connectionManager.removeOwnedHere(missing);
     }
 }
 const ServerWebsocketsTransport = WebsocketsTransport.createImplementation({
     implementation: ServerWebsocketsTransportImpl,
     dependencies: [
         WebsocketsConnectionManager,
-        WebsocketsServerAdapter
+        WebsocketsServerAdapter,
+        WebsocketsMessageBus,
+        ProcessOwner,
+        Logger
     ]
 });
 export { ServerWebsocketsTransport, ServerWebsocketsTransportImpl };

package/transport/ServerWebsocketsTransport.js.map

@@ -1 +1 @@
-{"version":3,"file":"transport/ServerWebsocketsTransport.js","sources":["../../src/transport/ServerWebsocketsTransport.ts"],"sourcesContent":["import { WebsocketsTransport } from \"@webiny/api-websockets/exports/api.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\nimport { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\n\nexport class ServerWebsocketsTransportImpl implements WebsocketsTransport.Interface {\n    private readonly connectionManager: WebsocketsConnectionManager.Interface<unknown>;\n    private readonly adapter: WebsocketsServerAdapter.Interface<unknown>;\n\n    public constructor(\n        connectionManager: WebsocketsConnectionManager.Interface<unknown>,\n        adapter: WebsocketsServerAdapter.Interface<unknown>\n    ) {\n        this.connectionManager = connectionManager;\n        this.adapter = adapter;\n    }\n\n    public async send<T extends GenericRecord = GenericRecord>(\n        connections: WebsocketsTransport.SendConnection[],\n        data: WebsocketsTransport.SendData<T>\n    ): Promise<void> {\n        // Connections with no socket in this process; see `removeOwnedHere`.\n        const missing: string[] = [];\n        for (const connection of connections) {\n            const { connectionId } = connection;\n            const socket = this.connectionManager.getSocket(connectionId);\n            if (socket === undefined) {\n                missing.push(connectionId);\n                continue;\n            }\n            try {\n                await this.adapter.send(socket, JSON.stringify(data));\n            } catch (ex) {\n                console.error(\n                    `Failed to send message to connection \"${connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n        await this.connectionManager.removeOwnedHere(missing);\n    }\n\n    public async disconnect(\n        connections: WebsocketsTransport.DisconnectConnection[]\n    ): Promise<void> {\n        const missing: string[] = [];\n        for (const connection of connections) {\n            const { connectionId } = connection;\n            const socket = this.connectionManager.getSocket(connectionId);\n            if (socket === undefined) {\n                missing.push(connectionId);\n                continue;\n            }\n            try {\n                this.adapter.close(socket);\n                await this.connectionManager.remove(connectionId);\n            } catch (ex) {\n                console.error(\n                    `Failed to disconnect connection \"${connectionId}\". Check logs for more information.`\n                );\n                console.log(ex);\n            }\n        }\n        await this.connectionManager.removeOwnedHere(missing);\n    }\n}\n\nexport const ServerWebsocketsTransport = WebsocketsTransport.createImplementation({\n    implementation: ServerWebsocketsTransportImpl,\n    dependencies: [WebsocketsConnectionManager, WebsocketsServerAdapter]\n});\n"],"names":["ServerWebsocketsTransportImpl","connectionManager","adapter","connections","data","missing","connection","connectionId","socket","undefined","JSON","ex","console","ServerWebsocketsTransport","WebsocketsTransport","WebsocketsConnectionManager","WebsocketsServerAdapter"],"mappings":";;;AAKO,MAAMA;IAIT,YACIC,iBAAiE,EACjEC,OAAmD,CACrD;QACE,IAAI,CAAC,iBAAiB,GAAGD;QACzB,IAAI,CAAC,OAAO,GAAGC;IACnB;IAEA,MAAa,KACTC,WAAiD,EACjDC,IAAqC,EACxB;QAEb,MAAMC,UAAoB,EAAE;QAC5B,KAAK,MAAMC,cAAcH,YAAa;YAClC,MAAM,EAAEI,YAAY,EAAE,GAAGD;YACzB,MAAME,SAAS,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAACD;YAChD,IAAIC,AAAWC,WAAXD,QAAsB;gBACtBH,QAAQ,IAAI,CAACE;gBACb;YACJ;YACA,IAAI;gBACA,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAACC,QAAQE,KAAK,SAAS,CAACN;YACnD,EAAE,OAAOO,IAAI;gBACTC,QAAQ,KAAK,CACT,CAAC,sCAAsC,EAAEL,aAAa,mCAAmC,CAAC;gBAE9FK,QAAQ,GAAG,CAACD;YAChB;QACJ;QACA,MAAM,IAAI,CAAC,iBAAiB,CAAC,eAAe,CAACN;IACjD;IAEA,MAAa,WACTF,WAAuD,EAC1C;QACb,MAAME,UAAoB,EAAE;QAC5B,KAAK,MAAMC,cAAcH,YAAa;YAClC,MAAM,EAAEI,YAAY,EAAE,GAAGD;YACzB,MAAME,SAAS,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAACD;YAChD,IAAIC,AAAWC,WAAXD,QAAsB;gBACtBH,QAAQ,IAAI,CAACE;gBACb;YACJ;YACA,IAAI;gBACA,IAAI,CAAC,OAAO,CAAC,KAAK,CAACC;gBACnB,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAACD;YACxC,EAAE,OAAOI,IAAI;gBACTC,QAAQ,KAAK,CACT,CAAC,iCAAiC,EAAEL,aAAa,mCAAmC,CAAC;gBAEzFK,QAAQ,GAAG,CAACD;YAChB;QACJ;QACA,MAAM,IAAI,CAAC,iBAAiB,CAAC,eAAe,CAACN;IACjD;AACJ;AAEO,MAAMQ,4BAA4BC,oBAAoB,oBAAoB,CAAC;IAC9E,gBAAgBd;IAChB,cAAc;QAACe;QAA6BC;KAAwB;AACxE"}
+{"version":3,"file":"transport/ServerWebsocketsTransport.js","sources":["../../src/transport/ServerWebsocketsTransport.ts"],"sourcesContent":["import { WebsocketsTransport } from \"@webiny/api-websockets/exports/api.js\";\nimport type { GenericRecord } from \"@webiny/api/types.js\";\nimport { Logger } from \"@webiny/api-core/features/logger/abstractions.js\";\nimport { ProcessOwner } from \"@webiny/event-handler-standalone\";\nimport { WebsocketsConnectionManager } from \"~/connectionManager/abstractions.js\";\nimport { WebsocketsServerAdapter } from \"~/adapter/abstractions.js\";\nimport { WebsocketsMessageBus } from \"~/bus/abstractions.js\";\nimport { sendToLocalSockets } from \"~/delivery/sendToLocalSockets.js\";\nimport { disconnectLocalSockets } from \"~/delivery/disconnectLocalSockets.js\";\n\ntype ToMessage = (connectionIds: string[]) => WebsocketsMessageBus.Message;\n\n/*\n * Sends to the sockets this process holds, and hands the rest to the process that holds them, through\n * the message bus. With one process there is no rest, apart from rows a previous run of the process\n * left behind.\n */\nexport class ServerWebsocketsTransportImpl implements WebsocketsTransport.Interface {\n    public constructor(\n        private readonly connectionManager: WebsocketsConnectionManager.Interface<unknown>,\n        private readonly adapter: WebsocketsServerAdapter.Interface<unknown>,\n        private readonly bus: WebsocketsMessageBus.Interface,\n        private readonly owner: ProcessOwner.Interface,\n        private readonly logger: Logger.Interface\n    ) {}\n\n    public async send<T extends GenericRecord = GenericRecord>(\n        connections: WebsocketsTransport.SendConnection[],\n        data: WebsocketsTransport.SendData<T>\n    ): Promise<void> {\n        const connectionIds = connections.map(connection => connection.connectionId);\n        const missing = await sendToLocalSockets({\n            connectionManager: this.connectionManager,\n            adapter: this.adapter,\n            logger: this.logger,\n            connectionIds,\n            data\n        });\n        await this.forward(missing, ids => {\n            return { action: \"send\", connectionIds: ids, data };\n        });\n    }\n\n    public async disconnect(\n        connections: WebsocketsTransport.DisconnectConnection[]\n    ): Promise<void> {\n        const connectionIds = connections.map(connection => connection.connectionId);\n        const missing = await disconnectLocalSockets({\n            connectionManager: this.connectionManager,\n            adapter: this.adapter,\n            logger: this.logger,\n            connectionIds\n        });\n        await this.forward(missing, ids => {\n            return { action: \"disconnect\", connectionIds: ids };\n        });\n    }\n\n    /*\n     * Connections this process has no socket for. Its own are gone, so they are removed; another\n     * process's are published to that process. A failed publish is logged, not thrown: the message\n     * is lost, as it would be if that process were restarting.\n     */\n    private async forward(missing: string[], toMessage: ToMessage): Promise<void> {\n        const byOwner = await this.connectionManager.groupByOwner(missing);\n        for (const [owner, connectionIds] of byOwner) {\n            if (owner === this.owner.id) {\n                for (const connectionId of connectionIds) {\n                    await this.connectionManager.remove(connectionId);\n                }\n                continue;\n            }\n            const message = toMessage(connectionIds);\n            try {\n                await this.bus.publish(owner, message);\n            } catch (error) {\n                this.logger.error(\n                    { error, owner, connectionIds },\n                    \"[websockets] could not publish to the process holding these connections\"\n                );\n            }\n        }\n    }\n}\n\nexport const ServerWebsocketsTransport = WebsocketsTransport.createImplementation({\n    implementation: ServerWebsocketsTransportImpl,\n    dependencies: [\n        WebsocketsConnectionManager,\n        WebsocketsServerAdapter,\n        WebsocketsMessageBus,\n        ProcessOwner,\n        Logger\n    ]\n});\n"],"names":["ServerWebsocketsTransportImpl","connectionManager","adapter","bus","owner","logger","connections","data","connectionIds","connection","missing","sendToLocalSockets","ids","disconnectLocalSockets","toMessage","byOwner","connectionId","message","error","ServerWebsocketsTransport","WebsocketsTransport","WebsocketsConnectionManager","WebsocketsServerAdapter","WebsocketsMessageBus","ProcessOwner","Logger"],"mappings":";;;;;;;;AAiBO,MAAMA;IACT,YACqBC,iBAAiE,EACjEC,OAAmD,EACnDC,GAAmC,EACnCC,KAA6B,EAC7BC,MAAwB,CAC3C;aALmBJ,iBAAiB,GAAjBA;aACAC,OAAO,GAAPA;aACAC,GAAG,GAAHA;aACAC,KAAK,GAALA;aACAC,MAAM,GAANA;IAClB;IAEH,MAAa,KACTC,WAAiD,EACjDC,IAAqC,EACxB;QACb,MAAMC,gBAAgBF,YAAY,GAAG,CAACG,CAAAA,aAAcA,WAAW,YAAY;QAC3E,MAAMC,UAAU,MAAMC,mBAAmB;YACrC,mBAAmB,IAAI,CAAC,iBAAiB;YACzC,SAAS,IAAI,CAAC,OAAO;YACrB,QAAQ,IAAI,CAAC,MAAM;YACnBH;YACAD;QACJ;QACA,MAAM,IAAI,CAAC,OAAO,CAACG,SAASE,CAAAA,MACjB;gBAAE,QAAQ;gBAAQ,eAAeA;gBAAKL;YAAK;IAE1D;IAEA,MAAa,WACTD,WAAuD,EAC1C;QACb,MAAME,gBAAgBF,YAAY,GAAG,CAACG,CAAAA,aAAcA,WAAW,YAAY;QAC3E,MAAMC,UAAU,MAAMG,uBAAuB;YACzC,mBAAmB,IAAI,CAAC,iBAAiB;YACzC,SAAS,IAAI,CAAC,OAAO;YACrB,QAAQ,IAAI,CAAC,MAAM;YACnBL;QACJ;QACA,MAAM,IAAI,CAAC,OAAO,CAACE,SAASE,CAAAA,MACjB;gBAAE,QAAQ;gBAAc,eAAeA;YAAI;IAE1D;IAOA,MAAc,QAAQF,OAAiB,EAAEI,SAAoB,EAAiB;QAC1E,MAAMC,UAAU,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAACL;QAC1D,KAAK,MAAM,CAACN,OAAOI,cAAc,IAAIO,QAAS;YAC1C,IAAIX,UAAU,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE;gBACzB,KAAK,MAAMY,gBAAgBR,cACvB,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAACQ;gBAExC;YACJ;YACA,MAAMC,UAAUH,UAAUN;YAC1B,IAAI;gBACA,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAACJ,OAAOa;YAClC,EAAE,OAAOC,OAAO;gBACZ,IAAI,CAAC,MAAM,CAAC,KAAK,CACb;oBAAEA;oBAAOd;oBAAOI;gBAAc,GAC9B;YAER;QACJ;IACJ;AACJ;AAEO,MAAMW,4BAA4BC,oBAAoB,oBAAoB,CAAC;IAC9E,gBAAgBpB;IAChB,cAAc;QACVqB;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}