580 lines
19 KiB
JavaScript
580 lines
19 KiB
JavaScript
"use strict";
|
||
var __importDefault = (this && this.__importDefault) || function (mod) {
|
||
return (mod && mod.__esModule) ? mod : { "default": mod };
|
||
};
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
exports.Namespace = exports.RESERVED_EVENTS = void 0;
|
||
const socket_1 = require("./socket");
|
||
const typed_events_1 = require("./typed-events");
|
||
const debug_1 = __importDefault(require("debug"));
|
||
const broadcast_operator_1 = require("./broadcast-operator");
|
||
const debug = (0, debug_1.default)("socket.io:namespace");
|
||
exports.RESERVED_EVENTS = new Set(["connect", "connection", "new_namespace"]);
|
||
/**
|
||
* A Namespace is a communication channel that allows you to split the logic of your application over a single shared
|
||
* connection.
|
||
*
|
||
* Each namespace has its own:
|
||
*
|
||
* - event handlers
|
||
*
|
||
* ```
|
||
* io.of("/orders").on("connection", (socket) => {
|
||
* socket.on("order:list", () => {});
|
||
* socket.on("order:create", () => {});
|
||
* });
|
||
*
|
||
* io.of("/users").on("connection", (socket) => {
|
||
* socket.on("user:list", () => {});
|
||
* });
|
||
* ```
|
||
*
|
||
* - rooms
|
||
*
|
||
* ```
|
||
* const orderNamespace = io.of("/orders");
|
||
*
|
||
* orderNamespace.on("connection", (socket) => {
|
||
* socket.join("room1");
|
||
* orderNamespace.to("room1").emit("hello");
|
||
* });
|
||
*
|
||
* const userNamespace = io.of("/users");
|
||
*
|
||
* userNamespace.on("connection", (socket) => {
|
||
* socket.join("room1"); // distinct from the room in the "orders" namespace
|
||
* userNamespace.to("room1").emit("holà");
|
||
* });
|
||
* ```
|
||
*
|
||
* - middlewares
|
||
*
|
||
* ```
|
||
* const orderNamespace = io.of("/orders");
|
||
*
|
||
* orderNamespace.use((socket, next) => {
|
||
* // ensure the socket has access to the "orders" namespace
|
||
* });
|
||
*
|
||
* const userNamespace = io.of("/users");
|
||
*
|
||
* userNamespace.use((socket, next) => {
|
||
* // ensure the socket has access to the "users" namespace
|
||
* });
|
||
* ```
|
||
*/
|
||
class Namespace extends typed_events_1.StrictEventEmitter {
|
||
/**
|
||
* Namespace constructor.
|
||
*
|
||
* @param server instance
|
||
* @param name
|
||
*/
|
||
constructor(server, name) {
|
||
super();
|
||
this.sockets = new Map();
|
||
/** @private */
|
||
this._fns = [];
|
||
/** @private */
|
||
this._ids = 0;
|
||
this.server = server;
|
||
this.name = name;
|
||
this._initAdapter();
|
||
}
|
||
/**
|
||
* Initializes the `Adapter` for this nsp.
|
||
* Run upon changing adapter by `Server#adapter`
|
||
* in addition to the constructor.
|
||
*
|
||
* @private
|
||
*/
|
||
_initAdapter() {
|
||
// @ts-ignore
|
||
this.adapter = new (this.server.adapter())(this);
|
||
}
|
||
/**
|
||
* Registers a middleware, which is a function that gets executed for every incoming {@link Socket}.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* myNamespace.use((socket, next) => {
|
||
* // ...
|
||
* next();
|
||
* });
|
||
*
|
||
* @param fn - the middleware function
|
||
*/
|
||
use(fn) {
|
||
this._fns.push(fn);
|
||
return this;
|
||
}
|
||
/**
|
||
* Executes the middleware for an incoming client.
|
||
*
|
||
* @param socket - the socket that will get added
|
||
* @param fn - last fn call in the middleware
|
||
* @private
|
||
*/
|
||
run(socket, fn) {
|
||
const fns = this._fns.slice(0);
|
||
if (!fns.length)
|
||
return fn(null);
|
||
function run(i) {
|
||
fns[i](socket, function (err) {
|
||
// upon error, short-circuit
|
||
if (err)
|
||
return fn(err);
|
||
// if no middleware left, summon callback
|
||
if (!fns[i + 1])
|
||
return fn(null);
|
||
// go on to next
|
||
run(i + 1);
|
||
});
|
||
}
|
||
run(0);
|
||
}
|
||
/**
|
||
* Targets a room when emitting.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // the “foo” event will be broadcast to all connected clients in the “room-101” room
|
||
* myNamespace.to("room-101").emit("foo", "bar");
|
||
*
|
||
* // with an array of rooms (a client will be notified at most once)
|
||
* myNamespace.to(["room-101", "room-102"]).emit("foo", "bar");
|
||
*
|
||
* // with multiple chained calls
|
||
* myNamespace.to("room-101").to("room-102").emit("foo", "bar");
|
||
*
|
||
* @param room - a room, or an array of rooms
|
||
* @return a new {@link BroadcastOperator} instance for chaining
|
||
*/
|
||
to(room) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).to(room);
|
||
}
|
||
/**
|
||
* Targets a room when emitting. Similar to `to()`, but might feel clearer in some cases:
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // disconnect all clients in the "room-101" room
|
||
* myNamespace.in("room-101").disconnectSockets();
|
||
*
|
||
* @param room - a room, or an array of rooms
|
||
* @return a new {@link BroadcastOperator} instance for chaining
|
||
*/
|
||
in(room) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).in(room);
|
||
}
|
||
/**
|
||
* Excludes a room when emitting.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // the "foo" event will be broadcast to all connected clients, except the ones that are in the "room-101" room
|
||
* myNamespace.except("room-101").emit("foo", "bar");
|
||
*
|
||
* // with an array of rooms
|
||
* myNamespace.except(["room-101", "room-102"]).emit("foo", "bar");
|
||
*
|
||
* // with multiple chained calls
|
||
* myNamespace.except("room-101").except("room-102").emit("foo", "bar");
|
||
*
|
||
* @param room - a room, or an array of rooms
|
||
* @return a new {@link BroadcastOperator} instance for chaining
|
||
*/
|
||
except(room) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).except(room);
|
||
}
|
||
/**
|
||
* Adds a new client.
|
||
*
|
||
* @return {Socket}
|
||
* @private
|
||
*/
|
||
async _add(client, auth, fn) {
|
||
var _a;
|
||
debug("adding socket to nsp %s", this.name);
|
||
const socket = await this._createSocket(client, auth);
|
||
if (
|
||
// @ts-ignore
|
||
((_a = this.server.opts.connectionStateRecovery) === null || _a === void 0 ? void 0 : _a.skipMiddlewares) &&
|
||
socket.recovered &&
|
||
client.conn.readyState === "open") {
|
||
return this._doConnect(socket, fn);
|
||
}
|
||
this.run(socket, (err) => {
|
||
process.nextTick(() => {
|
||
if ("open" !== client.conn.readyState) {
|
||
debug("next called after client was closed - ignoring socket");
|
||
socket._cleanup();
|
||
return;
|
||
}
|
||
if (err) {
|
||
debug("middleware error, sending CONNECT_ERROR packet to the client");
|
||
socket._cleanup();
|
||
if (client.conn.protocol === 3) {
|
||
return socket._error(err.data || err.message);
|
||
}
|
||
else {
|
||
return socket._error({
|
||
message: err.message,
|
||
data: err.data,
|
||
});
|
||
}
|
||
}
|
||
this._doConnect(socket, fn);
|
||
});
|
||
});
|
||
}
|
||
async _createSocket(client, auth) {
|
||
const sessionId = auth.pid;
|
||
const offset = auth.offset;
|
||
if (
|
||
// @ts-ignore
|
||
this.server.opts.connectionStateRecovery &&
|
||
typeof sessionId === "string" &&
|
||
typeof offset === "string") {
|
||
let session;
|
||
try {
|
||
session = await this.adapter.restoreSession(sessionId, offset);
|
||
}
|
||
catch (e) {
|
||
debug("error while restoring session: %s", e);
|
||
}
|
||
if (session) {
|
||
debug("connection state recovered for sid %s", session.sid);
|
||
return new socket_1.Socket(this, client, auth, session);
|
||
}
|
||
}
|
||
return new socket_1.Socket(this, client, auth);
|
||
}
|
||
_doConnect(socket, fn) {
|
||
// track socket
|
||
this.sockets.set(socket.id, socket);
|
||
// it's paramount that the internal `onconnect` logic
|
||
// fires before user-set events to prevent state order
|
||
// violations (such as a disconnection before the connection
|
||
// logic is complete)
|
||
socket._onconnect();
|
||
if (fn)
|
||
fn(socket);
|
||
// fire user-set events
|
||
this.emitReserved("connect", socket);
|
||
this.emitReserved("connection", socket);
|
||
}
|
||
/**
|
||
* Removes a client. Called by each `Socket`.
|
||
*
|
||
* @private
|
||
*/
|
||
_remove(socket) {
|
||
if (this.sockets.has(socket.id)) {
|
||
this.sockets.delete(socket.id);
|
||
}
|
||
else {
|
||
debug("ignoring remove for %s", socket.id);
|
||
}
|
||
}
|
||
/**
|
||
* Emits to all connected clients.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* myNamespace.emit("hello", "world");
|
||
*
|
||
* // all serializable datastructures are supported (no need to call JSON.stringify)
|
||
* myNamespace.emit("hello", 1, "2", { 3: ["4"], 5: Uint8Array.from([6]) });
|
||
*
|
||
* // with an acknowledgement from the clients
|
||
* myNamespace.timeout(1000).emit("some-event", (err, responses) => {
|
||
* if (err) {
|
||
* // some clients did not acknowledge the event in the given delay
|
||
* } else {
|
||
* console.log(responses); // one response per client
|
||
* }
|
||
* });
|
||
*
|
||
* @return Always true
|
||
*/
|
||
emit(ev, ...args) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).emit(ev, ...args);
|
||
}
|
||
/**
|
||
* Sends a `message` event to all clients.
|
||
*
|
||
* This method mimics the WebSocket.send() method.
|
||
*
|
||
* @see https://developer.mozilla.org/en-US/docs/Web/API/WebSocket/send
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* myNamespace.send("hello");
|
||
*
|
||
* // this is equivalent to
|
||
* myNamespace.emit("message", "hello");
|
||
*
|
||
* @return self
|
||
*/
|
||
send(...args) {
|
||
// This type-cast is needed because EmitEvents likely doesn't have `message` as a key.
|
||
// if you specify the EmitEvents, the type of args will be never.
|
||
this.emit("message", ...args);
|
||
return this;
|
||
}
|
||
/**
|
||
* Sends a `message` event to all clients. Sends a `message` event. Alias of {@link send}.
|
||
*
|
||
* @return self
|
||
*/
|
||
write(...args) {
|
||
// This type-cast is needed because EmitEvents likely doesn't have `message` as a key.
|
||
// if you specify the EmitEvents, the type of args will be never.
|
||
this.emit("message", ...args);
|
||
return this;
|
||
}
|
||
/**
|
||
* Sends a message to the other Socket.IO servers of the cluster.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* myNamespace.serverSideEmit("hello", "world");
|
||
*
|
||
* myNamespace.on("hello", (arg1) => {
|
||
* console.log(arg1); // prints "world"
|
||
* });
|
||
*
|
||
* // acknowledgements (without binary content) are supported too:
|
||
* myNamespace.serverSideEmit("ping", (err, responses) => {
|
||
* if (err) {
|
||
* // some servers did not acknowledge the event in the given delay
|
||
* } else {
|
||
* console.log(responses); // one response per server (except the current one)
|
||
* }
|
||
* });
|
||
*
|
||
* myNamespace.on("ping", (cb) => {
|
||
* cb("pong");
|
||
* });
|
||
*
|
||
* @param ev - the event name
|
||
* @param args - an array of arguments, which may include an acknowledgement callback at the end
|
||
*/
|
||
serverSideEmit(ev, ...args) {
|
||
if (exports.RESERVED_EVENTS.has(ev)) {
|
||
throw new Error(`"${String(ev)}" is a reserved event name`);
|
||
}
|
||
args.unshift(ev);
|
||
this.adapter.serverSideEmit(args);
|
||
return true;
|
||
}
|
||
/**
|
||
* Sends a message and expect an acknowledgement from the other Socket.IO servers of the cluster.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* try {
|
||
* const responses = await myNamespace.serverSideEmitWithAck("ping");
|
||
* console.log(responses); // one response per server (except the current one)
|
||
* } catch (e) {
|
||
* // some servers did not acknowledge the event in the given delay
|
||
* }
|
||
*
|
||
* @param ev - the event name
|
||
* @param args - an array of arguments
|
||
*
|
||
* @return a Promise that will be fulfilled when all servers have acknowledged the event
|
||
*/
|
||
serverSideEmitWithAck(ev, ...args) {
|
||
return new Promise((resolve, reject) => {
|
||
args.push((err, responses) => {
|
||
if (err) {
|
||
err.responses = responses;
|
||
return reject(err);
|
||
}
|
||
else {
|
||
return resolve(responses);
|
||
}
|
||
});
|
||
this.serverSideEmit(ev, ...args);
|
||
});
|
||
}
|
||
/**
|
||
* Called when a packet is received from another Socket.IO server
|
||
*
|
||
* @param args - an array of arguments, which may include an acknowledgement callback at the end
|
||
*
|
||
* @private
|
||
*/
|
||
_onServerSideEmit(args) {
|
||
super.emitUntyped.apply(this, args);
|
||
}
|
||
/**
|
||
* Gets a list of clients.
|
||
*
|
||
* @deprecated this method will be removed in the next major release, please use {@link Namespace#serverSideEmit} or
|
||
* {@link Namespace#fetchSockets} instead.
|
||
*/
|
||
allSockets() {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).allSockets();
|
||
}
|
||
/**
|
||
* Sets the compress flag.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* myNamespace.compress(false).emit("hello");
|
||
*
|
||
* @param compress - if `true`, compresses the sending data
|
||
* @return self
|
||
*/
|
||
compress(compress) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).compress(compress);
|
||
}
|
||
/**
|
||
* Sets a modifier for a subsequent event emission that the event data may be lost if the client is not ready to
|
||
* receive messages (because of network slowness or other issues, or because they’re connected through long polling
|
||
* and is in the middle of a request-response cycle).
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* myNamespace.volatile.emit("hello"); // the clients may or may not receive it
|
||
*
|
||
* @return self
|
||
*/
|
||
get volatile() {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).volatile;
|
||
}
|
||
/**
|
||
* Sets a modifier for a subsequent event emission that the event data will only be broadcast to the current node.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // the “foo” event will be broadcast to all connected clients on this node
|
||
* myNamespace.local.emit("foo", "bar");
|
||
*
|
||
* @return a new {@link BroadcastOperator} instance for chaining
|
||
*/
|
||
get local() {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).local;
|
||
}
|
||
/**
|
||
* Adds a timeout in milliseconds for the next operation.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* myNamespace.timeout(1000).emit("some-event", (err, responses) => {
|
||
* if (err) {
|
||
* // some clients did not acknowledge the event in the given delay
|
||
* } else {
|
||
* console.log(responses); // one response per client
|
||
* }
|
||
* });
|
||
*
|
||
* @param timeout
|
||
*/
|
||
timeout(timeout) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).timeout(timeout);
|
||
}
|
||
/**
|
||
* Returns the matching socket instances.
|
||
*
|
||
* Note: this method also works within a cluster of multiple Socket.IO servers, with a compatible {@link Adapter}.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // return all Socket instances
|
||
* const sockets = await myNamespace.fetchSockets();
|
||
*
|
||
* // return all Socket instances in the "room1" room
|
||
* const sockets = await myNamespace.in("room1").fetchSockets();
|
||
*
|
||
* for (const socket of sockets) {
|
||
* console.log(socket.id);
|
||
* console.log(socket.handshake);
|
||
* console.log(socket.rooms);
|
||
* console.log(socket.data);
|
||
*
|
||
* socket.emit("hello");
|
||
* socket.join("room1");
|
||
* socket.leave("room2");
|
||
* socket.disconnect();
|
||
* }
|
||
*/
|
||
fetchSockets() {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).fetchSockets();
|
||
}
|
||
/**
|
||
* Makes the matching socket instances join the specified rooms.
|
||
*
|
||
* Note: this method also works within a cluster of multiple Socket.IO servers, with a compatible {@link Adapter}.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // make all socket instances join the "room1" room
|
||
* myNamespace.socketsJoin("room1");
|
||
*
|
||
* // make all socket instances in the "room1" room join the "room2" and "room3" rooms
|
||
* myNamespace.in("room1").socketsJoin(["room2", "room3"]);
|
||
*
|
||
* @param room - a room, or an array of rooms
|
||
*/
|
||
socketsJoin(room) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).socketsJoin(room);
|
||
}
|
||
/**
|
||
* Makes the matching socket instances leave the specified rooms.
|
||
*
|
||
* Note: this method also works within a cluster of multiple Socket.IO servers, with a compatible {@link Adapter}.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // make all socket instances leave the "room1" room
|
||
* myNamespace.socketsLeave("room1");
|
||
*
|
||
* // make all socket instances in the "room1" room leave the "room2" and "room3" rooms
|
||
* myNamespace.in("room1").socketsLeave(["room2", "room3"]);
|
||
*
|
||
* @param room - a room, or an array of rooms
|
||
*/
|
||
socketsLeave(room) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).socketsLeave(room);
|
||
}
|
||
/**
|
||
* Makes the matching socket instances disconnect.
|
||
*
|
||
* Note: this method also works within a cluster of multiple Socket.IO servers, with a compatible {@link Adapter}.
|
||
*
|
||
* @example
|
||
* const myNamespace = io.of("/my-namespace");
|
||
*
|
||
* // make all socket instances disconnect (the connections might be kept alive for other namespaces)
|
||
* myNamespace.disconnectSockets();
|
||
*
|
||
* // make all socket instances in the "room1" room disconnect and close the underlying connections
|
||
* myNamespace.in("room1").disconnectSockets(true);
|
||
*
|
||
* @param close - whether to close the underlying connection
|
||
*/
|
||
disconnectSockets(close = false) {
|
||
return new broadcast_operator_1.BroadcastOperator(this.adapter).disconnectSockets(close);
|
||
}
|
||
}
|
||
exports.Namespace = Namespace;
|