java和Vue WebSocket 使用

最新推荐文章于 2025-02-21 18:15:01 发布

Trank-Lw

最新推荐文章于 2025-02-21 18:15:01 发布

阅读量1.1k

点赞数

分类专栏：工作中的小技术点

本文链接：https://blog.csdn.net/controller_Lw/article/details/107200727

版权

工作中的小技术点专栏收录该内容

13 篇文章

订阅专栏

就是记录下使用方便以后查看，上班没有详细整理，以后有时间再说使用的地方标红了

controller:
/** * 更新发布操作 * @param id * @return */
@Resource
private WebSocket webSocket;

@RequestMapping(value = "/doReleaseData", method = RequestMethod.GET)
public Result<SysAnnouncement> doReleaseData(@RequestParam(name="id",required=true) String id, HttpServletRequest request) {
   Result<SysAnnouncement> result = new Result<SysAnnouncement>();
   SysAnnouncement sysAnnouncement = sysAnnouncementService.getById(id);
   if(sysAnnouncement==null) {
      result.error500("未找到对应实体");
   }else {
      sysAnnouncement.setSendStatus(CommonSendStatus.PUBLISHED_STATUS_1);//发布中
      sysAnnouncement.setSendTime(new Date());
      String currentUserName = JwtUtil.getUserNameByToken(request);
      sysAnnouncement.setSender(currentUserName);
      boolean ok = sysAnnouncementService.updateById(sysAnnouncement);
      if(ok) {
         result.success("该系统通知发布成功");
         if(sysAnnouncement.getMsgType().equals(CommonConstant.MSG_TYPE_ALL)) {
            JSONObject obj = new JSONObject();
             obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_TOPIC);//消息类型
            obj.put(WebsocketConst.MSG_ID, sysAnnouncement.getId());//消息id（用于点击消息传给后台查询详情）
            obj.put(WebsocketConst.MSG_TXT, sysAnnouncement.getTitile()); //消息名字用于展示
             webSocket.sendAllMessage(obj.toJSONString());
         }else {
            // 2.插入用户通告阅读标记表记录
            String userId = sysAnnouncement.getUserIds();
            String[] userIds = userId.substring(0, (userId.length()-1)).split(",");
            String anntId = sysAnnouncement.getId();
            Date refDate = new Date();
            JSONObject obj = new JSONObject();
             obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_USER);
            obj.put(WebsocketConst.MSG_ID, sysAnnouncement.getId());
            obj.put(WebsocketConst.MSG_TXT, sysAnnouncement.getTitile());
             webSocket.sendMoreMessage(userIds, obj.toJSONString());
         }
      }
   }

   return result;
}

webSocket：

/** * @Description: Websocket 工具类 * */

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArraySet;

import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;

import org.jeecg.common.constant.WebsocketConst;
import org.springframework.stereotype.Component;

import com.alibaba.fastjson.JSONObject;

import lombok.extern.slf4j.Slf4j;

/**
* @Author scott
* @Date 2019/11/29 9:41
* @Description: 此注解相当于设置访问URL
*/
@Component
@Slf4j
@ServerEndpoint("/websocket/{userId}") //此注解相当于设置访问URL
public class WebSocket {

private Session session;

