Node.js之Websocket技术
我第一次听说websocket之时，HTML5标准尚未公布，当时只有少数前瞻性浏览器携带了这个API。
我对websocket最大的印象是，他可以解决我对“在线聊天系统”开发的疑惑（我一直想不通http如何保持长连接），这样我们无需使用轮询ajax和php无限循环去模拟，还记得2014年初那会我写了一个在线聊天室，那时候我真的应该用websocket技术的，php无限轮询的方式，只要3-4个人在线就可以让linux+apache服务器崩溃。
很可惜，我没有在php上使用过Websocket API，今天是我第一次将这个技术用来实践，使用node.js实现
引入模块
package.json
{ ???"name":"chat-websocket", ???"version":"0.0.1", ???"description":"use websocket to create a char server", ???"dependencies":{ ???????"express":"latest", ???????"express-ws":"latest" ???}}在引入http必备的express框架后，再引入基于express的中间件——express-ws
express-ws是express上的websocket中间件，为express提供了websocket请求处理功能
另外注意：《了不起的node.js》一书中使用的是websocket.io模块，两个东西原理，方法都差不多，我个人觉得io那个老了点，我选择了express-ws模块！主要是因为其与express配合效果更佳，专门为express而设计的中间件，何乐而不用呢？
做个测试
先来看看，node中使用express-ws的基本套路：
index.js
var express = require("express");//创建express下的http服务var app = express();//关联express-ws中间件var expressWs = require("express-ws")(app);//express托管静态文件app.use(express.static(__dirname+‘/views‘,{‘index‘:‘index.html‘}));app.ws(‘/ws‘,function(ws,req){ ???ws.on(‘message‘,function(msg){ ???????console.log(msg); ???})});;app.listen(80);这个index.js是服务端代码
解析：
- 先引入express模块，并使用express()方法获得app对象
- 引入express-ws并构造对象，构造函数传入的是app对象，意思大概是绑定到express服务器上
- 通常接受http请求使用的是app.use()，而websocket请求则使用app.ws()，该方法就是中间件扩展的
- message事件监听客户端发送的数据，并打印到终端上
views/index.html
<!DOCTYPE html><html lang="en"><head> ???<meta charset="UTF-8"> ???<title>测试</title></head><body> ???<input type="text" value="" id="msg"> ???<button id="submit">提交</button></body><script> ???window.onload=function(){ ???????var ws = new WebSocket(‘ws://localhost/ws‘) ???????var sub = document.getElementById("submit"); ???????var msg = document.getElementById("msg") ???????sub.addEventListener("click",function(){ ???????????ws.send(msg.value); ???????}); ???????ws.addEventListener("open",function(){ ???????????alert("WebSocket has been opened!"); ???????}) ???}</script></html>这些为前端代码：实现了一个简单输入框，输入后主动通过websocket发送给服务端，服务端那边会打印再console中
效果：
测试成功，每一次点击提交都会显示！
WebSocket开发在线聊天室
功能点
- 用户进入输入用户名可开始聊天
- 登入后提示当前在线用户
- 聊天内容会广播给聊天室其他在线用户
- 用户进入后提示xxx用户进入聊天室
- 用户关闭后提示xxx用户退出聊天室
package.json同上不变，具体设计逻辑参考另一篇博客，我用TCP API实现的聊天室程序http://www.cnblogs.com/devilyouwei/p/8423961.html
前端：index.html
<!DOCTYPE html><html lang="en"><head> ???<meta charset="UTF-8"> ???<title>测试</title> ???<style> ???????#inchat{ ???????????display:none; ???????} ???????ul{ ???????????list-style-type:none; ???????} ???</style></head><body> ???<div id="inputname"> ???????<input type="text" value="" id="msg" placeholder="输入一个聊天昵称"> ???????<button id="submit">提交</button> ???</div> ???<div id="inchat"> ???????<ul id="chat-content"> ???????</ul> ???????<textarea id="sendContent" value=""> ???????</textarea> ???????<button id="send">发送</button> ???</div></body><script> ???window.onload=function(){ ???????var ws = new WebSocket(‘ws://localhost/ws‘);//需要修改为相应地址 ???????var sub = document.getElementById("submit"); ???????var msg = document.getElementById("msg") ???????var ul ?= document.getElementById("chat-content"); ???????var send = document.getElementById("send"); ???????var sendContent = document.getElementById("sendContent"); ???????sendContent.value=""; ???????sub.addEventListener("click",function(){ ???????????ws.send(msg.value); ???????}); ???????send.addEventListener("click",function(){ ???????????ws.send(sendContent.value); ???????????sendContent.value="" ???????}); ???????ws.addEventListener("open",function(){ ???????????alert("WebSocket has been opened!"); ???????}) ???????ws.addEventListener("message",function(e){ ???????????var res = JSON.parse(e.data); ???????????//如果正在聊天中 ???????????if(res.ischat){ ???????????????var li = document.createElement("li"); ???????????????li.innerText = res.info; ???????????????ul.appendChild(li); ???????????}else{ ???????????????if(res.status == 1){ ???????????????????document.getElementById("inputname").style.display="none"; ???????????????????document.getElementById("inchat").style.display="block"; ???????????????}else{ ???????????????????alert(res.info); ???????????????} ???????????} ???????}); ???}</script></html>注意：放到公网访问需要把localhost改成ip或者网址！
后端：index.js
var express = require("express");//创建express下的http服务var app = express();//关联express-ws中间件var expressWs = require("express-ws")(app);var users = {};var count = 0;//express托管静态文件app.use(express.static(__dirname+‘/views‘,{‘index‘:‘index.html‘}));//用ws方法而不是use方法app.ws(‘/ws‘,function(ws,req){ ???var username = null; ???ws.on(‘message‘,function(msg){ ???????if(!username){ ???????????if(users[msg]){ ???????????????ws.send(JSON.stringify({status:0,info:"用户名重复请重试",ischat:false})); ???????????}else if(msg == ""){ ???????????????ws.send(JSON.stringify({status:0,info:"用户名不能为空",ischat:false})); ???????????}else{ ???????????????username = msg; ???????????????count++; ??//用户+1 ???????????????users[msg] = ws; ???????????????ws.send(JSON.stringify({status:1,info:"注册成功，欢迎"+username,ischat:false})); ???????????????console.log(username+"用户加入聊天室！当前在线："+count); ???????????????broadcast(username+"用户加入聊天室！当前在线："+count); ???????????} ???????}else{ ???????????broadcast(username+"："+msg); ???????????console.log(username+"："+msg); ???????} ???}); ???ws.on(‘close‘,function(){ ???????delete users[username]; ???????count--; ???????console.log(username+"用户退出聊天室！当前在线："+count); ???????broadcast(username+"用户退出聊天室！当前在线："+count); ???})});;app.listen(80);//需要广播给所有人（不排除自己）function broadcast(msg){ ???for(var i in users) ???????users[i].send(JSON.stringify({status:1,info:msg,ischat:true}));}注意1：express-ws中的ws.send()方法只能发送字符串，并没有express http的res.send()那么强大，故而我在传入js对象时，手动使用JSON.stringify()将对象转换为json字符串，待前端收到后再使用JSON.parse()转换回js对象。
注意2：WebSocket API中几个最重要的事件：open，close，message，error，对应了连接过程中的打开连接，关闭连接，消息传递，错误事件，无论前端还是后端都需要对这几个事件进行绑定监听，传入回掉函数做必要的处理
注意3：设计逻辑再讲一遍：users变量存储每一个连接引用，username作为局部变量，再每一次连接域内部，每一个客户端连入都会创建一个，单独的broadcast()方法遍历所有socket连接发送消息，最后再提醒一遍node.js开发一定要特别注意作用域范围！
效果