1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
| 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()
| })
| }
|
|