1
const socketURL = 'ws://xxxx.com/' // 连接地址
2
let ws = null    // websocket实例
3
let lockReconnect = false; // websocket连接状态，避免重复连接
4
let reconnectTimer = 2000  // 断开后2s自动重连
5

6
function setWebsocket() {
7
  ws = new WebSocket(socketURL)
8
  ws.addEventListener('open', () => {
9
    alert('服务连接成功');
10
  })
11
  ws.addEventListener('close', () => {
12
    alert('服务已断开');
13
    reconnect()
14
  })
15
  ws.addEventListener('error', () => {
16
    alert('服务发生异常');
17
    reconnect()
18
  })
19
  ws.addEventListener('message', event => {
20
    alert('收到消息：', event.data)
21
  })
22
}
23

24
// 重连
25
function reconnect() {
26
  if (lockReconnect) return;
27
  lockReconnect = true;
28
  setTimeout(function () {
29
    setWebsocket();
30
    lockReconnect = false;
31
  }, reconnectTimer);
32
}
33

34
// 页面关闭断开socket连接
35
window.onbeforeunload = function() {
36
  ws.close()
37
  return;
38
}
39

40
// 向外暴露
41
export {
42
    ws
43
}