Update TomcatRequestUpgradeStrategy for 8.0 RC1

While RC2 is not yet out, we need to support RC1 as well.
This change introduces temporary code that can be removed
once RC2 becomes available to avoid having to use Tomcat
snapshots.

Author: rstoyanchev

spring-websocket/src/main/java/org/springframework/web/socket/server/support/TomcatRequestUpgradeStrategy.java

@@ -17,6 +17,7 @@
 package org.springframework.web.socket.server.support;
 
 import java.io.IOException;
+import java.lang.reflect.Method;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.Map;
@@ -26,13 +27,17 @@
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import javax.websocket.Endpoint;
+import javax.websocket.server.ServerEndpointConfig;
 
+import org.apache.tomcat.websocket.server.WsHandshakeRequest;
+import org.apache.tomcat.websocket.server.WsHttpUpgradeHandler;
 import org.apache.tomcat.websocket.server.WsServerContainer;
 import org.springframework.http.server.ServerHttpRequest;
 import org.springframework.http.server.ServerHttpResponse;
 import org.springframework.http.server.ServletServerHttpRequest;
 import org.springframework.http.server.ServletServerHttpResponse;
 import org.springframework.util.Assert;
+import org.springframework.util.ReflectionUtils;
 import org.springframework.web.socket.server.HandshakeFailureException;
 import org.springframework.web.socket.server.endpoint.ServerEndpointRegistration;
 
@@ -60,6 +65,18 @@ public void upgradeInternal(ServerHttpRequest request, ServerHttpResponse respon
 		Assert.isTrue(response instanceof ServletServerHttpResponse);
 		HttpServletResponse servletResponse = ((ServletServerHttpResponse) response).getServletResponse();
 
+		if (hasDoUpgrade) {
+			doUpgrade(servletRequest, servletResponse, acceptedProtocol, endpoint);
+		}
+		else {
+			upgradeTomcat80RC1(servletRequest, acceptedProtocol, endpoint);
+		}
+	}
+
+	private void doUpgrade(HttpServletRequest servletRequest, HttpServletResponse servletResponse,
+			String acceptedProtocol, Endpoint endpoint) {
+
+		StringBuffer requestUrl = servletRequest.getRequestURL();
 		String path = servletRequest.getRequestURI(); // shouldn't matter
 		Map<String, String> pathParams = Collections.<String, String> emptyMap();
 
@@ -71,11 +88,11 @@ public void upgradeInternal(ServerHttpRequest request, ServerHttpResponse respon
 		}
 		catch (ServletException ex) {
 			throw new HandshakeFailureException(
-					"Servlet request failed to upgrade to WebSocket, uri=" + request.getURI(), ex);
+					"Servlet request failed to upgrade to WebSocket, uri=" + requestUrl, ex);
 		}
 		catch (IOException ex) {
 			throw new HandshakeFailureException(
-					"Response update failed during upgrade to WebSocket, uri=" + request.getURI(), ex);
+					"Response update failed during upgrade to WebSocket, uri=" + requestUrl, ex);
 		}
 	}
 
@@ -85,4 +102,36 @@ private WsServerContainer getContainer(HttpServletRequest servletRequest) {
 		return (WsServerContainer) servletContext.getAttribute(attribute);
 	}
 
-}
+	// FIXME: Remove this after RC2 is out
+
+	private void upgradeTomcat80RC1(HttpServletRequest request, String protocol, Endpoint endpoint) {
+
+		WsHttpUpgradeHandler upgradeHandler;
+		try {
+			upgradeHandler = request.upgrade(WsHttpUpgradeHandler.class);
+		}
+		catch (Exception e) {
+			throw new HandshakeFailureException("Unable to create UpgardeHandler", e);
+		}
+
+		WsHandshakeRequest webSocketRequest = new WsHandshakeRequest(request);
+		try {
+			Method method = ReflectionUtils.findMethod(WsHandshakeRequest.class, "finished");
+			ReflectionUtils.makeAccessible(method);
+			method.invoke(webSocketRequest);
+		}
+		catch (Exception ex) {
+			throw new HandshakeFailureException("Failed to upgrade HttpServletRequest", ex);
+		}
+
+		ServerEndpointConfig endpointConfig = new ServerEndpointRegistration("/shouldntmatter", endpoint);
+
+		upgradeHandler.preInit(endpoint, endpointConfig, getContainer(request), webSocketRequest,
+				protocol, Collections.<String, String> emptyMap(), request.isSecure());
+	}
+
+	private static boolean hasDoUpgrade = (ReflectionUtils.findMethod(WsServerContainer.class,
+			"doUpgrade", HttpServletRequest.class, HttpServletResponse.class,
+			ServerEndpointConfig.class, Map.class) != null);
+
+}