119 lines
3.4 KiB
JavaScript
119 lines
3.4 KiB
JavaScript
// 官方文档：https://docs.microsoft.com/zh-cn/aspnet/core/signalr/javascript-client?view=aspnetcore-6.0&viewFallbackFrom=aspnetcore-2.2&tabs=visual-studio
|
||
import * as signalR from '@microsoft/signalr'
|
||
import { getToken } from '@/utils/auth'
|
||
import { ElNotification, ElMessage } from 'element-plus'
|
||
import useSocketStore from '@/store/modules/socket'
|
||
import { webNotify } from './index'
|
||
export default {
|
||
// signalR对象
|
||
SR: {},
|
||
// 失败连接重试次数
|
||
failNum: 4,
|
||
baseUrl: '',
|
||
init(url) {
|
||
var socketUrl = window.location.origin + url
|
||
const connection = new signalR.HubConnectionBuilder()
|
||
.withUrl(socketUrl, { accessTokenFactory: () => getToken() })
|
||
.withAutomaticReconnect() //自动重新连接
|
||
.configureLogging(signalR.LogLevel.Warning)
|
||
.build()
|
||
this.SR = connection
|
||
// 断线重连
|
||
connection.onclose(async () => {
|
||
console.log('断开连接了')
|
||
console.assert(connection.state === signalR.HubConnectionState.Disconnected)
|
||
// 建议用户重新刷新浏览器
|
||
await this.start()
|
||
})
|
||
|
||
connection.onreconnected(() => {
|
||
console.log('断线重新连接成功')
|
||
})
|
||
|
||
connection.onreconnecting(async () => {
|
||
console.log('断线重新连接中... ')
|
||
ElMessage({
|
||
message: '与服务器通讯断开连接了，请稍后刷新浏览器尝试',
|
||
type: 'warning',
|
||
duration: 3000
|
||
})
|
||
await this.start()
|
||
})
|
||
|
||
this.receiveMsg(connection)
|
||
// 启动
|
||
// this.start();
|
||
},
|
||
/**
|
||
* 调用 this.signalR.start().then(async () => { await this.SR.invoke("method")})
|
||
* @returns
|
||
*/
|
||
async start() {
|
||
try {
|
||
console.log('signalR-1', this.SR.state)
|
||
//使用async和await 或 promise的then 和catch 处理来自服务端的异常
|
||
if (this.SR.state === signalR.HubConnectionState.Disconnected) {
|
||
await this.SR.start()
|
||
}
|
||
|
||
console.log('signalR-2', this.SR.state)
|
||
return true
|
||
} catch (error) {
|
||
console.error(error)
|
||
this.failNum--
|
||
// console.log(`失败重试剩余次数${that.failNum}`, error)
|
||
if (this.failNum > 0 && this.SR.state.Disconnected) {
|
||
setTimeout(async () => {
|
||
await this.start()
|
||
}, 5000)
|
||
}
|
||
return false
|
||
}
|
||
},
|
||
// 接收消息处理
|
||
receiveMsg(connection) {
|
||
connection.on('onlineNum', (data) => {
|
||
useSocketStore().setOnlineUserNum(data)
|
||
})
|
||
// 接收欢迎语
|
||
connection.on('welcome', (data) => {
|
||
ElNotification.info(data)
|
||
})
|
||
// 接收后台手动推送消息
|
||
connection.on('receiveNotice', (title, data) => {
|
||
ElNotification({
|
||
type: 'info',
|
||
title: title,
|
||
message: data,
|
||
dangerouslyUseHTMLString: true,
|
||
duration: 0
|
||
})
|
||
webNotify({ title: title, body: data })
|
||
})
|
||
// 接收系统通知/公告
|
||
connection.on('moreNotice', (data) => {
|
||
if (data.code == 200) {
|
||
useSocketStore().setNoticeList(data.data)
|
||
}
|
||
})
|
||
|
||
// 接收在线用户
|
||
connection.on('onlineUser', (data) => {
|
||
useSocketStore().setOnlineUsers(data)
|
||
})
|
||
|
||
// 接收聊天数据
|
||
connection.on('receiveChat', (data) => {
|
||
const title = `来自${data.userName}的消息通知`
|
||
ElNotification({
|
||
title: title,
|
||
message: data.message,
|
||
type: 'success',
|
||
duration: 0
|
||
})
|
||
|
||
webNotify({ title: title, body: data.message })
|
||
})
|
||
}
|
||
}
|