70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
import ElementUI from 'element-ui'
|
|
import util from '@/libs/util'
|
|
function initWebSocket (e) {
|
|
const token = util.cookies.get('token')
|
|
if (token) {
|
|
const wsUri = 'ws://127.0.0.1:8000/?auth=' + token
|
|
this.socket = new WebSocket(wsUri)// 这里面的this都指向vue
|
|
this.socket.onerror = webSocketOnError
|
|
this.socket.onmessage = webSocketOnMessage
|
|
this.socket.onclose = closeWebsocket
|
|
}
|
|
}
|
|
|
|
function webSocketOnError (e) {
|
|
ElementUI.Notification({
|
|
title: '',
|
|
message: 'WebSocket连接发生错误' + JSON.stringify(e),
|
|
type: 'error',
|
|
duration: 0
|
|
})
|
|
}
|
|
function webSocketOnMessage (e) {
|
|
const data = JSON.parse(e.data)
|
|
if (data.contentType === 'INFO') {
|
|
ElementUI.Notification({
|
|
title: 'websocket',
|
|
message: data.content,
|
|
type: 'success',
|
|
position: 'bottom-right',
|
|
duration: 3000
|
|
})
|
|
} else if (data.contentType === 'ERROR') {
|
|
ElementUI.Notification({
|
|
title: '',
|
|
message: data.content,
|
|
type: 'error',
|
|
position: 'bottom-right',
|
|
duration: 0
|
|
})
|
|
} else if (data.contentType === 'TEXT') {
|
|
ElementUI.Notification({
|
|
title: '温馨提示',
|
|
message: data.content,
|
|
type: 'success',
|
|
position: 'bottom-right',
|
|
duration: 0
|
|
})
|
|
} else {
|
|
console.log(data.content)
|
|
}
|
|
}
|
|
// 关闭websiocket
|
|
function closeWebsocket () {
|
|
console.log('连接已关闭...')
|
|
close()
|
|
}
|
|
function close () {
|
|
this.socket.close() // 关闭 websocket
|
|
this.socket.onclose = function (e) {
|
|
console.log(e)// 监听关闭事件
|
|
console.log('关闭')
|
|
}
|
|
}
|
|
function webSocketSend (message) {
|
|
this.socket.send(JSON.stringify(message))
|
|
}
|
|
export default {
|
|
initWebSocket, close, webSocketSend
|
|
}
|