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個客戶端的測試結果: