今天我來實現一個簡單的聊天室,後台用nodejs, 客戶端與服務端通信用socket.io,這是一個比較成熟的websocket框架.
初始工作
1.安裝express, 用這個來託管socket.io,以及靜態頁面,命令npm install express --save,--save可以使包添加到package.json文件裡.
2.安裝socket.io,命令npm install socket.io --save.
編寫服務端代碼
首先我們通過express來託管網站,並附加到socket.io實例裡,因為socket.io初次連接需要http協議
複製代碼代碼如下:
var express = require('express'),
io = require('socket.io');
var app = express();
app.use(express.static(__dirname));
var server = app.listen(8888);
var ws = io.listen(server);
添加服務器連接事件,當客戶端連接成功之後,發公告告訴所有在線用戶,並且,當用戶發送消息時,發廣播通知其它用戶.
複製代碼代碼如下:
ws.on('connection', function(client){
console.log('/033[96msomeone is connect/033[39m /n');
client.on('join', function(msg){
// 檢查是否有重複
if(checkNickname(msg)){
client.emit('nickname', '暱稱有重複!');
}else{
client.nickname = msg;
ws.sockets.emit('announcement', '系統', msg + ' 加入了聊天室!');
}
});
// 監聽發送消息
client.on('send.message', function(msg){
client.broadcast.emit('send.message',client.nickname, msg);
});
// 斷開連接時,通知其它用戶
client.on('disconnect', function(){
if(client.nickname){
client.broadcast.emit('send.message','系統', client.nickname + '離開聊天室!');
}
})
})
由於客戶端是通過暱稱來標識的,所以服務端需要一個檢測暱稱重複的函數
複製代碼代碼如下:
// 檢查暱稱是否重複
var checkNickname = function(name){
for(var k in ws.sockets.sockets){
if(ws.sockets.sockets.hasOwnProperty(k)){
if(ws.sockets.sockets[k] && ws.sockets.sockets[k].nickname == name){
return true;
}
}
}
return false;
}
編寫客服端代碼
由於服務端採用第三方websokcet框架,所以前端頁面需要單獨引用socket.io客戶端代碼,源文件可以從socket.io模塊裡找,windows下路徑為node_modules/socket.io/node_modules/socket.io-client/dist,這裡有開發版和壓縮版的,默認引用開發版就行.
前端主要處理輸入暱稱檢查,消息處理,完整代碼如下:
複製代碼代碼如下:
<!DOCTYPE html>
<html>
<head>
<title>socket.io 聊天室例子</title>
<meta charset="utf-8">
<link rel="stylesheet" href="css/reset.css"/>
<link rel="stylesheet" href="css/bootstrap.css"/>
<link rel="stylesheet" href="css/app.css"/>
</head>
<body>
<div>
<div id="chat">
<ul id="chat_conatiner">
</ul>
</div>
<div>
<textarea ></textarea>
<button id="clear">清屏</button>
<button id="send">發送</button>
</div>
</div>
<script type="text/javascript" src="js/socket.io.js"></script>
<script type="text/javascript">
var ws = io.connect('http://172.16.2.184:8888');
var sendMsg = function(msg){
ws.emit('send.message', msg);
}
var addMessage = function(from, msg){
var li = document.createElement('li');
li.innerHTML = '<span>' + from + '</span>' + ' : ' + msg;
document.querySelector('#chat_conatiner').appendChild(li);
// 設置內容區的滾動條到底部
document.querySelector('#chat').scrollTop = document.querySelector('#chat').scrollHeight;
// 並設置焦點
document.querySelector('textarea').focus();
}
var send = function(){
var ele_msg = document.querySelector('textarea');
var msg = ele_msg.value.replace('/r/n', '').trim();
console.log(msg);
if(!msg) return;
sendMsg(msg);
// 添加消息到自己的內容區
addMessage('你', msg);
ele_msg.value = '';
}
ws.on('connect', function(){
var nickname = window.prompt('輸入你的暱稱!');
while(!nickname){
nickname = window.prompt('暱稱不能為空,請重新輸入!')
}
ws.emit('join', nickname);
});
// 暱稱有重複
ws.on('nickname', function(){
var nickname = window.prompt('暱稱有重複,請重新輸入!');
while(!nickname){
nickname = window.prompt('暱稱不能為空,請重新輸入!')
}
ws.emit('join', nickname);
});
ws.on('send.message', function(from, msg){
addMessage(from, msg);
});
ws.on('announcement', function(from, msg){
addMessage(from, msg);
});
document.querySelector('textarea').addEventListener('keypress', function(event){
if(event.which == 13){
send();
}
});
document.querySelector('textarea').addEventListener('keydown', function(event){
if(event.which == 13){
send();
}
});
document.querySelector('#send').addEventListener('click', function(){
send();
});
document.querySelector('#clear').addEventListener('click', function(){
document.querySelector('#chat_conatiner').innerHTML = '';
});
</script>
</body>
</html>
這裡提供完整的代碼壓縮文件
總結
nodejs是一個好東西,尤其是在處理消息通訊,網絡編程方面,天生的異步IO.