node.js 从控制器调用 ws.send

Question

我想发送一个 websocket，使用 express-ws 从不同的控制器发送出去，而不是通过路由，我在 server.js 中有以下代码：

var SocketController = require('./routes/ws.routes');
var app  = express();
var server  = require('http').createServer(app);
var expressWs = require('express-ws')(app);

app.ws('/', SocketController.socketFunction);

SocketController 看起来像这样：

exports.socketFunction = function (ws, req) {
    ws.on('message', function(msg) {
        console.log(msg);
        ws.send("hello");
    });
}

是否可以调用

ws.send()

来自另一个控制器的事件？我如何获得“ws”对象？

谢谢！

Answer 1

您必须将套接字存储在内存中。要从不同的模块访问存储的套接字，您可以将这些套接字的引用存储在单独的模块中。

例如，您可以创建一个模块 socketCollection 来存储所有活动的套接字。

socketCollection.js：

exports.socketCollection = [];

您可以在定义 Web 套接字服务器的位置导入此模块：

var socketCollection = require('./socketCollection');
var SocketController = require('./routes/ws.routes');
var app  = express();
var server  = require('http').createServer(app);
var expressWs = require('express-ws')(app);

expressWs.getWss().on('connection', function(ws) {
  socketCollection.push({
    id: 'someRandomSocketId',
    socket: ws,
  });
});

app.ws('/', SocketController.socketFunction);

现在，每次新客户端连接到服务器时，它的引用都会保存到 'socketCollection' 您可以通过导入此数组从任何模块访问这些客户端

var socketCollection = require('./socketCollection');
var ws = findSocket('someRandomSocketId', socketCollection);

var findSocket = function(id, socketCollection) {
   var result  = socketCollection.filter(function(x){return x.id == id;} );
   return result ? result[0].socket : null;
};