使用集群将Socket.IO扩展到多个Node.js进程

小编典典

使用集群将Socket.IO扩展到多个Node.js进程

javascript

可以说我在四个工作进程（伪）上具有以下内容：

// on the server
var express = require('express');
var server = express();
var socket = require('socket.io');
var io = socket.listen(server);

// socket.io
io.set('store', new socket.RedisStore);

// set-up connections...
io.sockets.on('connection', function(socket) {

  socket.on('join', function(rooms) {
    rooms.forEach(function(room) {
      socket.join(room);
    });
  });

  socket.on('leave', function(rooms) {
    rooms.forEach(function(room) {
      socket.leave(room);
    });
  });

});

// Emit a message every second
function send() {
  io.sockets.in('room').emit('data', 'howdy');
}

setInterval(send, 1000);

在浏览器上…

// on the client
socket = io.connect();
socket.emit('join', ['room']);

socket.on('data', function(data){
  console.log(data);
});

问题： 由于四个独立的工作进程发送消息，因此我每秒收到四条消息。

如何确保邮件仅发送一次？

阅读 387

2020-05-01

共1个答案

小编典典

编辑： 在Socket.IO 1.0+中，现在可以使用更简单的Redis适配器模块，而不是通过多个Redis客户端设置存储。

var io = require('socket.io')(3000);
var redis = require('socket.io-redis');
io.adapter(redis({ host: 'localhost', port: 6379 }));

下面显示的示例看起来更像这样：

var cluster = require('cluster');
var os = require('os');

if (cluster.isMaster) {
  // we create a HTTP server, but we do not use listen
  // that way, we have a socket.io server that doesn't accept connections
  var server = require('http').createServer();
  var io = require('socket.io').listen(server);
  var redis = require('socket.io-redis');

  io.adapter(redis({ host: 'localhost', port: 6379 }));

  setInterval(function() {
    // all workers will receive this in Redis, and emit
    io.emit('data', 'payload');
  }, 1000);

  for (var i = 0; i < os.cpus().length; i++) {
    cluster.fork();
  }

  cluster.on('exit', function(worker, code, signal) {
    console.log('worker ' + worker.process.pid + ' died');
  }); 
}

if (cluster.isWorker) {
  var express = require('express');
  var app = express();

  var http = require('http');
  var server = http.createServer(app);
  var io = require('socket.io').listen(server);
  var redis = require('socket.io-redis');

  io.adapter(redis({ host: 'localhost', port: 6379 }));
  io.on('connection', function(socket) {
    socket.emit('data', 'connected to worker: ' + cluster.worker.id);
  });

  app.listen(80);
}

如果您有一个主节点需要发布到其他Socket.IO进程，但本身不接受套接字连接，请使用socket.io-
emitter而不是socket.io-
redis。

如果无法扩展，请使用来运行Node应用程序DEBUG=*。Socket.IO现在实现调试，该调试还将打印出Redis适配器调试消息。输出示例：

socket.io:server initializing namespace / +0ms
socket.io:server creating engine.io instance with opts {"path":"/socket.io"} +2ms
socket.io:server attaching client serving req handler +2ms
socket.io-parser encoding packet {"type":2,"data":["event","payload"],"nsp":"/"} +0ms
socket.io-parser encoded {"type":2,"data":["event","payload"],"nsp":"/"} as 2["event","payload"] +1ms
socket.io-redis ignore same uid +0ms

如果您的主进程和子进程都显示相同的解析器消息，则您的应用程序正在正确扩展。

如果您是从单个工作人员中解雇，则设置应该不会有问题。您正在执行的操作是从所有四个工作人员发出的，并且由于Redis的发布/订阅，消息不是重复的，而是按您要求应用程序执行的那样写了四次。这是Redis的简单示意图：

Client  <--  Worker 1 emit -->  Redis
Client  <--  Worker 2  <----------|
Client  <--  Worker 3  <----------|
Client  <--  Worker 4  <----------|

如您所见，当您从某个工作程序发出时，它会将发布内容发布到Redis，并将从其他已订阅Redis数据库的工作程序中镜像出来。这也意味着您可以使用连接同一实例的多个套接字服务器，并且一台服务器上的发射将在所有连接的服务器上触发。

使用群集时，当客户端连接时，它将连接到您的四个工作线程之一，而不是全部四个。这也意味着您从该工作人员发出的任何内容只会显示给客户端一次。因此，是的，应用程序正在扩展，但是您的操作方式是，您要从所有四个工作程序中释放资源，而Redis数据库则使它像在单个工作程序上调用了四次一样。如果客户端实际上连接到您的所有四个套接字实例，则它们每秒将接收十六条消息，而不是四条。

套接字处理的类型取决于您将要拥有的应用程序的类型。如果您要分别处理客户端，那么应该没有问题，因为连接事件将仅对每个客户端一个工作人员触发。如果需要全局“心跳”，则可以在主进程中使用套接字处理程序。由于工作人员在主进程死亡时死亡，因此您应抵消主进程的连接负载，并让子进程处理连接。这是一个例子：

var cluster = require('cluster');
var os = require('os');

if (cluster.isMaster) {
  // we create a HTTP server, but we do not use listen
  // that way, we have a socket.io server that doesn't accept connections
  var server = require('http').createServer();
  var io = require('socket.io').listen(server);

  var RedisStore = require('socket.io/lib/stores/redis');
  var redis = require('socket.io/node_modules/redis');

  io.set('store', new RedisStore({
    redisPub: redis.createClient(),
    redisSub: redis.createClient(),
    redisClient: redis.createClient()
  }));

  setInterval(function() {
    // all workers will receive this in Redis, and emit
    io.sockets.emit('data', 'payload');
  }, 1000);

  for (var i = 0; i < os.cpus().length; i++) {
    cluster.fork();
  }

  cluster.on('exit', function(worker, code, signal) {
    console.log('worker ' + worker.process.pid + ' died');
  }); 
}

if (cluster.isWorker) {
  var express = require('express');
  var app = express();

  var http = require('http');
  var server = http.createServer(app);
  var io = require('socket.io').listen(server);

  var RedisStore = require('socket.io/lib/stores/redis');
  var redis = require('socket.io/node_modules/redis');

  io.set('store', new RedisStore({
    redisPub: redis.createClient(),
    redisSub: redis.createClient(),
    redisClient: redis.createClient()
  }));

  io.sockets.on('connection', function(socket) {
    socket.emit('data', 'connected to worker: ' + cluster.worker.id);
  });

  app.listen(80);
}

在该示例中，有五个Socket.IO实例，一个是主实例，四个是子实例。主服务器从不调用，listen()因此该进程没有连接开销。但是，如果您在主进程上调用发射，它将被发布到Redis，并且四个工作进程将在其客户端上执行发射。这抵消了对工作人员的连接负载，如果工作人员死亡，则主应用程序中的主要应用程序逻辑将保持不变。

请注意，使用Redis，所有发射，即使是在名称空间或房间中，也会由其他工作进程处理，就像您从该进程触发了发射一样。换句话说，如果您有两个带有一个Redis实例的Socket.IO实例，则emit()在第一个工作线程中调用套接字将向其客户端发送数据，而第二个工作线程将执行与您从该工作线程调用emit相同的操作。

2020-05-01