npm install ws

// 第三方包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服務(wù)器，監(jiān)聽端口9000
const wss = new WebSocket.Server({ port: 9000 });

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


// 存儲(chǔ)所有連接
// 方便斷開連接
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實(shí)例存入
  connections.add({
    id: userId,
    Instantiation: ws,
  });


  // 請(qǐng)求頭信息-如果是uni-app（APP端-傳遞header-可以打?。? 參考uni-app文檔websocket
  // console.log("請(qǐng)求頭信息", 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)閉刪除連接池某個(gè)websocket連接
          connection.Instantiation.close();
          connections.delete(connection);
        }
      }

      return;
    }
  });
});