[关闭]
@MrXiao 2018-12-12T09:23:16.000000Z 字数 10434 阅读 1140

Spring4整合websocket

websocket

原文地址：Spring4整合websocket

  1. spring 4.0及以上增加了WebSocket的支持(这里使用4.2.5.RELEASE)。
  2. spring 支持STOMP协议的WebSocket通信。

  3. 应对不支持 WebSocket 的场景，许多浏览器不支持 WebSocket 协议；SockJS 是 WebSocket 技术的一种模拟。SockJS 会 尽可能对应 WebSocket API，但如果 WebSocket 技术 不可用的话，会从如下 方案中挑选最优可行方案：

    1. XHR streaming
    2. XDR streaming
    3. iFrame event source
    4. iFrame HTML file
    5. XHR polling
    6. XDR polling
    7. iFrame XHR polling
    8. JSONP polling

  4. WebSocket 是发送和接收消息的 底层API，而SockJS 是在 WebSocket 之上的 API；最后 STOMP（面向消息的简单文本协议）是基于 SockJS 的高级API。

  5. SockJS 所处理的URL 是 “http:” 或 “https:” 模式
  6. WebSocket 所处理的URL 是“ws:” or “wss:” 模式

下面我们来建立一个websocket实例

1、创建web工程

首先搭建一个简单可访问的web系统，这里提供一个我自己搭建的SSM框架供使用。

2、导入spring-websocket需要的jar

  1. <dependency>
  2.     <groupId>org.springframework</groupId>
  3.     <artifactId>spring-websocket</artifactId>
  4.     <version>${spring.version}</version>
  5. </dependency>
  6. <dependency>
  7.     <groupId>org.springframework</groupId>
  8.     <artifactId>spring-messaging</artifactId>
  9.     <version>${spring.version}</version>
  10. </dependency>
  12. <dependency>
  13.     <groupId>javax.servlet</groupId>
  14.     <artifactId>javax.servlet-api</artifactId>
  15.     <version>3.1.0</version>
  16. </dependency>
  18. <dependency>
  19.     <groupId>org.codehaus.jackson</groupId>
  20.     <artifactId>jackson-mapper-asl</artifactId>
  21.     <version>1.9.13</version>
  22. </dependency>
  23. <dependency>
  24.     <groupId>com.fasterxml.jackson.core</groupId>
  25.     <artifactId>jackson-core</artifactId>
  26.     <version>2.8.0</version>
  27. </dependency>
  28. <dependency>
  29.     <groupId>com.fasterxml.jackson.core</groupId>
  30.     <artifactId>jackson-databind</artifactId>
  31.     <version>2.8.0</version>
  32. </dependency>

3、创建Websocket处理器

可以扩展 AbstractWebSocketHandler ,也可以扩展TextWebSocketHandler(文本处理器)，TextWebSocketHandler 继承 AbstractWebSocketHandler

TestWebsocketHandler

  1. package com.websocket;
  3. import java.io.IOException;
  4. import java.util.ArrayList;
  6. import org.slf4j.Logger;
  7. import org.slf4j.LoggerFactory;
  8. import org.springframework.web.socket.CloseStatus;
  9. import org.springframework.web.socket.PongMessage;
  10. import org.springframework.web.socket.TextMessage;
  11. import org.springframework.web.socket.WebSocketSession;
  12. import org.springframework.web.socket.handler.TextWebSocketHandler;
  14. /**
  15.  * 〈websocket处理类〉
  16.  *
  17.  * @author xiaoyue
  18.  * @create 2018/12/11 15:17
  19.  * @since 1.0.0
  20.  */
  21. public class TestWebsocketHandler extends TextWebSocketHandler {
  23.     private Logger logger = LoggerFactory.getLogger(TextWebSocketHandler.class);
  25.     private static final ArrayList<WebSocketSession> sessions;// 这个会出现性能问题，最好用Map来存储，key用userid
  26.     static {
  27.         sessions = new ArrayList<WebSocketSession>();
  28.     }
  30.     // 接收文本消息，并发送出去
  31.     // js调用websocket.send时候，会调用该方法
  32.     @Override
  33.     protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
  34.         super.handleTextMessage(session, message);
  35.         TextMessage returnMessage = new TextMessage(message.getPayload() + " received at server");
  36.         // session.sendMessage(returnMessage);
  37.         sendMessageToUsers(returnMessage);
  38.     }
  40.     // 连接建立后处理
  41.     @SuppressWarnings("unchecked")
  42.     @Override
  43.     public void afterConnectionEstablished(WebSocketSession session) throws Exception {
  44.         sessions.add(session);
  45.         System.out.println("connect to the websocket success......当前数量:" + sessions.size());
  46.         // 处理离线消息
  47.     }
  49.     // 抛出异常时处理
  50.     @Override
  51.     public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
  52.         if (session.isOpen()) {
  53.             session.close();
  54.         }
  55.         logger.debug("websocket chat connection closed......");
  56.         sessions.remove(session);
  57.     }
  59.     // 连接关闭后处理
  60.     @Override
  61.     public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
  62.         logger.debug("websocket chat connection closed......");
  63.         String username = (String) session.getAttributes().get("WEBSOCKET_USERNAME");
  64.         System.out.println("用户" + username + "已退出！");
  65.         sessions.remove(session);
  66.         System.out.println("剩余在线用户" + sessions.size());
  67.     }
  69.     @Override
  70.     public boolean supportsPartialMessages() {
  71.         return false;
  72.     }
  74.     @Override
  75.     protected void handlePongMessage(WebSocketSession session, PongMessage message) throws Exception {
  76.         super.handlePongMessage(session, message);
  77.     }
  79.     /**
  80.      * 给某个用户发送消息
  81.      *
  82.      * @param userName
  83.      * @param message
  84.      */
  85.     public void sendMessageToUser(String userName, TextMessage message) {
  86.         for (WebSocketSession user : sessions) {
  87.             if (user.getAttributes().get("WEBSOCKET_USERNAME").equals(userName)) {
  88.                 try {
  89.                     if (user.isOpen()) {
  90.                         user.sendMessage(message);
  91.                     }
  92.                 } catch (IOException e) {
  93.                     e.printStackTrace();
  94.                 }
  95.                 break;
  96.             }
  97.         }
  98.     }
  100.     /**
  101.      * 给所有在线用户发送消息
  102.      *
  103.      * @param message
  104.      */
  105.     public void sendMessageToUsers(TextMessage message) {
  106.         for (WebSocketSession user : sessions) {
  107.             try {
  108.                 if (user.isOpen()) {
  109.                     user.sendMessage(message);
  110.                 }
  111.             } catch (IOException e) {
  112.                 e.printStackTrace();
  113.             }
  114.         }
  115.     }
  116. }

