Newer
Older
shipFront / src / store / modules / websocket.js
[wangxitong] on 30 May 2022 3 KB v1.1.2
import { Notification } from 'element-ui'
import router from '../../router'
import user from './user'
const websocket = {
  state: {
    wsUrl: 'ws://192.168.0.15:12209/websocket/',
    websocket: null,
    wsStatus: false, // websocket连接状态
    needRefresh: false // 是否需要刷新数据
  },
  mutations: {
    SET_WEBSOCKET: (state, websocket) => {
      state.websocket = websocket
    },
    SET_WS_STATUS: (state, wsStatus) => {
      state.wsStatus = wsStatus
    },
    SET_NEED_REFRESH: (state, needRefresh) => {
      state.needRefresh = needRefresh
    }
  },

  actions: {
    // 连接websocket
    initWebSocket({ commit }) {
      if (typeof (WebSocket) === 'undefined') {
        Notification({
          title: '提示',
          message: '当前浏览器无法接收实时报警信息,请使用谷歌浏览器或360浏览器极速模式!',
          type: 'warning',
          duration: 0
        })
      } else {
        // 获取token保存到vuex中的用户信息,此处仅适用于本项目,注意删除或修改
        // 实例化socket,这里我把用户名传给了后台,使后台能判断要把消息发给哪个用户,其实也可以后台直接获取用户IP来判断并推送
        const socketUrl = websocket.state.wsUrl + user.state.id
        console.log(socketUrl,'*********************')
        const socket = new WebSocket(socketUrl)
        commit('SET_WEBSOCKET', socket)
        commit('SET_WS_STATUS', true)
        // 监听socket打开
        socket.onopen = function() {
          console.log('浏览器WebSocket已打开')
        }
        // 监听socket消息接收
        socket.onmessage = function(msg) {
          debugger
          // 转换为json对象
          const data = JSON.parse(msg.data)
          console.log(data)
          if (data.type === 'alarm') {
            let message
            switch (data.alarmType) {
              case '1':
                message = data.robotId + '号' + data.alarmTypeName + ':' + data.alarmValue + '%'
                break
              case '2':
                message = data.robotId + '号' + data.alarmTypeName + ':' + data.alarmValue + '海里'
                break
              case '3':
                message = data.alarmTypeName + ':' + data.alarmValue + 'm/s'
                break
              case '4':
                message = data.robotId + '号' + data.alarmTypeName + ':' + data.alarmValue + '海里'
                break
              case '5':
                message = data.robotId + '号' + data.alarmTypeName + ':' + data.alarmValue + '海里'
                break
              case '6':
                message = data.alarmTypeName + ':' + data.alarmValue + 'kb/s'
                break
            }
              commit('SET_NEED_REFRESH', true)
              Notification({
                title: '新报警来了',
                // 这里也可以把返回信息加入到message中显示
                message: message,
                type: 'warning',
                // onClick: () => {
                //   router.push({
                //     path: '/overview',
                //     query: { refresh: true }
                //   })
                // }
              })
              setTimeout(function() {
                commit('SET_NEED_REFRESH', false)
              }, 2000)
          }else if(data.type === 'shelterPosition'){

          }
        }
        // 监听socket错误
        socket.onerror = function() {
          Notification({
            title: '服务器错误',
            message: '无法接收实时报警信息,请检查服务器后重新刷新页面',
            type: 'error',
            duration: 0
          })
        }
        // 监听socket关闭
        socket.onclose = function() {
          console.log('WebSocket已关闭')
          commit('SET_WS_STATUS', false)
        }
      }
    }
  }
}

export default websocket