Node.js环境下如何让ShareDB忽略WebSocket非标准操作消息？

这问题我之前帮人处理过，其实核心就是在ShareDB拿到WebSocket消息之前加一层拦截过滤，把非标准格式的消息拦下来就行。根据你用的WebSocket集成方式，给你两种具体的解决方案：

解决方案1：直接使用ws库时的自定义过滤 #

如果是自己用原生ws库搭建的WebSocket服务，你可以在消息传给ShareDB的Stream之前先做判断，只让符合ShareDB标准类型的消息通过：

const WebSocket = require('ws');
const ShareDB = require('sharedb');

// 初始化ShareDB实例
const db = new ShareDB();
// 创建WebSocket服务器
const wss = new WebSocket.Server({ port: 8080 });

wss.on('connection', (ws) => {
  // 创建ShareDB对应的WebSocket JSON流
  const shareDbStream = new ShareDB.WebsocketJSONStream(ws);

  // 拦截WebSocket的message事件，先过滤消息
  ws.on('message', (rawData) => {
    try {
      const message = JSON.parse(rawData);
      // 定义ShareDB支持的标准消息类型（可以根据官方文档补充）
      const validShareDbTypes = ['open', 'op', 'submit', 'query', 'subscribe', 'unsubscribe', 'close'];

      if (validShareDbTypes.includes(message.type)) {
        // 符合标准的消息，传给ShareDB处理
        shareDbStream.write(message);
      } else {
        // 处理自定义消息，比如你的`request-id`
        if (message.type === 'request-id') {
          // 这里可以给客户端返回响应，示例：
          ws.send(JSON.stringify({ type: 'id-response', id: `user-${Date.now()}` }));
        }
        // 其他非标准消息直接忽略，不传给ShareDB
      }
    } catch (err) {
      // 处理JSON解析失败的异常，避免服务崩溃
      console.error('Failed to parse WebSocket message:', err);
    }
  });

  // 将ShareDB的输出传回客户端
  shareDbStream.on('data', (data) => {
    ws.send(JSON.stringify(data));
  });

  // 处理Stream错误，避免服务挂掉
  shareDbStream.on('error', (err) => {
    console.error('ShareDB stream error:', err);
    ws.close();
  });
});

关键逻辑说明 #

• 我们直接接管了WebSocket的message事件，而不是让ShareDB的Stream直接绑定
• 先解析消息，判断type是否在ShareDB的标准类型列表里，只有符合条件的才传给ShareDB的Stream
• 自定义消息（比如request-id）可以在这里单独处理，不需要ShareDB介入

解决方案2：使用share-websocket包时的重写逻辑 #

如果你用的是share-websocket这个封装好的包，可以重写它的消息处理方法来实现过滤：

const ShareDB = require('sharedb');
const ShareWebSocket = require('share-websocket');
const WebSocket = require('ws');

const db = new ShareDB();
const wss = new WebSocket.Server({ port: 8080 });

wss.on('connection', (ws) => {
  const shareSocket = new ShareWebSocket(db, ws);
  
  // 保存原始的消息处理方法
  const originalHandleMessage = shareSocket.handleMessage;
  
  // 重写handleMessage方法，添加过滤逻辑
  shareSocket.handleMessage = (message) => {
    const validTypes = ['open', 'op', 'submit', 'query', 'subscribe', 'unsubscribe', 'close'];
    if (validTypes.includes(message.type)) {
      // 符合标准的消息，交给原始方法处理
      originalHandleMessage.call(shareSocket, message);
    } else {
      // 处理自定义消息
      if (message.type === 'request-id') {
        ws.send(JSON.stringify({ type: 'id-response', id: 'your-unique-id' }));
      }
      // 其他非标准消息直接忽略
    }
  };
});

注意事项 #

1. 更新标准类型列表：上面的validTypes是常见的ShareDB消息类型，你可以根据ShareDB官方文档补充遗漏的类型
2. 异常处理：一定要包裹JSON解析的try/catch，避免客户端发送非法JSON导致服务崩溃
3. 自定义消息扩展：如果之后有其他自定义消息类型，直接在过滤逻辑里添加对应的处理分支即可

这样处理后，ShareDB就只会收到符合它要求的消息，不会再因为request-id这类自定义消息报错了。

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