import { onMounted, onUnmounted } from "vue"
import ConnectWebSocket from "@/websocket"
import { getWebsocketUrl } from "@/websocket/util/config"

/**
 * 接收一个message函数
 * @param messageHandler
 */
export function useConnectWebSocket(messageHandler, url) {
  const webSocket = new ConnectWebSocket(url || getWebsocketUrl())

  onMounted(() => {
    webSocket?.registerMessageHandler(messageHandler)
    webSocket?.initSocket()
  })

  onUnmounted(() => {
    webSocket?.close()
  })
}