26 lines
610 B
TypeScript
26 lines
610 B
TypeScript
|
import { computed, onUnmounted, unref } from 'vue'
|
||
|
|
import actions from '@/shared/action'
|
||
|
|
|
||
|
|
export default function(onmessage: {(data: any):void}, store: any) {
|
||
|
|
const webSocket = computed(() => store.state.app.$webSocket);
|
||
|
|
console.log(webSocket)
|
||
|
|
if (unref(webSocket)) {
|
||
|
|
webSocket.value.onmessage = onmessage
|
||
|
|
} else {
|
||
|
|
actions.setGlobalState({
|
||
|
|
name: 'cop-web',
|
||
|
|
onmessage
|
||
|
|
})
|
||
|
|
}
|
||
|
|
onUnmounted(() => {
|
||
|
|
if (unref(webSocket)) {
|
||
|
|
webSocket.value.onmessage = null
|
||
|
|
} else {
|
||
|
|
actions.setGlobalState({
|
||
|
|
name: 'cop-web',
|
||
|
|
onmessage: null
|
||
|
|
})
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|