nodejs基于WS模塊實現(xiàn)WebSocket聊天功能的方法

更新時間：2018年01月12日 11:47:43 作者：空白的畫

這篇文章主要介紹了nodejs基于WS模塊實現(xiàn)WebSocket聊天功能的方法,結(jié)合實例形式分析了nodejs使用WS模塊進行WebSocket通信實現(xiàn)聊天功能的具體操作技巧,需要的朋友可以參考下

本文實例講述了nodejs基于WS模塊實現(xiàn)WebSocket聊天功能的方法。分享給大家供大家參考，具體如下：

WebSocket 模塊眾多，我選擇了一個較為簡單的模塊進行了實現(xiàn)。

工具：Sublime

技術(shù)：Node.js 引用模塊 ws

最終效果如下

本來打算制作網(wǎng)絡(luò)畫板，因為工作忘記了，這里就簡單實現(xiàn)了群聊功能

沒什么好介紹的，網(wǎng)上代碼案例太多了，（各位新手主要碰見的問題，可能也就是 node 導(dǎo)入模塊問題了）

介紹一下安裝模塊：

找到安裝node 目錄，進入到 npm 目錄下運行 cmd 輸入

npm install --save ws

源文件呢，最好放在 npm 目錄下，就可以引用到了（我也是新手各位見諒）
然后執(zhí)行 node 源文件.js 就可以訪問了呢；

源碼：保存成js 文件就可以了

//https://github.com/websockets/ws/blob/master/doc/ws.md#new-wsserveroptions-callback
var WebSocketServer = require('ws').Server,
  wss = new WebSocketServer({
    port: 3000, //監(jiān)聽接口
    verifyClient: socketVerify //可選，驗證連接函數(shù)
  });
function socketVerify(info) {
  console.log(info.origin);
  console.log(info.req.t);
  console.log(info.secure);
  // console.log(info.origin);
  // var origin = info.origin.match(/^(:?.+\:\/\/)([^\/]+)/);
  //if (origin.length >= 3 && origin[2] == "blog.luojia.me") {
  //  return true; //如果是來自blog.luojia.me的連接，就接受
  //}
  // console.log("連接",origin[2]);
  return true; //否則拒絕
  //傳入的info參數(shù)會包括這個連接的很多信息，你可以在此處使用console.log(info)來查看和選擇如何驗證連接
}
//廣播
wss.broadcast = function broadcast(s,ws) {
  // console.log(ws);
  // debugger;
  wss.clients.forEach(function each(client) {
    // if (typeof client.user != "undefined") {
      if(s == 1){
        client.send(ws.name + ":" + ws.msg);
      }
      if(s == 0){
        client.send(ws + "退出聊天室");
      }
    // }
  });
};
// 初始化
wss.on('connection', function(ws) {
  // console.log(ws.clients.session);
  // console.log("在線人數(shù)", wss.clients.length);
  ws.send('你是第' + wss.clients.length + '位');
  // 發(fā)送消息
  ws.on('message', function(jsonStr,flags) {
    var obj = eval('(' + jsonStr + ')');
    // console.log(obj);
    this.user = obj;
    if (typeof this.user.msg != "undefined") {
      wss.broadcast(1,obj);
    }
  });
  // 退出聊天
  ws.on('close', function(close) {
    try{
      wss.broadcast(0,this.user.name);
    }catch(e){
      console.log('刷新頁面了');
    }
  });
});

html 前端源碼：

<!DOCTYPE html>
<html>
<head>
  <meta charset="utf-8">
  <meta http-equiv="X-UA-Compatible" content="IE=edge">
  <title>聊天</title>
  <link rel="stylesheet" href="">
  <script language="JavaScript" src="http://code.jquery.com/jquery-1.11.0.js"></script>
  <script type="text/javascript">
  </script>
</head>
<style type="text/css" media="screen">
div {
  border: 1px solid #cccccc;
  width: 500px;
  min-height: 100px;
}
</style>
<body>
  <div id="show">
  </div>
  <input type="text" id="message" name="" value="" placeholder="">
  <input type="text" id="name" value="" placeholder="昵稱;">
  <a href="javascript:void(0)" rel="external nofollow" rel="external nofollow" onclick="send()">發(fā)送</a>
  <a href="javascript:void(0)" rel="external nofollow" rel="external nofollow" onclick="exit()">退出</a>
  <!-- <input type="text" id="token" name="" value="" placeholder="" readonly="false"> -->
</body>
<script type="text/javascript">
var ws = new WebSocket("ws://127.0.0.1:3000?t=test");
ws.onopen = function() {
  console.log("連接狀態(tài)", ws);
  $("#show").html("連接狀態(tài);" + ws.readyState + "</br>");
  console.log("open");
  ws.open("start");
};
ws.onmessage = function(evt) {
  // console.log(evt.data)
  // alert(evt.data);
  $("#show").append(evt.data + "</br>");
};
ws.onclose = function(evt) {
  console.log("WebSocketClosed!");
  console.log(evt);
};
ws.onerror = function(evt) {
  console.log("WebSocketError!");
};
function send() {
  var msg = $("#message").val();
  var key = $("#token").val();
  var name = $("#name").val();
  var str = "{name:'" + name + "',msg:'" + msg + "',key:'" + key + "'}";
  console.log("發(fā)送", str);
  ws.send(str);
};
function exit() {
  var r = ws.close();
  console.log("退出", r);
}
</script>
</html>

聊天中沒有做太多驗證，名字可以隨意改動，

參考

https://github.com/websockets/ws

https://github.com/websockets/ws/blob/master/doc/ws.md

希望本文所述對大家nodejs程序設(shè)計有所幫助。

您可能感興趣的文章: