springboot集成websocket

为什么有了http协议我们还需要websocket协议呢
答案很简单，因为 HTTP 协议有一个缺陷：通信只能由客户端发起，HTTP 协议做不到服务器主动向客户端推送信息。

springboo对WebSocket有良好的支持 直接引入 spring-boot-starter-websocket（spirngboot 开发不要太爽）
WebSocketConfig
启用WebSocket的支持也是很简单，几句代码搞定

```java

package com.imooc.sell.config;

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

import java.rmi.server.ServerCloneException;

@Component
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter(){
        return  new ServerEndpointExporter();
    }
}                   

  重点来了       
  WebSocketServer               
1. 可以把这个类看成一个ws协议的controller   
2. 关键注解 @ServerEndpoint("/webSocket")（一定要加前面的/不加就报错！ 大坑啊）  再加上   @Component就可以启用了  @OnOpen开启连接   @OnClose关闭连接   @OmMessage接受消息                  
3. 新建一个set 用来存储websocket方便推送信息               
4. 顺便一提 消息提醒类的需求 最好直接在方法上trycatch 不要抛 因为它毕竟只是一个消息提醒就算没有 也不能影响修改状态之类的主方法运行 不然就得不偿失  当然也得根据实际需求来   
5. 需要消息推送的时候直接调用sendmessage就ok了.                         






    ```java     

    package com.imooc.sell.service;

    import lombok.extern.slf4j.Slf4j;
    import org.springframework.stereotype.Component;

    import javax.websocket.OnClose;
    import javax.websocket.OnMessage;
    import javax.websocket.OnOpen;
    import javax.websocket.Session;
    import javax.websocket.server.ServerEndpoint;
    import java.io.IOException;
    import java.util.concurrent.CopyOnWriteArraySet;

    @Component
    @ServerEndpoint("/webSocket")
    @Slf4j
    public class WebSocket {
    /**与某个客户端的连接会话，需要通过它来给客户端发送数据*/
        private Session session;

        private static CopyOnWriteArraySet<WebSocket> webSocketSet = new CopyOnWriteArraySet<>();

        /**
         * 建立连接
         */
        @OnOpen
        public void onOpen(Session session){
            this.session=session;
            webSocketSet.add(this);
            log.info("【websocket消息】有新的连接, 总数:{}", webSocketSet.size());
        }
        /**
         * 关闭连接
         */
        @OnClose
        public void  onClose(){
            webSocketSet.remove(this);

            log.info("【websocket消息】有新的连接, 总数:{}", webSocketSet.size());
        }
        /**
         * 有消息
         */
        @OnMessage
        public void onMessage(String message){
            log.info("【websocket消息】收到客户端发来的消息:{}", message);
        }


        public  void sendMessage(String message){
            for (WebSocket webSocket:webSocketSet){
                log.info("【websocket消息】广播消息, message={}", message);
                try {
                    webSocket.session.getBasicRemote().sendText(message);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }



    }            

js代码

1. 首先判断浏览器是否支持websocket 太古老的浏览器不支持
2. 一次写上 建立连接 关闭连接 收到消息时候 websocket发生错误 的方法
3. 在onmessage的方法里写上自己需要的逻辑代码 一般都是弹框之类的
4. 在离开页面之前要一定关闭websocket

```js     

<script>
    var websocket = null;
    if('WebSocket' in window) {
        websocket = new WebSocket('ws://hhwan.natapp1.cc/sell/webSocket');
    }else {
        alert('该浏览器不支持websocket!');
    }

    websocket.onopen = function (event) {
        console.log('建立连接');
    }

    websocket.onclose = function (event) {
        console.log('连接关闭');
    }

    websocket.onmessage = function (event) {
        console.log('收到消息:' + event.data);
        $('#wbmessage').append("订单号:"+event.data);
        console.log();
        /*弹窗提醒, 播放音乐*/
        $('#myModal').modal('show');

        document.getElementById('notice').play();
    }

    websocket.onerror = function () {
        alert('websocket通信发生错误！');
    }

    window.onbeforeunload = function () {
        websocket.close();
    }

</script>