    private static CopyOnWriteArraySet<WebSocket> webSockets =new CopyOnWriteArraySet<>();
    private static Map<String,Session> sessionPool = new HashMap<String,Session>();
    //创建链接
    @OnOpen
    public void onOpen(Session session, @PathParam(value="userId")String userId) {
        try {
         this.session = session;
         webSockets.add(this);
         sessionPool.put(userId, session);
         log.info("【websocket消息】有新的连接，总数为:"+webSockets.size());
      } catch (Exception e) {
      }
    }
    //关闭链接
    @OnClose
    public void onClose() {
        try {
         webSockets.remove(this);
         log.info("【websocket消息】连接断开，总数为:"+webSockets.size());
      } catch (Exception e) {
      }
    }
    //
    @OnMessage
    public void onMessage(String message) {
        //todo 现在有个定时任务刷，应该去掉
       log.debug("【websocket消息】收到客户端消息:"+message);
       JSONObject obj = new JSONObject();
       obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_CHECK);//业务类型
       obj.put(WebsocketConst.MSG_TXT, "心跳响应");//消息内容
       session.getAsyncRemote().sendText(obj.toJSONString());
    }

    // 此为广播消息
    public void sendAllMessage(String message) {
       log.info("【websocket消息】广播消息:"+message);
        for(WebSocket webSocket : webSockets) {
            try {
               if(webSocket.session.isOpen()) {
                  webSocket.session.getAsyncRemote().sendText(message);
               }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    // 此为单点消息
    public void sendOneMessage(String userId, String message) {
        Session session = sessionPool.get(userId);
        if (session != null&&session.isOpen()) {
            try {
               log.info("【websocket消息】单点消息:"+message);
                session.getAsyncRemote().sendText(message);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    // 此为单点消息(多人)
    public void sendMoreMessage(String[] userIds, String message) {
       for(String userId:userIds) {
          Session session = sessionPool.get(userId);
            if (session != null&&session.isOpen()) {
                try {
                   log.info("【websocket消息】单点消息:"+message);
                    session.getAsyncRemote().sendText(message);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
       }

}

}

webSocket常用量（都是给前台判断的具体）

/**
 * @Description: Websocket常量类
 * @author: taoyan
 * @date: 2020年03月23日
 */
public class WebsocketConst {


    /**
     * 消息json key:cmd
     */
    public static final String MSG_CMD = "cmd";

    /**
     * 消息json key:msgId
     */
    public static final String MSG_ID = "msgId";

    /**
     * 消息json key:msgTxt
     */
    public static final String MSG_TXT = "msgTxt";

    /**
     * 消息json key:userId
     */
    public static final String MSG_USER_ID = "userId";

    /**
     * 消息类型 heartcheck
     */
    public static final String CMD_CHECK = "heartcheck";

    /**
     * 消息类型 user 用户消息
     */
    public static final String CMD_USER = "user";

    /**
     * 消息类型 topic 系统通知
     */
    public static final String CMD_TOPIC = "topic";

    /**
     * 消息类型 email
     */
    public static final String CMD_EMAIL = "email";

    /**
     * 消息类型 meetingsign 会议签到
     */
    public static final String CMD_SIGN = "sign";

}

WebSocketConfig：

/** *放在项目里的配置文件
*/

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration
public class WebSocketConfig {
/**
*    注入ServerEndpointExporter，
*    这个bean会自动注册使用了@ServerEndpoint注解声明的Websocket endpoint
*/
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}

}

前台Vue

<a-popover

trigger="click"

placement="bottomRight"

:autoAdjustOverflow="true"

:arrowPointAtCenter="true"

overlayClassName="header-notice-wrapper"

@visibleChange="handleHoverChange"

:overlayStyle="{ width: '300px', top: '50px' }">

<a-spin :spinning="loadding">

<a-tabs>

<a-tab-pane :tab="msg1Title" key="1">

<!--<a-list>

<a-list-item>

<a-list-item-meta title="你收到了 14 份新周报" description="一年前">

<a-avatar style="background-color: white" slot="avatar" src="https://gw.alipayobjects.com/zos/rmsportal/ThXAXghbEsBCCSDihZxY.png"/>

</a-list-item-meta>

</a-list-item>

<a-list-item>

<a-list-item-meta title="你推荐的 IT大牛已通过第三轮面试" description="一年前">

<a-avatar style="background-color: white" slot="avatar" src="https://gw.alipayobjects.com/zos/rmsportal/OKJXDXrmkNshAMvwtvhu.png"/>

</a-list-item-meta>

</a-list-item>

<a-list-item>

<a-list-item-meta title="这种模板可以区分多种通知类型" description="一年前">

<a-avatar style="background-color: white" slot="avatar" src="https://gw.alipayobjects.com/zos/rmsportal/kISTdvpyTAhtGxpovNWd.png"/>

</a-list-item-meta>

</a-list-item>

</a-list>-->

<a-list>

<a-list-item :key="index" v-for="(record, index) in announcement1">

<p><a @click="showAnnouncement(record)">{{ record.content }}</a></p>

<p style="color: rgba(0,0,0,.45);margin-bottom: 0px">{{ record.createTime }} 发布</p>

</div>

<a-tag @click="showAnnouncement(record)" v-if="record.priority === 'L'" color="blue">一般消息</a-tag>

<a-tag @click="showAnnouncement(record)" v-if="record.priority === 'M'" color="orange">重要消息</a-tag>

<a-tag @click="showAnnouncement(record)" v-if="record.priority === 'H'" color="red">紧急消息</a-tag>

</div>

</a-list-item>

<a-button @click="toMyAnnouncement()" type="dashed" block>查看更多</a-button>

</div>

</a-list>

</a-tab-pane>

<a-tab-pane :tab="msg2Title" key="2">

<a-list>

<a-list-item :key="index" v-for="(record, index) in announcement2">

<p><a @click="showAnnouncement(record)">{{ record.titile }}</a></p>

<p style="color: rgba(0,0,0,.45);margin-bottom: 0px">{{ record.createTime }} 发布</p>

</div>

<a-tag @click="showAnnouncement(record)" v-if="record.priority === 'L'" color="blue">一般消息</a-tag>

<a-tag @click="showAnnouncement(record)" v-if="record.priority === 'M'" color="orange">重要消息</a-tag>

<a-tag @click="showAnnouncement(record)" v-if="record.priority === 'H'" color="red">紧急消息</a-tag>

</div>

</a-list-item>

<a-button @click="toMyAnnouncement()" type="dashed" block>查看更多</a-button>

</div>

</a-list>

</a-tab-pane>

</a-tabs>

</a-spin>

</template>

<a-badge :count="msgTotal">

<a-icon style="font-size: 16px; padding: 4px" type="bell" />

</a-badge>

</span>

<show-announcement ref="ShowAnnouncement" @ok="modalFormOk"></show-announcement>

<dynamic-notice ref="showDynamNotice" :path="openPath" :formData="formData"/>

</a-popover>

</template>

import { getAction,putAction } from '@/api/manage'

import ShowAnnouncement from './ShowAnnouncement'

import store from '@/store/'

import DynamicNotice from './DynamicNotice'

export default {

name: "HeaderNotice",

components: {

DynamicNotice,

ShowAnnouncement,

data () {

return {

loadding: false,

url:{

listCementByUser:"/sys/annountCement/listByUser",

editCementSend:"/messageMchealth/getMsgById",

queryById:"/sys/annountCement/queryById",

hovered: false,

announcement1:[],

announcement2:[],

msg1Count:"0",

msg2Count:"0",

msg1Title:"通知(0)",

msg2Title:"",

stopTimer:false,

websock: null,

lockReconnect:false,

heartCheck:null,

formData:{},

openPath:''

}

computed:{

msgTotal () {

return parseInt(this.msg1Count)+parseInt(this.msg2Count);

}

mounted() {

this.loadData();

//this.timerFun();

this.initWebSocket();//初始化化

// this.heartCheckFun();

destroyed: function () { // 离开页面生命周期函数

this.websocketclose();

methods: {

timerFun() {

this.stopTimer = false;

let myTimer = setInterval(()=>{

// 停止定时器

if (this.stopTimer == true) {

clearInterval(myTimer);

return;

}

this.loadData()

},6000)

loadData (){

try {

// 获取系统消息

getAction(this.url.listCementByUser).then((res) => {

if (res.success) {

this.announcement1 = res.result.anntMsgList;

this.msg1Count = res.result.anntMsgTotal;

this.msg1Title = "通知(" + res.result.anntMsgTotal + ")";

this.announcement2 = res.result.sysMsgList;

this.msg2Count = res.result.sysMsgTotal;

this.msg2Title = "系统消息(" + res.result.sysMsgTotal + ")";

}

}).catch(error => {

console.log("系统消息通知异常",error);//这行打印permissionName is undefined

this.stopTimer = true;

console.log("清理timer");

});

} catch (err) {

this.stopTimer = true;

console.log("通知异常",err);

}

fetchNotice () {

if (this.loadding) {

this.loadding = false

return

}

this.loadding = true

setTimeout(() => {

this.loadding = false

}, 200)

showAnnouncement(record){

getAction(this.url.editCementSend,{id:record.id}).then((res)=>{

if(res.success){

this.loadData();

}

});

this.hovered = false;

if(record.openType==='component'){

this.openPath = record.openPage;

this.formData = {id:record.busId};

this.$refs.showDynamNotice.detail(record.openPage);

}else{

this.$refs.ShowAnnouncement.detail(record);

}

toMyAnnouncement(){

this.$router.push({

path: '/isps/userAnnouncement',

name: 'isps-userAnnouncement'

});

modalFormOk(){

handleHoverChange (visible) {

this.hovered = visible;

initWebSocket: function () {

// WebSocket与普通的请求所用协议有所不同，ws等同于http，wss等同于https

var userId = store.getters.userInfo.id;

var url = window._CONFIG['domianURL'].replace("https://","wss://").replace("http://","ws://")+"/websocket/"+userId;

console.log(url);

this.websock = new WebSocket(url);

this.websock.onopen = this.websocketOnopen;

this.websock.onerror = this.websocketOnerror;

this.websock.onmessage = this.websocketOnmessage;

this.websock.onclose = this.websocketOnclose;

websocketOnopen: function () {

console.log("WebSocket连接成功");

//心跳检测重置

//this.heartCheck.reset().start();

websocketOnerror: function (e) {

console.log("WebSocket连接发生错误");

this.reconnect();

websocketOnmessage: function (e) {

console.log("-----接收消息-------",e.data);

var data = eval("(" + e.data + ")"); //解析对象

if(data.cmd == "topic"){

//系统通知

this.loadData();

}else if(data.cmd == "user"){

//用户消息

this.loadData();

}

//心跳检测重置

//this.heartCheck.reset().start();

websocketOnclose: function (e) {

console.log("connection closed (" + e.code + ")");

this.reconnect();

websocketSend(text) { // 数据发送

try {

this.websock.send(text);

} catch (err) {

console.log("send failed (" + err.code + ")");

}

openNotification (data) {

var text = data.msgTxt;

const key = `open${Date.now()}`;

this.$notification.open({

message: '消息提醒',

placement:'bottomRight',

description: text,

key,

btn: (h)=>{

return h('a-button', {

props: {

type: 'primary',

size: 'small',

on: {

click: () => this.showDetail(key,data)

}

}, '查看详情')

});

reconnect() {

var that = this;

if(that.lockReconnect) return;

that.lockReconnect = true;

//没连接上会一直重连，设置延迟避免请求过多

setTimeout(function () {

console.info("尝试重连...");

that.initWebSocket();

that.lockReconnect = false;

}, 5000);

heartCheckFun(){

var that = this;

//心跳检测,每20s心跳一次

that.heartCheck = {

timeout: 20000,

timeoutObj: null,

serverTimeoutObj: null,

reset: function(){

clearTimeout(this.timeoutObj);

//clearTimeout(this.serverTimeoutObj);

return this;

start: function(){

var self = this;

this.timeoutObj = setTimeout(function(){

//这里发送一个心跳，后端收到后，返回一个心跳消息，

//onmessage拿到返回的心跳就说明连接正常

that.websocketSend("HeartBeat");

console.info("客户端发送心跳");

//self.serverTimeoutObj = setTimeout(function(){//如果超过一定时间还没重置，说明后端主动断开了

// that.websock.close();//如果onclose会执行reconnect，我们执行ws.close()就行了.如果直接执行reconnect 会触发onclose导致重连两次

//}, self.timeout)

}, this.timeout)

}

showDetail(key,data){

this.$notification.close(key);

var id = data.msgId;

getAction(this.url.queryById,{id:id}).then((res) => {

if (res.success) {

var record = res.result;

this.showAnnouncement(record);

}

})

}

</script>

.header-notice-wrapper {

top: 50px !important;

}

</style>

.header-notice{

display: inline-block;

transition: all 0.3s;

span {

vertical-align: initial;

}

</style>