4、创建WebSocket握手拦截器

TestHandshakeInterceptor

  1. package com.websocket;
  3. import java.util.Map;
  5. import javax.servlet.http.HttpSession;
  7. import org.springframework.http.server.ServerHttpRequest;
  8. import org.springframework.http.server.ServerHttpResponse;
  9. import org.springframework.http.server.ServletServerHttpRequest;
  10. import org.springframework.web.socket.WebSocketHandler;
  11. import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
  13. /**
  14.  * 〈握手（handshake）接口〉
  15.  *
  16.  * @author xiaoyue
  17.  * @create 2018/12/11 15:18
  18.  * @since 1.0.0
  19.  */
  20. public class TestHandshakeInterceptor extends HttpSessionHandshakeInterceptor {
  22.     // 常用在注册用户信息，绑定WebSocketSession，在handler里根据用户信息获取WebSocketSession发送消息
  23.     @Override
  24.     public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
  25.             Map<String, Object> attributes) throws Exception {
  26.         System.out.println("Before Handshake");
  27.         if (request instanceof ServletServerHttpRequest) {
  28.             ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
  29.             HttpSession session = servletRequest.getServletRequest().getSession(false);
  30.             if (session != null) {
  31.                 // 使用userName区分WebSocketHandler，以便定向发送消息
  32.                 String userName = (String) session.getAttribute("SESSION_USERNAME");
  33.                 if (userName == null) {
  34.                     userName = "default-system";
  35.                 }
  36.                 attributes.put("WEBSOCKET_USERNAME", userName);
  37.             }
  38.         }
  39.         return super.beforeHandshake(request, response, wsHandler, attributes);
  40.     }
  42.     @Override
  43.     public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
  44.             Exception ex) {
  45.         System.out.println("After Handshake");
  46.         super.afterHandshake(request, response, wsHandler, ex);
  47.     }
  49. }

5、创建Websocket配置文件

采用注解的方式。

WebSocketConfig

  1. package com.websocket;
  3. import org.springframework.context.annotation.Bean;
  4. import org.springframework.context.annotation.Configuration;
  5. import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
  6. import org.springframework.web.socket.WebSocketHandler;
  7. import org.springframework.web.socket.config.annotation.EnableWebSocket;
  8. import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
  9. import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
  11. /**
  12.  * 〈WebSocket入口〉
  13.  *
  14.  * @author xiaoyue
  15.  * @create 2018/12/11 15:37
  16.  * @since 1.0.0
  17.  */
  18. @Configuration
  19. // @EnableWebMvc//这个标注可以不加，如果有加，要extends WebMvcConfigurerAdapter
  20. @EnableWebSocket
  21. public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {
  23.     // 注册处理器，绑定对应的url
  24.     @Override
  25.     public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
  26.         System.out.println("==========================注册socket");
  27.         String websocket_url = "/websocket/socketServer.do";
  28.         registry.addHandler(testWebSocketHandler(), websocket_url).addInterceptors(testHandshakeInterceptor());
  29.         String sockjs_url = "/sockjs/socketServer.do";
  30.         registry.addHandler(testWebSocketHandler(), sockjs_url).addInterceptors(testHandshakeInterceptor())
  31.                 .withSockJS();
  32.     }
  34.     @Bean
  35.     public WebSocketHandler testWebSocketHandler() {
  36.         return new TestWebsocketHandler();
  37.     }
  39.     @Bean
  40.     public TestHandshakeInterceptor testHandshakeInterceptor() {
  41.         return new TestHandshakeInterceptor();
  42.     }
  44. }

