一、首先用npm安装 socketJS 和 stompJS
npm install sockjs-client
npm install stompjs
二、在页面中引入这2个js
import SockJS from 'sockjs-client'
import Stomp from 'stompjs'
三、初始化websocket连接,定义一个对象(socketClient)接收订阅服务的实例化
// 初始化ws连接
initSocketConnection () {
const _url = 'Your Websocket Url'
const _socket = new SockJS(_url)
this.socketClient = Stomp.over(_socket)
// 向服务器发起websocket连接并发送CONNECT帧
this.socketClient.connect(
{ login: '', passcode: '' },
// 连接成功的回调函数
function connectCallback (success) {
console.log('webSocket连接成功:', success)
},
// 连接失败时的回调函数
function errorCallBack (error) {
console.log('webSocket连接失败:', error)
}
)
}
四、开始订阅(如需多个订阅频道,可以用一个变量monitorList)
// 点击开始订阅按钮
onStartMonitor () {
// 多个订阅频道时找出当前激活的频道
const _activeRoom = this.monitorList.find(item => item.name === this.activeMonitorTabName)
_activeRoom.isPaused = false
this.startChatRoom()
},
// 点击暂停订阅按钮
onPauseMonitor () {
// 多个订阅频道时找出当前激活的频道
const _activeRoom = this.monitorList.find(item => item.name === this.activeMonitorTabName)
_activeRoom.isPaused = true
this.pauseChatRoom()
},
// 执行开始订阅
startChatRoom () {
// 找出当前激活的频道
const _activeRoom = this.monitorList.find(item => item.name === this.activeMonitorTabName)
// 自己定义查询的参数
const params = {
'chatRoomId': _activeRoom.chatRoomId,
}
// 转换格式为base64,也可以不转,看你的服务器需求
const query = window.btoa(JSON.stringify(params))
// 为当前频道增加订阅服务
_activeRoom.subscribe = this.socketClient.subscribe(`/client/roomChat/?jsonStr=${query}`, (message) => {
// 可以看看message里面的东西,取出想要的即可
const _result = JSON.parse(message.body)
if (_result.data.chatLogs && _result.data.chatLogs.length > 0) {
const _chatData = _result.data.chatLogs
_activeRoom.tableData = _chatData.concat(_activeRoom.tableData)
}
})
},
// 暂停订阅
pauseChatRoom (name) {
let _activeRoom = this.monitorList.find(item => item.name === this.activeMonitorTabName)
if (name) {
_activeRoom = this.monitorList.find(item => item.name === name)
}
if (_activeRoom.subscribe) {
_activeRoom.subscribe.unsubscribe()
}
},
// 销毁ws连接
disconnect () {
if (this.socketClient) {
this.socketClient.disconnect()
}
},
// vue生命周期,离开或关闭页面时销毁ws连接
beforeDestroy () {
this.disconnect()
},
五、增加订阅频道功能
// 增加频道
onAddMonitor () {
if (this.monitorList.length >= 10) {
return this.$Message.error('最多只能添加10个频道')
}
// 使用一个计数器来标记唯一的频道id
this.counter++
const _obj = { ...this.monitorTemplate }
_obj.id = new Date().getTime()
_obj.name = _obj.name + '_' + _obj.id
_obj.chatRoomId = this.counter + '_1'
this.monitorList.push(_obj)
if (this.monitorList.length === 1) {
this.activeMonitorTabName = this.monitorList[0].name
}
},
六、所需要的data数据
data() {
return {
monitorList: [],
monitorTemplate: {
id: null,
label: '频道',
name: 'monitorTab',
chatRoomId: '',
isPaused: true,
subscribe: null,
tableData: [],
},
activeMonitorTabName: '',
socketClient: null,
}
}
相关网站:
sockjs
https://github.com/sockjs/sockjs-node
stomp-websocket
https://github.com/jmesnil/stomp-websocket