首页 > 解决方案 > 使用 node.js 管理多个 Websocket 客户端连接

问题描述

我在节点中实现了一个 websocket 客户端,我的 webhook 试图处理来自聊天机器人服务的多个连接。例如:一个新用户收入,在外部聊天机器人服务上建立了一个 websocket 连接。websocket URL是通过我的代码中的XMLHttpRequest获取的。然后我使用这个 url 使用 ws 对象 ( ) 连接到聊天机器人服务new WebSocket('wssUrlObtainedThroughAjaxRequest','default-protocol')。所以每个用户都有一个WebSocket。问题是我的代码是按顺序运行的。因此,如果两个人向我的 webhook 节点发送消息,由于并行性,事情将无法正常工作。好吧,我将在这里发布示例代码以使其更好地理解。

const express = require('express');
const PORT = process.env.PORT || 8002;
let WebSocket = require('ws');
let CONNECTIONS = new Map();
...
...
...
const app = express()
    .use(bodyparser.urlencoded({extended: false}))
    .use(bodyparser.json());

app.post('/', (req, res) => { 
...
...
...
    } else if (req.body.type === 'MESSAGE') {

    let DM = req.body.space.name;
    let msg = req.body.message.text;

    ws = (CONNECTIONS.get(DM)!=null) ? CONNECTIONS.get(DM) : null;

    if(ws==null || ws.readyState==3){
      controlws.gerarURLWS();
      ws = new WebSocket(controlws.urlws, 'talk-protocol');
      CONNECTIONS.set(DM,ws);
    }

    // Executes on websocket openning
    ws.onopen = function (event) {
      console.log('Canal aberto;');
      keepAliveWS();
      ws.send(JSON.stringify(msgKoreAi(msg))); 
    }

    if(ws.readyState==1)
      ws.send(JSON.stringify(msgKoreAi(msg))); 

    ws.onmessage = async function (event) {
      let resp = JSON.parse(event.data);

      if (resp.type == "bot_response") {
        text = resp.message[0].component.payload.text;

        if(text==null){ // tem quick reply
          //mount card hangouts response 
          let qreplies = resp.message[0].component.payload.payload.quick_replies; 
          card = '{"sections": [{"widgets": [{"buttons": [';
          for(let i=0; i<qreplies.length; i++){
            if(i!=qreplies.length-1)
              card+='{"textButton": {"text": "'+qreplies[i].payload+'","onClick": {"action": {"actionMethodName": "'+qreplies[i].payload+'"}}}},';
            else
              card+='{"textButton": {"text": "'+qreplies[i].payload+'","onClick": {"action": {"actionMethodName": "'+qreplies[i].payload+'"}}}}';
          }      
          card+=']}]}],"name": "respostas"}';
          card = JSON.parse(card);

          text = resp.message[0].component.payload.payload.text; 
          {
            await assyncMessage(DM, text);
            await assyncMessage(DM, card);
          }
          return;
        }

        //Send assync messages if synchronous was already sent
        if(res.headersSent){
          {
            return await assyncMessage(DM, text);
          }
        }
        else
          return res.json({text});
      }
    }
    return;
  }
...
...
...
app.listen(PORT, () => {
  console.log(`Server is running in port - ${PORT}`);
});

标签: javascriptnode.jssessionwebsocket

解决方案


推荐阅读