nodejs基于WS模块实现WebSocket聊天功能的方法

网络编程 2024/11/13 佚名

3 2 1

本文实例讲述了nodejs基于WS模块实现WebSocket聊天功能的方法。分享给大家供大家参考，具体如下：

WebSocket 模块众多，我选择了一个较为简单的模块进行了实现。

工具：Sublime

技术：Node.js 引用模块 ws

最终效果如下

本来打算制作网络画板，因为工作忘记了，这里就简单实现了群聊功能

没什么好介绍的，网上代码案例太多了，（各位新手主要碰见的问题，可能也就是 node 导入模块问题了）

介绍一下安装模块：

找到安装node 目录，进入到 npm 目录下运行 cmd 输入

npm install --save ws

源文件呢，最好放在 npm 目录下，就可以引用到了（我也是新手各位见谅）
然后执行 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, //监听接口
    verifyClient: socketVerify //可选，验证连接函数
  });
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(/^(:"blog.luojia.me") {
  //  return true; //如果是来自blog.luojia.me的连接，就接受
  //}
  // console.log("连接",origin[2]);
  return true; //否则拒绝
  //传入的info参数会包括这个连接的很多信息，你可以在此处使用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("在线人数", wss.clients.length);
  ws.send('你是第' + wss.clients.length + '位');
  // 发送消息
  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="/UploadFiles/2021-04-02/jquery-1.11.0.js">

聊天中没有做太多验证，名字可以随意改动，
参考
https://github.com/websockets/ws
https://github.com/websockets/ws/blob/master/doc/ws.md
希望本文所述对大家nodejs程序设计有所帮助。

nodejs,WS模块,WebSocket,聊天功能