使用原生node写一个聊天室

在学习node的时候都会练习做一个聊天室的项目，主要使用socket.io模块和http模块。这里我们使用更加原始的方式去写一个在命令行聊天的聊天室。

http模块，socket.io都是高度封装之后的模块，我们使用更加原始的net模块来做。

socket

做聊天室，我们首先要了解一下socket，用百度百科上的定义：网络上的两个程序通过一个双向的通信连接实现数据的交换，这个连接的一端称为一个socket。socket对TCP/IP封装提供网络开发的接口，提供网络通信的接口。详细可以看看各种百科的资料。

服务端

1.基本接口介绍

创建服务端

const server = net.createServer((socket) => {
       //这里的毁掉函数中的参数就是一个socket
}
//监听端口和主机
server.listen({
    port:4433,
    host:'127.0.0.1'//要想别人访问到，要写服务启动所在机子的ip地址，默认localhost
}, () => {
    //这里是启动成功后的回调
    console.log('server bound')
})

服务创建成功之后，socket可以提供客户端（访问者）的一些信息，这里主要用的两个属性

socket.remoteAddress :客户端的ip地址
socket.remotePort:客户端访问的接口，这里是随机分配，注意这里的接口跟我们创建服务的接口不同，表示的意义也不一样建立网络通信连接至少要一对端口号，服务端监听的端口是为了跟服务端所在机子的其他服务区分，客户端访问这个端口的时候自然也要区分，分配不同端口

socket的event事件

//监听客户端发送过来的消息
socket.on('data', (data) => {
      //这里data是一个buffer要转化为字符串，然后去一下两端空格
        let receive = JSON.parse(data.toString().trim())；
    })
//给客户端发送消息
socket.write(string[,encoding]);

错误处理error

socket.on('error', (err) => {
      //在测试中发现，不去监听这个错误事件，当你客户端掉线之后服务端也会断开所以做一些错误处理
    })

2.聊天设计

聊天过程我们分三部分

1.登录

2.跟所有人聊

3.跟特定对象聊

首先我们确定聊天信息的发送格式，就像http里面有头部，内容等。我们发送的信息，应该包括发送者名字，给谁发以及发送的消息，设定为一下格式

const send = {
    name:"",
    message:"",
    to:""
}

服务端要根据给谁发做不同的处理

1.登录（login）：我们要给所有的人发信息，XXX登录了聊天室，还要记录这个访问的socket，以便于后续给特定的人发消息，格式{username：socket}

2.跟所有人聊天（server）：除了发消息本人要给其他人发送

3.跟某一个人聊天（client）:找到相应的socket发送消息

socket.on('data', (data) => {
        let receive = JSON.parse(data.toString().trim());
        switch(receive.to){
            case "login":
                DealInfo.login(receive, socket);
                break;
            case "server":
                DealInfo.server(receive, socket);
                break;
            default:
                DealInfo.client(receive, socket);
                break;
        }
    })

之前我们有监听socket的错误处理，也就是当用户掉线的时候，我们就可以把记录的socket删除掉，并提示当前用户数。这里可以从记录的socket个数求取，也可以根据以下方法

server.getConnections((err, count) => {
            if(err){
                throw err;
            }
            console.log(`${deleteKey}下线了 当前在线人数${count}`);
        })

客户端

1.基本接口介绍

创建socket连接

const client = net.connect({
            port:4433,
            host:'127.0.0.1'//默认localhost
        }, () => {
              //连接成功之后的回调
        })

client的事件

//监听服务端发送过来的信息
client.on('data', (data) => {
                //这里data是一个buffer
                let receive = JSON.parse(data.toString().trim());
            }).on('error', (err) => {
                //错误处理
            })
//给服务端发送消息
client.wirte(string[, encoding]);

2.聊天设计

基本设置

2.1用户名

这里我们使用readline这个模块

const readline = require('readline');
const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
});
rl.question('What is your name >', (name) => {
  //回调中的参数就所输入的信息，当输入非空的时候作为用户名，之后按我们的聊天格式输出预先样式
    //设立设定prompt的输出内容，只要调用rl.prompt()即可
    rl.setPrompt(`${name.trim()}>`);
    rl.prompt();
}

监听用户输入

 rl.on('line', (line) => {
                let stdinInfo = line.trim().split(':');
                if(stdinInfo.length == 2){
                    //跟某个用户聊天
                    send.to = stdinInfo[0];
                    send.message = stdinInfo[1];
                }else{
                    //跟所有人聊天
                    send.to = "server";
                    send.message = stdinInfo[0];
                }
                client.write(JSON.stringify(send));
                rl.prompt();
            })

详细代码地址：https://github.com/Stevenzwzhai/node-socket-chatroom