electron websocket 封装

Electron 是一个基于 Chromium 和 Node.js 的框架，用于构建跨平台的桌面应用程序。它支持使用 WebSocket 进行实时通信，而 WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，可以实现实时通信、推送和广播等功能。

以下是 Electron 中使用 WebSocket 的简单封装示例：

const WebSocket = require('ws');

class WebSocketClient {
  constructor(url) {
    this.url = url;
    this.socket = null;
  }

  connect() {
    this.socket = new WebSocket(this.url);

    this.socket.on('open', () => {
      console.log('connected');
    });

    this.socket.on('message', (data) => {
      console.log('received:', data);
    });

    this.socket.on('close', () => {
      console.log('disconnected');
    });
  }

  send(data) {
    if (this.socket.readyState === WebSocket.OPEN) {
      this.socket.send(data);
    }
  }

  close() {
    this.socket.close();
  }
}

// usage
const client = new WebSocketClient('ws://localhost:8080');
client.connect();

client.send('hello');

// when you want to close the connection
client.close();

在这个示例中，我们创建了一个 `WebSocketClient` 类来封装 WebSocket 的操作。它包含了 `connect()`、`send()` 和 `close()` 方法，分别用于连接、发送数据和关闭连接。在 `connect()` 方法中，我们使用 `new WebSocket(url)` 创建了一个 WebSocket 连接，并通过 `on()` 方法注册了连接打开、收到消息和关闭连接时的回调函数。

需要注意的是，在 Electron 中使用 WebSocket 时，需要在主进程和渲染进程中都引入 `ws` 模块，因为它使用了 Node.js 的原生模块，而渲染进程中默认没有这些模块。可以通过在渲染进程中使用 `preload` 脚本或将 `nodeIntegration` 设置为 `true` 来解决这个问题。