node.js項目如何創(chuàng)建websocket模塊

更新時間：2024年09月06日 09:17:46 作者：-風過無痕

這篇文章主要介紹了node.js項目如何創(chuàng)建websocket模塊問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

前言

node.js是沒有提供內(nèi)置的websocket模塊，下載第三方包ws模塊來完成
注意的是這個模塊好像在斷開連接這一塊是有點問題，需要自己補充代碼
通過內(nèi)置url模塊解決路徑參數(shù)-邏輯代碼解決斷開連接問題-方便擴展業(yè)務需求
可以不用-但不能沒有這些

代碼實現(xiàn)

1.下包

npm install ws

2.參考代碼

// 第三方包ws
const WebSocket = require("ws");

// 內(nèi)置url模塊-處理websocket路徑參數(shù)
const url = require("url");

// 注釋
// 前端使用-參考主頁文章uni-app使用websocket
// sendMessage方法是前端的
// userType代表類型（5-斷開連接/1-發(fā)送消息等等-自定義）
// 斷開連接數(shù)據(jù)格式
// sendMessage(JSON.stringify({
//   userType:'5',
//   moblieType:'用戶id'
// }))


// 創(chuàng)建WebSocket服務器，監(jiān)聽端口9000
const wss = new WebSocket.Server({ port: 9000 });

// 假設ip是 - 192.168.2.22
// 此時websocket前端連接地址就是
// ws://101.43.100.203:3010:9000?userId=用戶id


// 存儲所有連接
// 方便斷開連接
var connections = new Set();

wss.on("connection", function connection(ws, req) {

  // 使用url模塊-接收前端發(fā)來的userId（路徑參數(shù)）
  const parameters = url.parse(req.url, true).query;
  const userId = parameters.userId;
  
  // 用戶id+ws實例存入
  connections.add({
    id: userId,
    Instantiation: ws,
  });


  // 請求頭信息-如果是uni-app（APP端-傳遞header-可以打?。? 參考uni-app文檔websocket
  // console.log("請求頭信息", req.headers);

  console.log("客戶端連接成功！");

  // 接收來自客戶端的消息
  ws.on("message", function incoming(getMessage) {
    console.log("客戶端: %s", JSON.parse(getMessage));
    
    // 傳遞字符串文字-自取
    // console.log("客戶端: %s", getMessage);
    // const clientMsg = Buffer.from(getMessage.msg, "hex");

    let data = JSON.parse(getMessage);

    // 斷開websocket連接-參考上方注釋數(shù)據(jù)格式
    if (data.userType == 5) {
      console.log("用戶id-斷開連接", userId);

      for (const connection of connections) {
        if (connection.id == data.moblieType) {
          
          // 關(guān)閉刪除連接池某個websocket連接
          connection.Instantiation.close();
          connections.delete(connection);
        }
      }

      return;
    }
  });
});