import ConnectWebSocket from '@/websocket'
import { getWebsocketUrl } from '@/websocket/util/config'

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

	webSocket?.registerMessageHandler(messageHandler)
	webSocket?.initSocket()

	return webSocket
}