springboot业务功能实战（四）告别轮询，websocket的集成使用

后端代码

首先加入pom文件

             <dependency>
                 <groupId>org.springframework.boot</groupId>
                 <artifactId>spring-boot-starter-websocket</artifactId>
                 <!-- <version>1.3.5.RELEASE</version> -->
             </dependency>
  
 

加入配置类

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

加入连接发送消息方法

      @Component
      @ServerEndpoint("/websocket/{userName}")
      // 此注解相当于设置访问URL
      public class WebSocket {
         private Session session;
         private static CopyOnWriteArraySet<WebSocket> webSockets = new CopyOnWriteArraySet<>();
         private static Map<String, Session> sessionPool = new HashMap<String, Session>();
         private final static Logger logger = LoggerFactory.getLogger(LoginIntercept.class);
          @OnOpen
         public void onOpen(Session session, @PathParam(value = "userName") String userName) {
              this.session = session;
              webSockets.add(this);
             if (sessionPool.containsKey(userName)) {
                  sessionPool.put(userName + String.valueOf(session.getId()), session);
              } else {
                  sessionPool.put(userName, session);
              }
              logger.info("【websocket消息】有新的连接，总数为:" + webSockets.size());
          }
          @OnClose
         public void onClose() {
              webSockets.remove(this);
              logger.info("【websocket消息】连接断开，总数为:" + webSockets.size());
          }
          @OnMessage
         public void onMessage(String message) {
              logger.info("【websocket消息】收到客户端消息:" + message);
          }
         /**
       * 功能描述: 此为广播消息
       *
       * @param: [message] (消息)
       * @return: void ()
       */
         public void sendAllMessage(String message) {
             for (WebSocket webSocket : webSockets) {
                  logger.info("【websocket消息】广播消息:" + message);
                 try {
                     if (webSocket.session.isOpen()) {
                          webSocket.session.getAsyncRemote().sendText(message);
                      }
                  } catch (Exception e) {
                      e.printStackTrace();
                  }
              }
          }
         /**
       * 功能描述:此为单点消息 (发送文本) 现在可以发送给多客户端
       *
       * @param: [userName, message] (接收人,发送消息)
       * @return: void ()
       */
         public void sendTextMessage(String userName, String message) {
             // 遍历sessionPool
             for (String key : sessionPool.keySet()) {
                 // 存在当前用户
                 if (key.toString().indexOf(userName) != -1) {
                      Session session = sessionPool.get(key);
                     if (session != null && session.isOpen()) {
                         try {
                              session.getAsyncRemote().sendText(message);
                          } catch (Exception e) {
                              e.printStackTrace();
                          }
                      }
                  }
              }
          }
         /**
       * 功能描述: 此为单点消息 (发送文本) 现在可以发送给多客户端
       *
       * @param: [userName, message] (接收人,发送消息)
       * @return: void ()
       */
         public void sendObjMessage(String userName, Object message) {
             // 遍历sessionPool
             for (String key : sessionPool.keySet()) {
                 // 存在当前用户
                 if (key.toString().indexOf(userName) != -1) {
                      Session session = sessionPool.get(key);
                     if (session != null && session.isOpen()) {
                         try {
                              session.getAsyncRemote().sendObject(message);
                          } catch (Exception e) {
                              e.printStackTrace();
                          }
                      }
                  }
              }
          }
      }
  
 

发送信息

      @RestController
      @RequestMapping("websocket")
      public class WebSocketController {
         @GetMapping("setMessage")
         @ApiOperation(value = "发送信息接口", notes = "发送信息接口")
         public Result
              webSocket(@ApiParam(name = "定时任务日志实体", value = "定时任务日志实体", required = false) @RequestBody MessageVO messageVO) {
              Result result = new Result();
              String userName = messageVO.getUserName();
              String message = messageVO.getMessage();
              WebSocket webSocket = new WebSocket();
              webSocket.sendTextMessage(userName, message);
             return result;
          }
      }
  
 

前段代码

      import sysConfig from "../config";
      import {Notification} from 'element-ui';
      import {EVENT_TYPE} from "../const";
      export function openSocket(userId) {
       let ws = new WebSocket(`${sysConfig.SOCKET_URL}/${userId}`);
       // let ws = new WebSocket(`ws://121.40.165.18:8800`);
        ws.onopen = function (evt) {
          Notification({
           title: '欢迎回来！',
           message: `${sysConfig.SOCKET_URL}/${userId}`
          });
        };
        ws.onmessage = function (e) {
         console.log(typeof e.data);
         try{
           if(e.data!=undefined || e.data!=null){
             let json= JSON.parse(e.data);
              Notification({
               title: json.messageTitle,
               message: json.messageText
              });
             //通知页面更新
             window.postMessage(EVENT_TYPE.updateMessage,'/');
            }
          }catch(err){
             console.log("webSocke异常,异常信息:"+err)
          }
         //ws.close();
        };
        ws.onclose = function (evt) {
         console.log('Connection closed.');
        };
      }
  
 

文章来源: baocl.blog.csdn.net，作者：小黄鸡1992，版权归原作者所有，如需转载，请联系作者。

原文链接：baocl.blog.csdn.net/article/details/105543004