websocket整合rabbitMQ——实现消息推送

        
        org.springframework.boot
            spring-boot-starter-amqp
            2.1.4.RELEASEorg.springframework.boot
            spring-boot-starter-websocket
            2.1.4.RELEASE

org.projectlombok
    lombok
2.1.4.RELEASE

@Slf4j
@ServerEndpoint("/websocket")//把当前类标识成一个WebSocket的服务端，值是访问的URL地址
@Component//spring注入
public class WebSocketServer {

    //与某个客户端的连接会话，需要通过它来给客户端发送数据
    private Session session;

    //存放session的集合，很重要！！
    private static CopyOnWriteArraySet sessions = new CopyOnWriteArraySet();

    //用于存所有的连接服务的客户端，这个对象存储是安全的（因为HashMap不支持线程同步）
    private static ConcurrentHashMap webSocketSet = new ConcurrentHashMap();


    //监听队列，从队列中把刚发送的消息取出来
    @RabbitListener(queues = "pointQueue")
    public void getMessAge(String message, Channel channel, @Header(AmqpHeaders.DELIVERY_TAG) long tag) throws Exception {

        System.out.println("发来消息：" + message);

        channel.basicAck(tag, false);//确认收到--消费

        //用来判断session中是否存在数据
        if (sessions.size() != 0) {
            for (WebSocketServer s : sessions) {
                if (s != null) {
                    s.session.getBasicRemote().sendText(message);//向已连接客户端发送信息
                }
            }
        }
    }


    //连接成功
    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        //这个一定要写，第一次很容易忽略！
        sessions.add(this);
        log.info("[WebSocket] 连接成功，当前连接人数为：={}", webSocketSet.size());

    }

    //连接断开
    @OnClose
    public void onClose() {
        //释放
        sessions.remove(this);
        log.info("[WebSocket] 退出成功，当前连接人数为：={}", webSocketSet.size());
    }

    //收到消息
    @OnMessage
    public String onMessage(String message) {
        log.info("[WebSocket] 收到消息：{}", message);

        //这里是自己测试的，可以忽略
        if (message.equals("一只羊")) {//不能用==
            return "1:1个大白羊";
        }
        if (message.equals("两只羊")) {
            return "2:2个大白羊";
        }
        if (message.equals("三只羊")) {
            return "3:个大白羊";
        }

        return "你已成功连接，这是webSocket服务端的返回信息！";
    }

}

    //websocket 放入到spring容器
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }

script type="text/javascript">

    //设置一个变量，用于存储websocket
    var socket = null;

    //判断当前浏览器是否支持WebSocket
    if (‘WebSocket‘ in window) {
        //alert(‘当前浏览器支持WebSocket‘)
        //实现化WebSocket对象，指定要连接的服务器地址与端口  建立连接
        //这个url必须写对！ws固定的，ip是自己配置的，路径是自己命名的
        socket = new WebSocket("ws://127.0.0.1:7777/websocket");
    }
    else {
        alert(‘当前浏览器 Not support websocket‘)
    }


    //连接成功（回调方法）
    socket.onopen = function () {
        alert("Socket 连接成功");
    }

    //连接错误（回调方法）
    socket.onerror = function () {
        alert(‘连接出错‘)
        //可尝试刷新页面--循环
        //window.location.reload();
    };

    //接收消息（回调方法）
    socket.onmessage = function (event) {
        alert("你收到了一条返回消息："+event.data);
    }

    //连接关闭（回调方法）
    socket.onclose = function () {
        alert("连接关闭");
    }

    //窗口关闭，关闭websocket（否则server端会抛异常）
    window.onbeforeunload = function () {
        socket.close();
    }
    

script>