maven依赖

pool.xml

    
        
     
      4.0.0
         
     
      springws
         
     
      springws
         
     
      1.0-SNAPSHOT
         
             
                  
       
        org.springframework
                   
       
        spring-context
                   
       
        4.3.9.RELEASE
               
              
                  
       
        org.springframework
                   
       
        spring-webmvc
                   
       
        4.3.9.RELEASE
               
              
                  
       
        org.springframework
                   
       
        spring-websocket
                   
       
        4.3.9.RELEASE
               
          
     
    

web.xml

    
        
         
             
      
       springmvc
              
      
       org.springframework.web.servlet.DispatcherServlet
              
                  
       
        contextConfigLocation
                   
       
        /WEB-INF/springmvc-servlet.xml
               
              
          
         
             
      
       springmvc
              
      
       /
          
     
    

springmvc-servlet.xml

    
        
             
                  
                       
                    
               
          
         
         
         
         
         
         
         
         
             
              
              
              
          
         
             
                  
                       
        
         text/plain;charset=UTF-8
                    
               
          
     
    

显示的控制器：

indexController.java

package com.chen;import org.springframework.stereotype.Controller;import org.springframework.ui.ModelMap;import org.springframework.web.bind.annotation.RequestMapping;/** * Created by yuerwang on 2017/7/6. */@Controller@RequestMapping("/Index")public class IndexController {    @RequestMapping("/index")    public void index(ModelMap mp){         mp.put("chen",123);    }}

显示页面

index.jsp

<%--  Created by IntelliJ IDEA.  Date: 2017/7/6  Time: 14:39  To change this template use File | Settings | File Templates.--%><%@ page contentType="text/html;charset=UTF-8" language="java" %>    ${chen}提交

处理websocket

WebsocketConfig.java

package com.chen.conf;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;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;@Configuration@EnableWebSocketpublic 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();    }}

ChatHandshakeInterceptor.java

package com.chen.conf;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.HandshakeInterceptor;import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;import javax.servlet.http.HttpSession;import java.util.Map;/** * Created by yuerwang on 2017/7/6. */public class ChatHandshakeInterceptor extends HttpSessionHandshakeInterceptor {    @Override    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map
    
      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);    }}
    

ChatMessageHandler.java(主要处理逻辑)

package com.chen.conf;import java.io.IOException;import java.util.ArrayList;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;/** * Created by yuerwang on 2017/7/6. */public class ChatMessageHandler extends TextWebSocketHandler{    private static final ArrayList
    
      users;//这个会出现性能问题，最好用Map来存储，key用userid    static {        users = new ArrayList
     
      ();    }    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 {        for (WebSocketSession user:users){            System.out.println(user);        }        System.out.println("发送的是："+message);        //回复群体消息        //sendMessageToUsers(new TextMessage("全体消息："+message.getPayload()));        //回复发送者消息        session.sendMessage(message);        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();        }        users.remove(session);    }    @Override    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {        users.remove(session);    }    @Override    public boolean supportsPartialMessages() {        return false;    }}
     
    

Constants.java

package com.chen.conf;public class Constants {   public static final String WEBSOCKET_USERNAME = "ws_user";     public static final String SESSION_USERNAME = "se_user";}