websocket搭建的聊天室

在前后端数据交互的时候我们经常使用的是ajax，用的是传统的http协议，而http协议有个致命的缺点，就是请求一结束，连接就断开了，我们为了保持这个链接的，通常会使用cookie，而自从h5出现websocket之后，妈妈再也不用担心我的请求了，可以愉快的玩耍了，websocket是TCP协议，也就是所谓的“长连接”，可以一直保持，客户端与服务端的连接，交互起来自然是so easy ,to happy,这两天利用工作之余，写了一个基于nodejs和websocket的聊天室，只为说明原理，没有做任何的样式，不喜勿喷

这是客户端：

<!DOCTYPE html>
<html>
	<head>
		<meta charset="UTF-8">
		<title></title>
	</head>
	<body>
		<input type="text" id="input"/>
		<input type="button" value="发送" id="btn"/>
	</body>
	<script>
		var websocket = new WebSocket("ws://localhost:3000/");
		
		function showMessage(type,str){
			var div = document.createElement('div');
			div.innerHTML = str;
			
			if(type == 'enter'){
				div.style.color = 'blue';
			}else if(type == 'leave'){
				div.style.color = 'red';
			}
			document.body.appendChild(div);
			
		}
		websocket.onopen = function(){			//连接成功后的回调
			document.getElementById('btn').onclick = function(){
				var txt = document.getElementById('input').value;
				if(txt){
					websocket.send(txt);
				}
			}
		}
		websocket.onclose = function(){		//关闭连接的回调
											
		}
		websocket.onmessage = function(e){		//发送消息的回掉
			var msg = JSON.parse(e.data);
			showMessage(msg.type,msg.data);
		}
		
	</script>
</html>

　这是服务端：

var ws = require("nodejs-websocket")    //导入nodejs-websocket模块

var PORT = 3000;
var clientCount = 0;
// Scream server example: "hi" -> "HI!!!"
var server = ws.createServer(function (conn) {        //建立连接时
    console.log("New connection")
    clientCount++;
    conn.nickname = 'user'+clientCount;                //每一个用户名
    var msg = {};
    msg.type = "enter";        //消息的类型，
    msg.data = conn.nickname + '  进入了聊天室';    
    msg = JSON.stringify(msg);    //数据格式化
    broadcast(msg);        //用于广播数据
    
    console.log(msg);
    conn.on("text", function (str) {
        console.log("Received "+str)
        var msg = {};
        msg.type = 'message';
        msg.data = conn.nickname + '：' + str;
        msg = JSON.stringify(msg);
        broadcast(msg);
    })
    conn.on("close", function (code, reason) {        //链接关闭的回调，注意这个一定要写，否则在关闭页面的时候，服务会抛异常；
        console.log("Connection closed")
        var msg = {};
        msg.type = 'leave';
        msg.data = conn.nickname +'离开了';
        msg = JSON.stringify(msg);
        broadcast(msg);
        
    })
    conn.on("error", function (err) {    //出错时候的回调；
        console.log(err);
    })
}).listen(PORT);

function broadcast(str){
    server.connections.forEach(function(connection){    //遍历所有的链接
        connection.sendText(str);                        //给每一个链接发送数据
    });
}

有说的不对的欢迎大家吐槽，大家相互学习。