如何在JavaScript中限制WebSocket服务器的单事件入站消息数量？

首先纠正一个误区：你代码里的event.length始终是undefined，因为WebSocket的message事件回调里的event对象本身没有这个属性——每条消息都会单独触发一次message事件，你遇到的“单事件包含多条消息”其实是短时间内服务器连续推送了多条消息，导致message事件被多次触发，进而多次调用trigger()。

要实现按时间节点限制接收消息数量的需求，可以通过维护消息计数器和时间窗口的方式处理，以下是具体方案：

核心思路 #

• 预设时间节点与对应节点允许处理的消息数量
• 维护当前时间窗口的起始标记、允许处理数、已处理数三个状态变量
• 每次触发message事件时，先判断是否进入新的时间窗口，再根据计数器决定是否处理当前消息

代码实现 #

// 定义时间节点配置：键为时间字符串，值为对应允许的消息数量
const timeLimitConfig = {
  '16:00:00': 1,
  '16:00:12': 5,
  // 可按需添加更多时间节点
};

// 维护当前窗口状态
let currentTimeMarker = null;
let allowedMsgCount = 0;
let processedMsgCount = 0;

client.on('connect', async (connection) => {
  connection.on('message', async event => {
    if (event.type !== 'utf8') return;
    
    const now = new Date();
    // 格式化为HH:MM:SS的时间字符串
    const currentTimeStr = `${now.getHours().toString().padStart(2, '0')}:${now.getMinutes().toString().padStart(2, '0')}:${now.getSeconds().toString().padStart(2, '0')}`;
    
    // 检查是否触发了预设的时间节点，更新窗口状态
    if (timeLimitConfig[currentTimeStr]) {
      currentTimeMarker = currentTimeStr;
      allowedMsgCount = timeLimitConfig[currentTimeStr];
      processedMsgCount = 0;
    }
    
    // 当前窗口未达处理上限时，才执行消息处理逻辑
    if (processedMsgCount < allowedMsgCount) {
      const data = JSON.parse(event.utf8Data);
      await trigger(data);
      processedMsgCount++;
    } else {
      // 超过上限的消息可选择丢弃，或存入队列等待后续处理
      console.log(`时间窗口${currentTimeMarker}已达处理上限，丢弃消息:`, event.utf8Data);
    }
  })
})

async function trigger (data) {
  // ... 你的异步业务逻辑
}

补充说明 #

1. 如果你的限流规则是固定间隔周期（比如每12秒调整一次允许数量），而非固定时间点，可以把时间窗口改为基于时间戳的间隔计算，例如用Math.floor(Date.now() / 12000)划分12秒一个的窗口。
2. 若需避免消息丢失，超过上限的消息可存入缓存队列，等下一个时间窗口释放额度后再批量处理。
3. 由于trigger是异步函数，代码中用await确保了计数器在处理完成后才更新，避免并发场景下的计数错误。

内容的提问来源于stack exchange，提问作者aspirinemaga