Nodejs Socket+TCP，实现小小简单聊天程序

好的，下面是关于如何使用 Node.js 实现一个简单的 TCP 聊天程序的详细内容，包括服务器端和客户端的代码。

Node.js Socket+TCP，实现小小简单聊天程序

这段时间我在重新学习 Node.js，今天我完成了一个基于控制台模式的聊天小程序。这个程序主要用到了 Node.js 的 net 模块来实现 TCP 通信。让我们一起来看看具体的实现过程吧！

实现内容

该聊天程序包含一个服务器和多个客户端。服务器会监视所有客户端的连接状态，并实现广播功能。具体来说，只要有一个客户端连接成功，它就会立即加入当前的聊天状态。这个应用虽然简单，但足以展示基本的 TCP 通信原理。

服务器端代码

var sockets = []; // 存储所有客户端连接

var server = require('net').createServer(function(socket) {
  console.log('new connection');
  
  sockets.push(socket); // 新连接时，将 socket 添加到数组中
  
  socket.setEncoding('utf8'); // 设置编码为 UTF-8
  
  socket.on('data', function(data) {
    // 广播数据给其他所有客户端
    sockets.forEach(function(otherSocket) {
      if (otherSocket !== socket) {
        otherSocket.write(data + "\n");
      }
    });
  });

  socket.on('end', function() {
    console.log('client connection ended!');
  });

  socket.on('close', function() {
    console.log('connection closed');
    var index = sockets.indexOf(socket); // 找到并移除关闭的 socket
    sockets.splice(index, 1);
  });
});

server.listen(4006, function() {
  console.log('waiting for client connection···');
  var address = server.address();
  console.log('opened server on %j', address);
}).on('close', function() {
  console.log('server closed!');
}).on('error', function(err) {
  console.log('Server error:' + err.message);
});

客户端代码

var net = require('net');
var definePort = 4006;
var quitting = false;
var retryTimeout = 3000;
var retriedTimes = 0;
var maxRetries = 10;
var client;
var myName = "";

process.stdin.resume();

process.stdin.on('data', function(data) {
  if (data.toString().trim().toLowerCase() === 'quit') {
    quitting = true;
    console.log('quitting···');
    client.end();
    process.stdin.pause();
  } else {
    if (myName !== "") {
      client.write(myName + ":" + data);
    } else {
      myName = data.trim();
      console.log('Your nickname is set as:', myName);
      return false;
    }
  }
});

(function connect() {
  function reconnect() {
    if (retriedTimes >= maxRetries) {
      throw new Error('Maximum retries reached. Please check your network.');
    } else {
      retriedTimes += 1;
      setTimeout(connect, retryTimeout);
    }
  }

  client = net.connect({ port: definePort }, function() {
    client.setEncoding('utf8');
    retriedTimes = 0;
    console.log('Connected to server.');
    console.log('Type "quit" to exit the chat mode.');
    console.log('Please enter your nickname:');
  }).on('data', function(data) {
    console.log(data.toString());
  }).on('end', function() {
    console.log('Disconnected from server!');
  }).on('error', function(err) {
    console.log('Client error:', err.message);
  }).on('close', function() {
    if (!quitting) {
      console.log('Connection closed, will try to reconnect.');
      reconnect();
    }
  });

  client.pipe(process.stdout, { end: false });
}());

问题及解决方案

1. 显示格式问题：

   • 可以通过在 console.log 中添加更多的格式化信息来改善显示格式。

2. 中文乱码问题：

   • 确保客户端和服务端都设置了正确的编码（如 UTF-8）。在本示例中，我们已经设置了 setEncoding('utf8')，所以应该不会有乱码问题。如果仍然存在乱码，可以尝试检查系统环境变量是否正确设置为 UTF-8 编码。

希望这些代码和解释能帮助你更好地理解如何使用 Node.js 实现一个简单的聊天程序。如果有任何问题或需要进一步的帮助，请随时提问！