root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Add assert AbstractStandardRequestUpgradeStrategy

This commit is contained in:
Rossen Stoyanchev 2013-12-02 13:59:17 -05:00
parent 10f5d96a78
commit 079fb2db73
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spring-websocket/src/main/java/org/springframework/web/socket/server/support/AbstractStandardUpgradeStrategy.java
View File

@ -32,6 +32,7 @@ import org.springframework.http.HttpHeaders;
import org.springframework.http.server.ServerHttpRequest; import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse; import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest; import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.util.Assert;
import org.springframework.web.socket.support.WebSocketExtension; import org.springframework.web.socket.support.WebSocketExtension;
import org.springframework.web.socket.WebSocketHandler; import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.adapter.StandardWebSocketHandlerAdapter; import org.springframework.web.socket.adapter.StandardWebSocketHandlerAdapter;
@ -64,7 +65,11 @@ public abstract class AbstractStandardUpgradeStrategy implements RequestUpgradeS
protected ServerContainer getContainer(HttpServletRequest request) { protected ServerContainer getContainer(HttpServletRequest request) {
ServletContext servletContext = request.getServletContext(); ServletContext servletContext = request.getServletContext();
return (ServerContainer) servletContext.getAttribute("javax.websocket.server.ServerContainer"); String attrName = "javax.websocket.server.ServerContainer";
ServerContainer container = (ServerContainer) servletContext.getAttribute(attrName);
Assert.notNull(container, "No 'javax.websocket.server.ServerContainer' ServletContext attribute. " +
"Are you running in a Servlet container that supports JSR-356?");
return container;
} }
protected List<WebSocketExtension> getInstalledExtensions(WebSocketContainer container) { protected List<WebSocketExtension> getInstalledExtensions(WebSocketContainer container) {