5、创建Controller,控制页面跳转

WebsocketController

  1. package com.websocket;
  3. import org.springframework.context.annotation.Bean;
  4. import org.springframework.stereotype.Controller;
  5. import org.springframework.web.bind.annotation.RequestMapping;
  6. import org.springframework.web.bind.annotation.ResponseBody;
  7. import org.springframework.web.servlet.ModelAndView;
  8. import org.springframework.web.socket.TextMessage;
  10. import javax.servlet.http.HttpServletRequest;
  11. import javax.servlet.http.HttpServletResponse;
  12. import javax.servlet.http.HttpSession;
  14. /**
  15.  * 〈〉
  16.  *
  17.  * @author xiaoyue
  18.  * @create 2018/12/11 16:38
  19.  * @since 1.0.0
  20.  */
  21. @Controller
  22. public class WebsocketController {
  23.     @Bean // 这个注解会从Spring容器拿出Bean
  24.     public TestWebsocketHandler infoHandler() {
  25.         return new TestWebsocketHandler();
  26.     }
  28.     @RequestMapping("/websocket/login")
  29.     public ModelAndView login(HttpServletRequest request, HttpServletResponse response) throws Exception {
  30.         String username = request.getParameter("username");
  31.         System.out.println(username + "登录");
  32.         HttpSession session = request.getSession(false);
  33.         session.setAttribute("SESSION_USERNAME", username);
  34.         // response.sendRedirect("/quicksand/jsp/websocket.jsp");
  35.         return new ModelAndView("websocket");
  36.     }
  38.     @RequestMapping("/websocket/send")
  39.     @ResponseBody
  40.     public String send(HttpServletRequest request) {
  41.         String username = request.getParameter("username");
  42.         infoHandler().sendMessageToUser(username, new TextMessage("你好，测试！！！！"));
  43.         return null;
  44.     }
  45. }

6、测试页面

登录页面，index.jsp

  1. <%@ page language="java" contentType="text/html; charset=utf-8"
  2.          pageEncoding="utf-8"%>
  3. <!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
  4. <html>
  5. <body>
  6. <h2>Hello World!</h2>
  7. <body>
  8. <form action="/websocket/login.do">
  9.     登录名：<input type="text" name="username"/>
  10.     <input type="submit" value="登录"/>
  11. </form>
  12. </body>
  13. </body>
  14. </html>

消息发送接收页面 websocket.jsp

  1. <%@ page language="java" contentType="text/html; charset=utf-8"
  2.          pageEncoding="utf-8" %>
  3. <!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
  4. <html>
  5. <head>
  6.     <meta http-equiv="Content-Type" content="text/html; charset=utf-8">
  7.     <title>Java API for WebSocket (JSR-356)</title>
  8. </head>
  9. <body>
  10. <script type="text/javascript" src="http://cdn.bootcss.com/jquery/3.1.0/jquery.min.js"></script>
  11. <script type="text/javascript" src="http://cdn.bootcss.com/sockjs-client/1.1.1/sockjs.js"></script>
  12. <script type="text/javascript">
  13.     var websocket = null;
  14.     if ('WebSocket' in window) {
  15.         //Websocket的连接
  16.         websocket = new WebSocket("ws://localhost:8080/websocket/socketServer.do");//WebSocket对应的地址
  17.     }
  18.     else if ('MozWebSocket' in window) {
  19.         //Websocket的连接
  20.         websocket = new MozWebSocket("ws://localhost:8080/ws/websocket/socketServer.do");//SockJS对应的地址
  21.     }
  22.     else {
  23.         //SockJS的连接
  24.         websocket = new SockJS("http://localhost:8080/ws/sockjs/socketServer.do");    //SockJS对应的地址
  25.     }
  26.     websocket.onopen = onOpen;
  27.     websocket.onmessage = onMessage;
  28.     websocket.onerror = onError;
  29.     websocket.onclose = onClose;
  31.     function onOpen(openEvt) {
  32.         //alert(openEvt.Data);
  33.     }
  35.     function onMessage(evt) {
  36.         alert(evt.data);
  37.     }
  38.     function onError() {
  39.     }
  40.     function onClose() {
  41.     }
  43.     function doSend() {
  44.         if (websocket.readyState == websocket.OPEN) {
  45.             var msg = document.getElementById("inputMsg").value;
  46.             websocket.send(msg);//调用后台handleTextMessage方法
  47.             alert("发送成功!");
  48.         } else {
  49.             alert("连接失败!");
  50.         }
  51.     }
  53.     window.close = function () {
  54.         websocket.onclose();
  55.     }
  56. </script>
  57. 请输入：<textarea rows="3" cols="100" id="inputMsg" name="inputMsg"></textarea>
  58. <button onclick="doSend();">发送</button>
  59. </body>

注意事项

  1. 因为是基于Spring注解，所以要将以上websocket的三个文件添加到Spring的扫描路径中。
  2. 建议所有的servlet和filter都要加<async-supported>true</async-supported>，支持异步处理。
添加新批注
在作者公开此批注前，只有你和作者可见。
回复批注