const socket = new WebSocket('ws://localhost:8080');

socket.onopen = () => {
  console.log('连接已建立');
  // 发送消息到服务端
  socket.send('Hello Server!');
};

socket.onmessage = (event) => {
  console.log('收到消息:', event.data);
  // 可在此处处理服务端推送的数据
};

const WebSocket = require('ws');
const wss = new WebSocket.Server({ port: 8080 });

wss.on('connection', (ws) => {
  console.log('客户端已连接');

  ws.on('message', (data) => {
    console.log('收到:', data);
    // 回传消息
    ws.send(`Echo: ${data}`);
  });
});

function createSocket() {
  const socket = new WebSocket('ws://localhost:8080');
  let heartBeatInterval;
  let reconnnectTimer;

  function heartbeat() {
    // 发送心跳包
    socket.send('ping');
    // 设置超时，若未收到 pong 则断开
    clearTimeout(reconnnectTimer);
    reconnnectTimer = setTimeout(() => {
      socket.close();
    }, 3000); // 等待 3 秒
  }

  socket.onopen = () => {
    console.log('连接打开');
    // 开始心跳
    heartBeatInterval = setInterval(heartbeat, 5000); // 每 5 秒发一次
  };

  socket.onmessage = (event) => {
    if (event.data === 'pong') {
      // 清除重连计时器
      clearTimeout(reconnnectTimer);
      return;
    }
    console.log('收到数据:', event.data);
  };

  socket.onclose = () => {
    console.log('连接关闭，尝试重连...');
    clearInterval(heartBeatInterval);
    clearTimeout(reconnnectTimer);
    // 延迟重连，避免频繁连接
    setTimeout(createSocket, 2000);
  };

  socket.onerror = () => {
    console.error('连接出错');
    socket.close(); // 触发 onclose
  };

  return socket;
}

createSocket();

wss.on('connection', (ws) => {
  // 监听消息
  ws.on('message', (data) => {
    if (data === 'ping') {
      ws.send('pong');
      return;
    }
    console.log('收到:', data);
  });

  // 可选：服务端也可主动发送 ping，由客户端回应
});