Spring4.0.6 Websocket詳細配置 之 消息模塊javascript
歷經一週把整個消息模塊開發完畢,其中在websocket這塊遇到比較多的問題是中文亂碼,由於項目中用ajax跟後端交互,用@Response註解時候出現中文亂碼,須要Spring MVC相關配置,這塊遇到配置會在另外一個文章體現。html
鑑於網上提供的一些文章,都介紹不是很到位,關鍵部分都沒體現,致使在真實項目中出現各類各樣的問題。前端
===============================================java
環境介紹:jquery
Jdk 1.7web
Tomcat7.0.52 (支持Websocket協議)ajax
Spring4.0.26 (支持Websocket)spring
web.xml(配置了前端自動優化HtmlCompressor和Druid監控),自動優化會影響Websocket js腳本,後面會講apache
=================================================json
配置步驟:
1. 引入Spring相關Jar,特別須要下面這兩個
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-websocket</artifactId>
<version>4.0.6.RELEASE</version>
</dependency>
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-messaging</artifactId>
<version>4.0.6.RELEASE</version>
</dependency>
2. 編寫WebSocketConfig implements WebSocketConfigurer
WebSocketConfig.java
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.web.servlet.config.annotation.EnableWebMvc; import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter; import org.springframework.web.socket.config.annotation.EnableWebSocket; import org.springframework.web.socket.config.annotation.WebSocketConfigurer; import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry; import org.springframework.web.socket.handler.TextWebSocketHandler; import cn.com.ship.message.handler.ChatMessageHandler; import cn.com.ship.message.handler.TextMessageHandler; @Configuration //@EnableWebMvc//這個標註能夠不加,若是有加,要extends WebMvcConfigurerAdapter @EnableWebSocket public class WebSocketConfig implements WebSocketConfigurer { @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(chatMessageHandler(),"/websocket/chatMessageServer.do").addInterceptors(new ChatHandshakeInterceptor()); registry.addHandler(chatMessageHandler(), "/sockjs/chatMessageServer.do").addInterceptors(new ChatHandshakeInterceptor()).withSockJS(); } @Bean public TextWebSocketHandler chatMessageHandler(){ return new ChatMessageHandler(); } }
3. 編寫ChatMessageHandler extends TextWebSocketHandler
ChatMessageHandler.java
import java.io.IOException; import java.util.ArrayList; import org.apache.log4j.Logger; import org.springframework.web.socket.CloseStatus; import org.springframework.web.socket.TextMessage; import org.springframework.web.socket.WebSocketSession; import org.springframework.web.socket.handler.TextWebSocketHandler; import cn.com.ship.message.common.Constants; import cn.com.ship.message.common.MessageCriteria; public class ChatMessageHandler extends TextWebSocketHandler{ private static final ArrayList<WebSocketSession> users;//這個會出現性能問題,最好用Map來存儲,key用userid private static Logger logger = Logger.getLogger(ChatMessageHandler.class); static { users = new ArrayList<WebSocketSession>(); } public ChatMessageHandler() { // TODO Auto-generated constructor stub } /** * 鏈接成功時候,會觸發UI上onopen方法 */ @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { System.out.println("connect to the websocket success......"); users.add(session); //這塊會實現本身業務,好比,當用戶登陸後,會把離線消息推送給用戶 //TextMessage returnMessage = new TextMessage("你將收到的離線"); //session.sendMessage(returnMessage); } /** * 在UI在用js調用websocket.send()時候,會調用該方法 */ @Override protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { super.handleTextMessage(session, message); } /** * 給某個用戶發送消息 * * @param userName * @param message */ public void sendMessageToUser(String userName, TextMessage message) { for (WebSocketSession user : users) { if (user.getAttributes().get(Constants.WEBSOCKET_USERNAME).equals(userName)) { try { if (user.isOpen()) { user.sendMessage(message); } } catch (IOException e) { e.printStackTrace(); } break; } } } /** * 給全部在線用戶發送消息 * * @param message */ public void sendMessageToUsers(TextMessage message) { for (WebSocketSession user : users) { try { if (user.isOpen()) { user.sendMessage(message); } } catch (IOException e) { e.printStackTrace(); } } } @Override public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { if(session.isOpen()){ session.close(); } logger.debug("websocket connection closed......"); users.remove(session); } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception { logger.debug("websocket connection closed......"); users.remove(session); } @Override public boolean supportsPartialMessages() { return false; } }
4. 編寫websocket握手攔截器ChatHandshakeInterceptor extends HttpSessionHandshakeInterceptor
ChatHandshakeInterceptor.java
package cn.com.ship.message.websocket; import java.util.Map; import javax.servlet.http.HttpSession; import org.springframework.http.server.ServerHttpRequest; import org.springframework.http.server.ServerHttpResponse; import org.springframework.http.server.ServletServerHttpRequest; import org.springframework.web.socket.WebSocketHandler; import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor; import cn.com.ship.message.common.Constants; public class ChatHandshakeInterceptor extends HttpSessionHandshakeInterceptor { @Override public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception { System.out.println("Before Handshake"); if (request instanceof ServletServerHttpRequest) { ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request; HttpSession session = servletRequest.getServletRequest().getSession(false); if (session != null) { //使用userName區分WebSocketHandler,以便定向發送消息 String userName = (String) session.getAttribute(Constants.SESSION_USERNAME); if (userName==null) { userName="default-system"; } attributes.put(Constants.WEBSOCKET_USERNAME,userName); } } return super.beforeHandshake(request, response, wsHandler, attributes); } @Override public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception ex) { System.out.println("After Handshake"); super.afterHandshake(request, response, wsHandler, ex); } }
4. 重點在Spring mvc相關配置(常常出現問題就是:中文亂碼,若是是用ajax交互)
<?xml version="1.0" encoding="UTF-8"?> <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:jee="http://www.springframework.org/schema/jee" xmlns:tx="http://www.springframework.org/schema/tx" xmlns:aop="http://www.springframework.org/schema/aop" xmlns:p="http://www.springframework.org/schema/p" xmlns:util="http://www.springframework.org/schema/util" xmlns:tool="http://www.springframework.org/schema/tool" xmlns:context="http://www.springframework.org/schema/context" xmlns:task="http://www.springframework.org/schema/task" xmlns:mvc="http://www.springframework.org/schema/mvc" xmlns:websocket="http://www.springframework.org/schema/websocket" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx.xsd http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop.xsd http://www.springframework.org/schema/jee http://www.springframework.org/schema/jee/spring-jee.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsd http://www.springframework.org/schema/util http://www.springframework.org/schema/util/spring-util.xsd http://www.springframework.org/schema/tool http://www.springframework.org/schema/tool/spring-tool.xsd http://www.springframework.org/schema/task http://www.springframework.org/schema/task/spring-task.xsd http://www.springframework.org/schema/websocket http://www.springframework.org/schema/websocket/spring-websocket-4.1.xsd http://www.springframework.org/schema/mvc http://www.springframework.org/schema/mvc/spring-mvc-4.1.xsd"> <!-- 這個bean要放在context:component-scan這個前面,否則會出現中文亂碼 --> <bean class="org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter"> <property name="messageConverters"> <list> <ref bean="stringHttpMessageConverter" /> <ref bean="byteArrayHttpMessageConverter" /> <ref bean="jsonHttpMessageConverter" /> <ref bean="jsonHttpMessageConverter4JS" /> </list> </property> </bean> <!-- 啓動SpringMVC Controller的註解功能,完成請求和註解POJO的映射 --> <context:component-scan base-package="cn.com.ship.*.**.controller" /> <!-- websocket相關掃描,主要掃描:WebSocketConfig.java 這個類路徑 --> <context:component-scan base-package="cn.com.ship.message.websocket"/> <!-- 下面標籤能夠不加 等價於全部component-scan--> <context:annotation-config /> <!-- 這個重點,標註必須加,websocket用到--> <mvc:annotation-driven/> <bean id="byteArrayHttpMessageConverter" class="org.springframework.http.converter.ByteArrayHttpMessageConverter" /> <bean id="stringHttpMessageConverter" class="org.springframework.http.converter.StringHttpMessageConverter"> <property name="supportedMediaTypes"> <list> <value>text/plain;charset=UTF-8</value> </list> </property> </bean> <bean id="jsonHttpMessageConverter" class="cn.com.ship.external.spring.converter.json.MappingJackson2HttpMessageConverter"> <property name="objectMapper"> <bean class="com.fasterxml.jackson.databind.ObjectMapper"> <!--對屬性值爲null的不序列化反序列化--> <property name="serializationInclusion"> <util:constant static-field="com.fasterxml.jackson.annotation.JsonInclude.Include.NON_NULL"/> </property> </bean> </property> <property name="supportedMediaTypes"> <list> <value>application/json</value> </list> </property> </bean> <bean id="jsonHttpMessageConverter4JS" class="cn.com.ship.external.spring.converter.json.MappingJackson2HttpMessageConverter"> <property name="objectMapper"> <bean class="com.fasterxml.jackson.databind.ObjectMapper"> <property name="dateFormat"> <bean class="java.text.SimpleDateFormat"> <constructor-arg type="java.lang.String" value="yyyy-MM-dd HH:mm:ss" /> </bean> </property> <!--對屬性值爲null的不序列化反序列化--> <property name="serializationInclusion"> <util:constant static-field="com.fasterxml.jackson.annotation.JsonInclude.Include.NON_NULL"/> </property> </bean> </property> <property name="supportedMediaTypes"> <list> <value>text/json</value> </list> </property> </bean> <bean id="viewResolver" class="org.springframework.web.servlet.view.UrlBasedViewResolver"> <property name="viewClass" value="org.springframework.web.servlet.view.JstlView" /> <property name="prefix" value="/" /> <property name="suffix" value=".jsp" /> </bean> <!-- 文件上傳配置 --> <bean id="multipartResolver" class="org.springframework.web.multipart.commons.CommonsMultipartResolver" p:defaultEncoding="utf-8"> <property name="maxUploadSize" value="1024000000" /> <property name="resolveLazily" value="true" /> </bean> </beans>
注意:MappingJackson2HttpMessageConverter.java,來自Spring代碼,而且修改了一點點,這個找到附件位置下載
5. jsp相關Websocket腳本編寫
ws.jsp
<%@ page language="java" contentType="text/html; charset=UTF-8" pageEncoding="UTF-8"%> <html> <head> <title>Java API for WebSocket (JSR-356)</title> </head> <body> <script type="text/javascript" src="http://localhost:8080/ship/js/jquery/jquery.min.js"></script> <script type="text/javascript" src="http://localhost:8080/ship/js/sockjs-0.3.4.min.js"></script> <script type="text/javascript"> var websocket = null; if ('WebSocket' in window) { websocket = new WebSocket("ws://localhost:8080/ship/webSocketServer.do"); } else if ('MozWebSocket' in window) { websocket = new MozWebSocket("ws://localhost:8080/ship/webSocketServer.do"); } else { websocket = new SockJS("http://localhost:8080/ship/sockjs/webSocketServer.do"); } websocket.onopen = onOpen; websocket.onmessage = onMessage; websocket.onerror = onError; websocket.onclose = onClose; function onOpen(openEvt) { //alert(openEvt.Data); } function onMessage(evt) { alert(evt.data); } function onError() {} function onClose() {} function doSend() { if (websocket.readyState == websocket.OPEN) { var msg = document.getElementById("inputMsg").value; websocket.send(msg);//調用後臺handleTextMessage方法 alert("發送成功!"); } else { alert("鏈接失敗!"); } } </script> 請輸入:<textarea rows="5" cols="10" id="inputMsg" name="inputMsg"></textarea> <button onclick="doSend();">發送</button> </body> </html>
login.jsp
<%@ page language="java" contentType="text/html; charset=UTF-8" pageEncoding="UTF-8"%> <!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd"> <html> <head> <meta http-equiv="Content-Type" content="text/html; charset=UTF-8"> <title>Java API for WebSocket (JSR-356)</title> </head> <body> <!-ship是個人項目名--> <form action="/ship/websocket/login.do"> 登陸名:<input type="text" name="username"/> <input type="submit" value="登陸"/> </form> </body> </html>
5. 調用端Controller編寫 WebsocketController.java
package cn.com.ship.websocket.controller; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpSession; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; @Controller public class WebsocketController { @Bean//這個註解會從Spring容器拿出Bean public InfoHandler infoHandler() { return new InfoHandler(); } @RequestMapping("/websocket/login") public void login(HttpServletRequest request, HttpServletResponse response) throws Exception { String username = request.getParameter("username"); HttpSession session = request.getSession(false); session.setAttribute(Constants.SESSION_USERNAME, username); response.sendRedirect("/ship/websocket/ws.jsp"); } @RequestMapping("/websocket/send") @ResponseBody public String send(HttpServletRequest request) { String username = request.getParameter("username"); infoHandler().sendMessageToUser(username, new TextMessage("你好,測試!!!!")); return null; } }
6. 測試Websocket是否鏈接成功
先在login.jsp上隨便輸入用戶名,而後WebsocketController處理完請求,會重定向到ws.jsp
若是後臺打印出現消息,證實Websocket鏈接成功(前兩名消息是在ChatHandshakeInterceptor.java,最後一句是在ChatMessageHandler.java)
Before Handshake
After Handshake
connect to the websocket success......
開發中遇到各個問題,會開另一篇文章來描述!!!