如何编辑原生代码扩展功能？WebSocket发送消息日志实现咨询

你的思路是完全可行的，通过重写WebSocket实例的send方法来扩展功能，是原生环境下无需修改原有代码结构就能实现消息记录需求的最优方案。

优化后的实现代码 #

// 保存原生send方法的引用
const originalSend = socket.send;

// 重写send方法
socket.send = function(...args) {
    // 执行自定义逻辑：记录发送的消息
    console.log('WebSocket发送消息:', args);
    
    // 调用原生send方法，用apply绑定上下文确保this指向WebSocket实例
    const result = originalSend.apply(this, args);
    
    // 可选：保留原生方法的返回值，不影响原有逻辑
    return result;
};

关键细节说明 #

• 上下文绑定：不要用箭头函数重写send方法——箭头函数的this会继承外层作用域，无法指向WebSocket实例，使用普通function配合apply/call才能保证原生方法的上下文正确。
• 错误处理：如果需要捕获发送过程中的异常，可以添加try-catch块，避免影响原有业务逻辑：

  socket.send = function(...args) {
      console.log('WebSocket发送消息:', args);
      try {
          return originalSend.apply(this, args);
      } catch (error) {
          console.error('WebSocket发送失败:', error);
          throw error; // 抛出错误，不阻断原有错误处理流程
      }
  };
  

• 恢复原生方法：如果后续需要取消消息记录功能，直接恢复即可：

  socket.send = originalSend;
  

冲突规避方案（若重写send存在冲突） #

如果你的环境中有其他代码也重写了send方法，可以用Proxy包装整个WebSocket实例，更安全地拦截send调用：

const socketProxy = new Proxy(socket, {
    get(target, prop) {
        if (prop === 'send') {
            return function(...args) {
                console.log('WebSocket发送消息:', args);
                return target.send.apply(target, args);
            };
        }
        return Reflect.get(target, prop);
    }
});

// 后续使用socketProxy替代原socket实例即可
socketProxy.send('测试消息');

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