Nodejs之TCP服务端与客户端聊天程序详解
TCP是用来计算机之间进行通信的,通过编写客户端和服务端聊天的代码,对于服务器与客户端的工作步骤有了深刻的了解,在这里根据了不起的Node.js一书来总结一下。
TCP聊天程序需要用到Nodejs中的net模块,net模块是对TCP的封装,用于创建TCP服务端与客户端的。
服务器端
count:连接的客户端个数;
users:用于存储客户端用户昵称的对象,对象值都为该客户端的Socket,Socket是一个接口,用于服务端与客户端通信。
net.createServer:是用来创建TCP服务器,传递的函数为connetion事件的监听器,connection事件在有客户端连接上服务器后触发,传进去的参数conn是一个socket,即net.Socket的实例,是既可读也可写的。
conn:是回调函数的参数,每一个客户端对应的conn都是不一样的,服务器会自动识别。
conn.write:是将数据发送到客户端上。
nickname:是额户端的昵称,当nickname为undefined时,代表客户端还没有输入昵称,需要输入昵称验证,若不为空则已经存储,代表输入的数据是对话,需显示到其他的客户端上。
conn.on('data',function (data){…}):是用来接收客户端传来的数据。
/*server.js*/ var net=require('net'); var count=0; var users={}; var server=net.createServer(function (conn) { conn.setEncoding('utf8'); conn.write(count+' people are connected' +'\nplease write your nickname:'); count++; var nickname; conn.on('data',function (data) { data=data.replace(/\r\n/,''); if(!nickname){ if(users[data]){ conn.write('\n nickname already used,try again:'); return; }else { nickname=data; users[nickname]=conn;//每个客户端对应的conn是不一样的 for(var i in users){ users[i].write(nickname+' join the romm\n'); } } }else { conn.write('发送者nickname: '+nickname+'\n'); for(var i in users){ if(i!=nickname){ users[i].write(nickname+': '+data+'\n'); } } } }); conn.on('close',function () {//这里防止连接出错,使用close而非end console.log(nickname+' client disconnected\n'); count--; delete users[nickname]; }); conn.on('error',function (err) { console.log('Error in connection:'+ err+'\n'); }); }).listen(3000,function () { console.log('server is listenning on 3000'); });
客户端
process.stdin.resume():process.stdin流来接受用户的键盘输入,这个可读流初始化时处于暂停状态,调用流上的resume()方法来恢复流。
process.stdin.on('data', function(data){…}):是用来获取键盘的输入,并用client.write()将它发送到服务端。
stop:用来判断客户端是否是正常断开连接。
net.createConnection:是用来创建于服务端的连接,并且给connect事件自动添加监听器,监听器不接收参数。返回的是 'net.Socket'。
/*client.js*/ var net=require('net'); var stop = false; process.stdin.resume(); process.stdin.on('data', function(data){ if(data.toString().trim().toLowerCase()==='stop'){ stop=true; console.log('connection stop\n'); client.end(); }else{ client.write(data); } }); var client=net.createConnection(3000,function () { console.log('connected to server!\n'); }); client.on('data', function(data) { console.log(data.toString());//打印服务器发送的数据 }); client.on('error', function(err) { console.log('Error in connection:'+ err+'\n'); }); client.on('close', function() { if(! stop) { console.log('connection got closed'); } });
TCP聊天程序
这里是在git中运行的,连接了3个客户端的测试结果: