Nodejs聊天室及静态资源服务器的实现

---

写在最前

测试环境

➜  ~ node -v                               
v14.16.0
➜  ~ npm -v               
7.6.3

需要mime包

mime可以识别文件类型，方便网页展示。

安装

• 终端输入npm install mime

如果下载很慢，可以换npm阿里源

• 终端输入npm config set registry https://registry.npm.taobao.org/

---

多人聊天服务器

./lib/chat_server.js

var net = require('net');
var clientList={}; //此处用字典记录所有客户端

var server = net.createServer(function(socket){
    var userid = '';
    // 获取id信息，并记录到字典
    socket.once('data',function(data){
        userid = data.slice(0,3).toString();
        clientList[userid]=socket;
    }); 
	// 监听客户端的消息，并广播给所有人
    socket.on('data',function(data){
        broadcast(data);
    })
	// 将退出聊天的客户端从字典中删除
    socket.on('end',function(){
        delete clientList[userid];
        console.log(userid+"退出聊天");     
    });
});
// 	广播
function broadcast(data){
    console.log(data.toString());
    for(var key in clientList){
        clientList[key].write(data);
    }
}
// server.listen(PORT);
// 此处为方便server.js调用，若单独使用可用⬆️代替⬇️
exports.listen = function(T){
    server.listen(T);
}
});

多人聊天客户端

./lib/chat_client.js

const net = require('net');

// 接收三个参数，依次为：主机名、端口号、用户名。
var hostname = process.argv[2];
var port = process.argv[3];
var user = process.argv[4];

const client = net.createConnection({host: hostname,port: port}, (socket) => {
  	// 通过自定义hash生成唯一的用户id
    var ID = hashUser(user,1000);
    // 通知服务器，同时确保将ID顺利传达。
    client.write(ID+'('+user+"): 加入聊天");

	// 接收键盘输入
    process.stdin.setEncoding('utf8');
    process.stdin.on('readable',() => {
      var chunk = process.stdin.read();
      chunk = chunk.slice(0,-1);
      // 输入886退出聊天
      if(chunk === '886'){
        process.stdin.emit('end');
        client.emit('end');
      }else if( chunk !== null){
        client.write(ID+'('+user+"): "+chunk);
      }

      // 在“旧”流模式下，默认情况下标准输入流已暂停，所以需要resume
      process.stdin.resume();
    });
});

// 自定义hash函数，用于生成唯一id标识
function hashUser(str,idsize){
  var hashCode = 999;
  for(var i=0;i<str.length;i++){
    hashCode=37* hashCode + str.charCodeAt(i) //获取编码
  }
  return hashCode%idsize;
}

// 接收并在终端打印广播（聊天消息）
client.on('data', (data) => {
    console.log(data.toString());
});

// 
client.on('end', () => {
    console.log('断开连接');
});

静态资源服务器

./server.js

var http = require('http');
var fs = require('fs');
var path = require('path');
var mime = require('mime');
var cache = {}; // 媒体缓存，避免多次加载磁盘导致访问缓慢

// 各种状态码总结
// 2开头 成功
// 4开头 请求的资源 （403无权限）
// 5开头 服务器问题
// 3开头 以前有，现在🈚️了
function send404(response){
    response.writeHead(404,{'Content-Type': 'text/plain'});
    response.write('Error 404: resource not found');
    response.end();
}

// 发送文件
function sendFile(response, filePath, fileContents){
	// mime可以识别文件类型，方便网页展示。
    response.writeHead(200,{"content-type": mime.lookup(path.basename(filePath))});
    response.end(fileContents);
}

// 静态资源服务器，若cache缓存中是否已经存在则直接send
function serveStatic(response, cache, absPath){
    if(cache[absPath]){
        sendFile(response,absPath,cache[absPath]);
    }else{
        fs.exists(absPath,function(exists){
            if(exists){
                fs.readFile(absPath,function(err,data){
                    if(err){
                        send404(response);
                    }else{
                        cache[absPath]=data;
                        sendFile(response, absPath, data);
                    }
                })
            }else{
                send404(response);
            }
        })
    }
}

// 创建服务器
var server = http.createServer(function(request, response){
    var filePath = false;
	// 默认访问index.html
    if(request.url == '/'){
        filePath = 'public/index.html';
    }else{
        filePath = 'public' + request.url;
    }

    var absPath = './' + filePath;
    serveStatic(response, cache, absPath);
});

// 监听7301端口
server.listen(7301,function(){
    console.log("server listening on port 7301");
})

// 导入前述的 聊天服务器
var chatServer = require('./lib/chat_server');
chatServer.listen(7302,"127.0.0.1");

—分割线—

核心逻辑可以简化成以下代码⬇️

---

简化版server

const net = require('net');
var clientSet = new Set();
const server = net.createServer((socket) => {
    clientSet.add(socket);
    socket.on('data', (data) => {
        broadcast(data);
    })
    socket.on('end', function() {
        clientSet.delete(socket);
    });
});
function broadcast(data) {
    console.log(data.toString());
    clientSet.forEach((set) => {
        set.write(data);
    });
}
server.listen(10000, "127.0.0.1");

简化版client

const net = require('net');
const config = {
    host: "127.0.0.1",
    port: 10000
}
const client = net.createConnection(config, () => {
    process.stdin.setEncoding('utf8');
    process.stdin.on('readable', () => {
        var chunk = process.stdin.read();
        if (chunk !== null)
            client.write(chunk);
        process.stdin.resume();
    });
});
client.on('data', (data) => {
    console.log(data.toString());
});
client.on('end', () => {
    console.log('disconnected from server');
});