1、最基础的聊天室代码(有注释)
/**
* 1.创建一个服务器
* 2. 客户端可以连接服务器
* 3.客户端可以发言,然后广播给大家
* 4.客户端连接和退出后都要通知大家。
* 5.显示当前的在线人数
*/
let net = require('net'); // 利用TCP协议实现,所以引入net模块,用telnet充当客户端
let clients = {}; // 所有客户端的socket存在这个对象里
// 创建TCP服务器
let server = net.createServer(function (socket) {
// 获取当前人数,也就是TCP链接数
server.getConnections(function (err, count) {
socket.write(`weclome,there is ${count} users now,please input your username\r\n`);
})
let nickname; // 每个人的名字
setEncoding('utf8');// 编码设置为utf8
socket.on('data', function (data) { // 接收数据,socket是既是可写流也是可读流,所以有data事件
data = data.replace(/\r\n/, ''); // 将回车(windows回车是\r\n,linux是\n)替换
if (data == 'byebye') { // 如果输入的字符是byebye则关闭这个socket链接
socket.end();
} else {
if (nickname) { // 输入名字后,所有信息广播给其他人
broadcast(nickname, `${nickname}:${data}`);
} else { // 第一次需要输入名字,并将这个人的sokcet存入clients对象,后面要用来广播
nickname = data;
clients[nickname] = socket;
broadcast(nickname, `welcome ${nickname} joined us!`);
}
}
});
socket.on('end', function () { // 用户推出销毁socket,因为socket数量是有限的
clients[nickname].destroy()
delete clients[nickname]
broadcast(nickname, `welcome ${nickname} joined us!`);
});
}).listen(8088);
// 广播方法,将除了自己的socket外,全部用socket发送数据
function broadcast(nickname, msg) {
for (let key in clients) {
if (key != nickname) {
clients[key].write(msg + '\r\n');
}
}
}
复制代码
2、增加私聊功能
// 添加私聊方法
function sendTo(toUser, text, selfSocket){
let userSocekt = clients[toUser]
if(userSocekt){
userSocekt.write(`${nickname}: ${text}`)
}else{
selfSocket.write(`用户名不正确或者对方已经下线`)
}
}
// 同时创建net服务器改造一下,根据不同的需求,分类广播和私聊功能
let server = net.createServer(function (socket) {
// 获取当前人数,也就是TCP链接数
server.getConnections(function (err, count) {
socket.write(`weclome,there is ${count} users now,please input your username\r\n`);
})
let nickname; // 每个人的名字
setEncoding('utf8');// 编码设置为utf8
socket.on('data', function (data) { // 接收数据,socket是既是可写流也是可读流,所以有data事件
// 发送数据格式为b: 发送内容,b代表要广播, s: 用户名:内容,s代表要私聊
let type = data.slice(0) // 取出第一个字符知道要用私聊还是广播功能
switch(type){
case 'b':
let text = data.slice(2)
boardcast(text)
break;
case 's':
let values = data.split(':')
let toUser = values[1]
let text = values[2]
sendTo(toUser, text, socket)
break;
default:
socket.write('此命令不能识别, 请重新输入')
break;
}
});
socket.on('end', function () { // 用户推出销毁socket,因为socket数量是有限的
clients[nickname].destroy()
delete clients[nickname]
broadcast(nickname, `welcome ${nickname} joined us!`);
});
}).listen(8088);
复制代码