node.js学习(简单聊天室)
在掘金查看该文章
1. TCP服务搭建
1.1 socket
先来粗略了解下socket
套接字（socket）是一个抽象层，应用程序可以通过它发送或接收数据，可对其进行像对文件一样的打开、读写和关闭等操作。套接字允许应用程序将I/O插入到网络中，并与网络中的其他应用程序进行通信。网络套接字是IP地址与端口的组合。 (摘自百度百科)
socket用于在两个基于TCP/IP协议的应用程序之间相互通信.最早出现在UNIX系统中,是UNIX系统主要的信息传递方式.在windows系统中，socket称为winsock.
两种形式的socket：流式套接字,对应与TCP协议.
数据报套接字,对应与UDP协议.
2.创建TCP服务端
server.js(服务端)
const net = require("net");
const sever = net.createServer();
// const clients = [];
const users = [];
const types = require("./types");
sever.on("connection", clientSocket => {
console.log("有连接进来,请注意```");
// clients.push(clientSocket)
clientSocket.on("data", data => {
console.log("监听data事件,有人说:", data.toString());
data = JSON.parse(data.toString().trim());
switch (data.type) {
case types.login:
if (users.find(item => item.nickName === data.nickName)) {
return clientSocket.write(
JSON.stringify({
type: types.login,
success: false,
message: "昵称已存在"
})
);
}
clientSocket.nickName = data.nickName;
users.push(clientSocket);
clientSocket.write(
JSON.stringify({
type: types.login,
success: true,
message: "登录成功",
nickName:data.nickName,
sumUsers: users.length
})
);
users.forEach(user=>{
if(user!==clientSocket){
user.write(JSON.stringify({
type:types.log,
message:`${data,nickName} 进入聊天室,当前在线用户数${user.length}`
}))
}
})
break;
// 群聊天
case types.broadcast:
users.forEach(item => {
item.write(JSON.stringify({
type:types.broadcast,
message:data.message,
nickName:clientSocket.nickName
}))
})
break;
// 点对点
case types.p2p:
const user = users.find(item => item.nickName === data.nickName)
if(!user){
return clientSocket.write(JSON.stringify({
type:types.p2p,
success:false,
message:"该用户不存在"
}))
}
console.log('clientSocket.nickName',clientSocket.nickName)
user.write(JSON.stringify({
type:types.p2p,
message:data.message,
nickName:clientSocket.nickName,
success:true
}))
break;
default:
break;
}
});
// 离线
clientSocket.on("end",()=>{
console.log("有用户离线了~~~")
const index =users.findIndex(user => user.nickName === clientSocket.nickName)
if(index !== -1){
const offlineUser = users[inde]
users.splice(index,1)
users.forEach(user=>{
if(user!==clientSocket){
user.write(JSON.stringify({
type:types.log,
message:`${offlineUser,nickName} 离开了聊天室,当前在线用户数${user.length}`
}))
}
})
}
})
// clientSocket.write('hello,返回的是buffer,用tostring转一下哦')
});
sever.listen(2000, () => {
console.log("server running 127.0.0.1 2000");
});
3.创建客户端
client.js(客户端)
const net = require("net");
const types = require("./types");
let nickName = null;
const client = net.createConnection({
host: "127.0.0.1",
port: 2000
});
client.on("connect", () => {
console.log("连接成功了~~~");
process.stdout.write("请输入昵称:");
// 连接完毕后,可以监听终端的信息,发给服务端
process.stdin.on("data", data => {
data = data.toString().trim();
console.log("nickName", nickName);
if (!nickName) {
client.write(
JSON.stringify({
type: types.login,
nickName: data
})
);
}
const matches = /^@(\w+)\s(.+)$/.exec(data);
if (matches) {
//符合 @xxx xxx 格式
return client.write(
JSON.stringify({
type: types.p2p,
nickName: matches[1],
message: matches[2]
})
);
}
//群聊天
client.write(
JSON.stringify({
type: types.broadcast,
message: data
})
);
});
});
client.on("data", data => {
// console.log("服务端发来的data:::", data.toString());
data = JSON.parse(data.toString().trim());
switch (data.type) {
case types.login:
if (!data.success) {
console.log("登录失败", `${data.message}`);
process.stdout.write("请输入昵称");
} else {
process.stdout.write("登录成功,当前在线人数:", data.sumUsers);
nickName = data.nickName;
}
break;
case types.broadcast:
console.log(`${data.nickName}:${data.message}`);
break;
case types.p2p:
if (!data.success) {
return console.log(`发送失败:${data.message}`);
}
console.log(`${data.nickName}对你说:${data.message}`);
break;
case types.log:
console.log(`${data.message}`);
break;
default:
console.log("未知消息类型哦~");
break;
}
});
types.js
module.exports = {
login: 0,
broadcast: 1,
p2p: 2,
log: 3
};
总结
- 通过net模块建立TCP服务
- TCP必须建立连接(3次握手)后才能通信
- socket通信模型
- 和使用其他node模块(如koa)一样的思路,都是先建立服务(server),指定端口号