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

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

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

    return webSocket;
}