/*
 * @Author: shuishen 1109946754@qq.com
 * @Date: 2024-09-09 19:13:20
 * @LastEditors: shuishen 1109946754@qq.com
 * @LastEditTime: 2024-09-09 19:14:05
 * @FilePath: \drone-web-manage\src\websocket\index.js
 * @Description: 
 * 
 * Copyright (c) 2024 by shuishen, All Rights Reserved. 
 */
import ReconnectingWebSocket from "reconnecting-websocket"

/**
 * ConnectWebSocket 类
 * TODO: 优化messageHandler: EventEmitter。暂时传入回调函数
 */
class ConnectWebSocket {
  constructor(url) {
    this._url = url
    this._socket = null
    this._hasInit = false
    this._messageHandler = null
  }

  initSocket() {
    if (this._hasInit) {
      return
    }
    if (!this._url) {
      return
    }
    // 会自动重连，无需处理重连逻辑
    this._socket = new ReconnectingWebSocket(this._url, [], {
      maxReconnectionDelay: 20000, // 断开后最大的重连时间： 20s，每多一次重连，会增加 1.3 倍，5 * 1.3 * 1.3 * 1.3...
      minReconnectionDelay: 5000, // 断开后最短的重连时间： 5s
      maxRetries: 5
    })
    this._hasInit = true

    this._socket.addEventListener("open", this._onOpen.bind(this))
    this._socket.addEventListener("close", this._onClose.bind(this))
    this._socket.addEventListener("error", this._onError.bind(this))
    this._socket.addEventListener("message", this._onMessage.bind(this))
  }

  _onOpen() {
    console.log("连接成功")
  }

  _onClose() {
    console.log("连接已断开")
  }

  _onError() {
    console.log("连接 error")
  }

  registerMessageHandler(messageHandler) {
    this._messageHandler = messageHandler
  }

  _onMessage(msg) {
    const data = JSON.parse(msg.data)
    this._messageHandler && this._messageHandler(data)
    // console.log('接受消息', msg)
  }

  sendMessage = message => {
    this._socket?.send(JSON.stringify(message.data))
  }

  close() {
    this._socket?.close()
  }
}

export default ConnectWebSocket