48 lines
1.1 KiB
TypeScript
48 lines
1.1 KiB
TypeScript
import type { Params } from "@valkyr/json-rpc";
|
|
|
|
export class SocketRegistry {
|
|
readonly #sockets = new Set<WebSocket>();
|
|
|
|
/**
|
|
* Add a socket to the pool.
|
|
*
|
|
* @param socket - WebSocket to add.
|
|
*/
|
|
add(socket: WebSocket): this {
|
|
this.#sockets.add(socket);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Remove a socket from the pool.
|
|
*
|
|
* @param socket - WebSocket to remove.
|
|
*/
|
|
del(socket: WebSocket): this {
|
|
this.#sockets.delete(socket);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Sends a JSON-RPC notification to all connected sockets.
|
|
*
|
|
* @param method - Method to send the notification to.
|
|
* @param params - Message data to send to the clients.
|
|
*/
|
|
notify(method: string, params: Params): this {
|
|
this.send(JSON.stringify({ jsonrpc: "2.0", method, params }));
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Transmits data to all registered WebSocket connections. Data can be a string,
|
|
* a Blob, an ArrayBuffer, or an ArrayBufferView.
|
|
*
|
|
* @param data - Data to send to each connected socket.
|
|
*/
|
|
send(data: string | ArrayBufferLike | Blob | ArrayBufferView): this {
|
|
this.#sockets.forEach((socket) => socket.send(data));
|
|
return this;
|
|
}
|
|
}
|