80 lines
3.3 KiB
JavaScript
80 lines
3.3 KiB
JavaScript
"use strict";
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
const env_1 = require("./env");
|
||
const console_1 = require("console");
|
||
/**
|
||
* 集群行为备忘：
|
||
* 当socket.io通过adapter在集群间通信时，测试行为如下（测试环境为pm2 + cluster-adapter，其它adpater启用时需要再测一次）：
|
||
* 1）当client连接到node1并join room1时，只有node1上会有create room事件（room结构本身在结点间并不共享）
|
||
* 2）当某一个node执行 .adapter.to('room1').emit()时，连接到任一结点的client均能收到消息（但使用room可以实现跨结点推包）
|
||
* 3) serverSideEmit执行时如果有callback，而不是所有的接收者都执行callback的话，会抛出一个异常（意味着不需要本结点来判定是否收到全部的返回值了）
|
||
*/
|
||
class DataSubscriber {
|
||
ns;
|
||
nsServer;
|
||
constructor(ns, nsServer) {
|
||
this.ns = ns;
|
||
this.nsServer = nsServer;
|
||
this.startup();
|
||
}
|
||
/**
|
||
* 来自外部的socket连接，监听数据变化
|
||
*/
|
||
startup() {
|
||
this.ns.on('connection', async (socket) => {
|
||
try {
|
||
const { instanceId } = (0, env_1.getClusterInfo)();
|
||
// console.log('on connection', instanceId);
|
||
socket.on('sub', async (events) => {
|
||
events.forEach((event) => socket.join(event));
|
||
});
|
||
socket.on('unsub', (events) => {
|
||
// console.log('instance:', process.env.NODE_APP_INSTANCE, 'on unsub', JSON.stringify(ids));
|
||
events.forEach((id) => {
|
||
socket.leave(id);
|
||
});
|
||
});
|
||
}
|
||
catch (err) {
|
||
socket.emit('error', err.toString());
|
||
}
|
||
});
|
||
if (this.nsServer) {
|
||
this.nsServer.on('connection', async (socket) => {
|
||
try {
|
||
const { instanceId } = (0, env_1.getClusterInfo)();
|
||
console.log('on nsServer connection', instanceId);
|
||
socket.on('sub', async (events) => {
|
||
console.log('on nsServer sub', instanceId, events);
|
||
events.forEach((event) => socket.join(event));
|
||
});
|
||
socket.on('unsub', (events) => {
|
||
// console.log('instance:', process.env.NODE_APP_INSTANCE, 'on unsub', JSON.stringify(ids));
|
||
events.forEach((id) => {
|
||
socket.leave(id);
|
||
});
|
||
});
|
||
}
|
||
catch (err) {
|
||
socket.emit('error', err.toString());
|
||
}
|
||
});
|
||
}
|
||
}
|
||
publishEvent(event, records, sid) {
|
||
const { instanceId } = (0, env_1.getClusterInfo)();
|
||
// console.log('publishEvent', instanceId);
|
||
if (sid) {
|
||
this.ns.to(event).except(sid).emit('data', records, event);
|
||
}
|
||
else {
|
||
this.ns.to(event).emit('data', records, event);
|
||
}
|
||
}
|
||
publishServerEvent(identifier, event, ...rest) {
|
||
(0, console_1.assert)(this.nsServer);
|
||
this.nsServer.to(identifier).emit(event, ...rest);
|
||
}
|
||
}
|
||
exports.default = DataSubscriber;
|