spring与websocket整合

首先spring必须用4.0以上，

依赖包：

　　　　<dependency>  
            <groupId>com.fasterxml.jackson.core</groupId>  
            <artifactId>jackson-annotations</artifactId>  
            <version>2.3.0</version>  
        </dependency>  
        <dependency>  
            <groupId>com.fasterxml.jackson.core</groupId>  
            <artifactId>jackson-core</artifactId>  
            <version>2.3.1</version>  
        </dependency>  
        <dependency>  
            <groupId>com.fasterxml.jackson.core</groupId>  
            <artifactId>jackson-databind</artifactId>  
            <version>2.3.3</version>  
        </dependency>  
        <dependency>    
           <groupId>org.springframework</groupId>    
           <artifactId>spring-websocket</artifactId>    
           <version>4.0.5.RELEASE</version>    
        </dependency>    
        <dependency>    
           <groupId>org.springframework</groupId>    
           <artifactId>spring-messaging</artifactId>    
           <version>4.0.5.RELEASE</version>    
        </dependency> 

编写WebSocketConfig、HandshakeInterceptor与WebSocketHander

import org.springframework.context.annotation.Configuration;  
import org.springframework.web.servlet.config.annotation.EnableWebMvc;  
import org.springframework.web.socket.config.annotation.EnableWebSocket;  
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;  
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;  
  
@Configuration  
@EnableWebMvc  
@EnableWebSocket  
public class WebSocketConfig implements WebSocketConfigurer {  
    @Override  
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {  
          
        /** 
         * 支持websocket 的 connection 
         */  
        registry.addHandler(new WebSocketHander(),"/websocket").addInterceptors(new HandshakeInterceptor());  
          
        /** 
         * 不支持websocket的connenction,采用sockjs 
         */  
        registry.addHandler(new WebSocketHander(),"/sockjs/websocket").addInterceptors(new HandshakeInterceptor()).withSockJS();  
    }  
}  

import java.util.Map;  
  
import org.springframework.http.server.ServerHttpRequest;  
import org.springframework.http.server.ServerHttpResponse;  
import org.springframework.web.socket.WebSocketHandler;  
public class HandshakeInterceptor implements org.springframework.web.socket.server.HandshakeInterceptor {  
  
    /** 
     * Interceptor before hander  
     */  
    @Override  
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map<String, Object> map) throws Exception {  
        return true;  
    }  
  
    @Override  
    public void afterHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Exception e) {  
  
    }  
}  

import java.io.IOException;  
import java.util.ArrayList;  
  
import javax.annotation.Resource;  
  
import org.slf4j.Logger;  
import org.springframework.web.socket.CloseStatus;  
import org.springframework.web.socket.TextMessage;  
import org.springframework.web.socket.WebSocketHandler;  
import org.springframework.web.socket.WebSocketMessage;  
import org.springframework.web.socket.WebSocketSession;  
  
public class WebSocketHander implements WebSocketHandler {  
    
   private static final Logger logger = org.slf4j.LoggerFactory.getLogger(WebSocketHander.class);  
      
    private static final ArrayList<WebSocketSession> users = new ArrayList<WebSocketSession>();  
  
    /** 
     * after connection establish 
     */  
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {  
        logger.info("connect success...");  
        users.add(session);  
    }  
  
    /** 
     * process the received message  
     */  
    @Override  
    public void handleMessage(WebSocketSession webSocketSession, WebSocketMessage<?> webSocketMessage) throws Exception {  
        sendMessageToUsers(new TextMessage(webSocketMessage.getPayload() + "hello"));  
    }  
  
    @Override  
    public void handleTransportError(WebSocketSession webSocketSession, Throwable throwable) throws Exception {  
        if(webSocketSession.isOpen()){  
            webSocketSession.close();  
        }  
        logger.info("connenction error,close the connection...");  
        users.remove(webSocketSession);  
    }  
  
    @Override  
    public void afterConnectionClosed(WebSocketSession webSocketSession, CloseStatus closeStatus) throws Exception {  
        logger.info("close the connenction..."+closeStatus.toString());  
        users.remove(webSocketSession);  
    }  
  
    @Override  
    public boolean supportsPartialMessages() {  
        return false;  
    }  
    /** 
     * 给所有在线用户发送消息 
     * 
     * @param message 
     */  
    public void sendMessageToUsers(TextMessage message) {  
        for (WebSocketSession user : users) {  
            try {  
                if (user.isOpen()) {  
                    user.sendMessage(message);  
                }  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
    }
}  

前端页面js

<script type="text/javascript">  
        $(function(){  
            var websocket;  
            if ('WebSocket' in window) {  
                alert("WebSocket");  
                websocket = new WebSocket("ws://localhost:8080/项目名称/websocket");  
            } else if ('MozWebSocket' in window) {  
                alert("MozWebSocket");  
                websocket = new MozWebSocket("ws://echo");  
            } else {  
                alert("SockJS");  
                websocket = new SockJS("http://localhost:8080/<span style="font-family: Arial, Helvetica, sans-serif;">项目名称</span><span style="font-family: Arial, Helvetica, sans-serif;">/sockjs/websocket");</span>  
            }  
            websocket.onopen = function (evnt) {  
                $("#tou").html("链接服务器成功!")  
            };  
            websocket.onmessage = function (evnt) {  
                $("#msg").html($("#msg").html() + "<br/>" + evnt.data);  
            };  
            websocket.onerror = function (evnt) {  
            };  
            websocket.onclose = function (evnt) {  
                $("#tou").html("与服务器断开了链接!")  
            }  
            $('#send').bind('click', function() {  
                send();  
            });  
            function send(){  
                if (websocket != null) {  
                    var message = document.getElementById('message').value;  
                    websocket.send(message);  
                } else {  
                    alert('未与服务器链接.');  
                }  
            }  
        });  
    </script> 

可能出现的错误：

Async support must be enabled on a servlet and for all filters involved in async request processing

 

如果直接运行，基本上会包上面的错误，我们还需要在web.xml中做如下的配置

<web-app version="3.0"   
xmlns="http://java.sun.com/xml/ns/javaee"   
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"   
xsi:schemaLocation="http://java.sun.com/xml/ns/javaee   
http://java.sun.com/xml/ns/javaee/web-app_3_0.xsd">  

但是2.5的webapp也可以

然后在这个servlet中加入：

<async-supported>true</async-supported>  

如：

<servlet>  
    <servlet-name>presto-engine</servlet-name>  
    <servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>  
    <init-param>  
        <param-name>contextConfigLocation</param-name>  
        <param-value>classpath:presto_servlet.xml</param-value>  
    </init-param>  
    <load-on-startup>1</load-on-startup>  
    <async-supported>true</async-supported>  
</servlet> 

 

 

The extension [x-webkit-deflate-frame] is not supported

在HandshakeInterceptor的beforeHandshake方法中加入

 

public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map<String, Object> map) throws Exception {  
    /** 
     * This is a bug,bug fix:The extension [x-webkit-deflate-frame] is not supported 
     */  
    if(request.getHeaders().containsKey("Sec-WebSocket-Extensions")) {  
        request.getHeaders().set("Sec-WebSocket-Extensions", "permessage-deflate");  
      }  
      return true;  
  }

转自：http://blog.csdn.net/ouyang111222/article/details/50545411