续 Springboot+WebSocket+RabbitMQ

在上一篇文章《Springboot+WebSocket+RabbitMQ》中有个漏洞，今天给大家修复一下，上一篇文章因为博主的业务需求需要，本人将websocket的服务端和rabbitmq的监听消费端整合在了一起，结果出来了一个意想不到的结果，程序后台控制台一直不停地打印消息，重复地接受前台和队列发来的消息，后来经过仔细研究，应该是监听队列消息与接受前台消息冲突了，后来我就把websocket前台与后台建立连接发消息时作了非空判断，然后将以前整合好的拆分开来，具体代码如下：

websocket前台页面：

//将消息显示在网页上
function setMessageInnerHTML(innerHTML){
    console.log(innerHTML);
    if(innerHTML!=null||innerHTML!=""){
        alert(innerHTML);
    }

}
//关闭连接
function closeWebSocket(){
    websocket.close();
}
//发送消息
function send(){
    var message = document.getElementById('text').value;
    //websocket.send(message);
}

分离后的websocket服务端websocketserver

package com.iecas.controller;

import lombok.extern.slf4j.Slf4j;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.rabbit.annotation.RabbitHandler;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import java.io.IOException;
import java.util.concurrent.CopyOnWriteArraySet;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

import org.springframework.stereotype.Controller;

/**
 * Created by qs5 on 2018/10/11.
 * @author wangzhen
 * @version 1.0
 */
@Slf4j
@ServerEndpoint(value = "/webSocket")
@Controller
public class WebSocketServer {
    private static Logger logger = LoggerFactory.getLogger(WebSocketServer.class);
    //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
    private static int onlineCount = 0;
    //concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
    public static CopyOnWriteArraySet<WebSocketServer> wsClientMap = new CopyOnWriteArraySet<WebSocketServer>();
    //与某个客户端的连接会话，需要通过它来给客户端发送数据
    private Session session;

    /**
     * 连接建立成功调用的方法
     * @param session 当前会话session
     */
    @OnOpen
    public void onOpen (Session session){
        this.session = session;
        wsClientMap.add(this);
        addOnlineCount();
        logger.info(session.getId()+"有新链接加入，当前链接数为：" + wsClientMap.size());
    }
    /**
     * 连接关闭
     */
    @OnClose
    public void onClose (){
        wsClientMap.remove(this);
        subOnlineCount();
        logger.info("有一链接关闭，当前链接数为：" + wsClientMap.size());
    }
    /**
     * 收到客户端消息
     * @param message 客户端发送过来的消息
     * @param session 当前会话session
     * @throws IOException
     */
    @OnMessage
    public void onMessage (String message, Session session)  {
        logger.info("客户端发送过来的消息:" + message);
        for (WebSocketServer webSocketServer : wsClientMap) {
            try{
                webSocketServer.sendMessage(message);
            } catch (Exception e){

                e.printStackTrace();
                continue;
            }

        }

    }
    /**
     * 发生错误
     */
    @OnError
    public void onError(Session session, Throwable error) {
        logger.info("wsClientMap发生错误!");
        error.printStackTrace();
    }

    /**
     * 给所有客户端群发消息
     * @param message 消息内容
     * @throws IOException
     */
   /* public void sendMsgToAll(String message) throws IOException {
        for ( WebSocketServer item : wsClientMap ){
            item.session.getBasicRemote().sendText(message);
        }
        logger.info("成功群送一条消息:" + wsClientMap.size());
    }*/
    //实现服务器主动推送
    public void sendMessage (String message) throws IOException {
        this.session.getBasicRemote().sendText(message);

        logger.info("成功发送一条消息:" + message);
    }

    public static synchronized  int getOnlineCount (){
        return onlineCount;
    }

    public static synchronized void addOnlineCount (){
        WebSocketServer.onlineCount++;
    }

    public static synchronized void subOnlineCount (){
        WebSocketServer.onlineCount--;
    }
}

分离后的监听消费端MsgReceiverController

package com.iecas.controller;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.rabbit.annotation.RabbitHandler;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import java.io.IOException;

import org.springframework.stereotype.Controller;


/**
 * Created by qs5 on 2018/11/11.
 * @author wangzhen
 */
@RabbitListener(queues = "task_REPqueue")
@Controller
public class MsgReceiverController {
    private static Logger logger = LoggerFactory.getLogger(MsgReceiverController.class);
    //右键监听结果
    @RabbitHandler
    public void receiver(String body)throws IOException{
        logger.info("<=============监听到task_REPqueued队列消息============>"+body);
        if(body!=null||body!=""){

            for ( WebSocketServer item : WebSocketServer.wsClientMap ){

                item.sendMessage(body);
            }
        }
        System.out.println("===>"+body);
    }
}

这样控制台循环打印异常消息的问题就解决了，至此终结，如果有帮到各位，可以关注博主，不定期更新实用干货给大家，当然本人也有可能会佛系更新，反正不一定啦。。。
最后送给大家一句微语：
对自己好，就是对生活认真。时间不多，你要尽力而为。时间很久，你会水到渠成。