Skip to content

S
spring-cloud-netflix
Commit 053296de by Dave Syer
Options

Ensure request body is not buffered when direct to ZuulServlet

A change in Zuul in NetflixOSS means that it buffers all requests unless you switch it off at the level of the servlet. And even when you do that you need to ensure that Spring Cloud doesn't re-introduce a buffer through the Servlet30RequestWrapper. This change restores the feature in Spring Cloud Angel that multipart files posted to /zuul/* go straight through the servlet and are not buffered in memory. Fixes gh-773
parent 249f524a
Showing with 127 additions and 77 deletions
spring-cloud-netflix-core/src/main/java/org/springframework/cloud/netflix/zuul/ZuulConfiguration.java
...@@ -95,9 +95,14 @@ public class ZuulConfiguration { ...@@ -95,9 +95,14 @@ public class ZuulConfiguration {
} }
@Bean @Bean
@ConditionalOnMissingBean(name = "zuulServlet")
public ServletRegistrationBean zuulServlet() { public ServletRegistrationBean zuulServlet() {
return new ServletRegistrationBean(new ZuulServlet(), ServletRegistrationBean servlet = new ServletRegistrationBean(new ZuulServlet(),
this.zuulProperties.getServletPattern()); this.zuulProperties.getServletPattern());
// The whole point of exposing this servlet is to provide a route that doesn't
// buffer requests.
servlet.addInitParameter("buffer-requests", "false");
return servlet;
} }
// pre filters // pre filters
......
spring-cloud-netflix-core/src/main/java/org/springframework/cloud/netflix/zuul/filters/ProxyRequestHelper.java
...@@ -41,11 +41,11 @@ import org.springframework.web.util.WebUtils; ...@@ -41,11 +41,11 @@ import org.springframework.web.util.WebUtils;
import com.netflix.zuul.context.RequestContext; import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.util.HTTPRequestUtils; import com.netflix.zuul.util.HTTPRequestUtils;
import lombok.extern.apachecommons.CommonsLog;
import static org.springframework.http.HttpHeaders.CONTENT_ENCODING; import static org.springframework.http.HttpHeaders.CONTENT_ENCODING;
import static org.springframework.http.HttpHeaders.CONTENT_LENGTH; import static org.springframework.http.HttpHeaders.CONTENT_LENGTH;
import lombok.extern.apachecommons.CommonsLog;
/** /**
* @author Dave Syer * @author Dave Syer
*/ */
...@@ -230,7 +230,8 @@ public class ProxyRequestHelper { ...@@ -230,7 +230,8 @@ public class ProxyRequestHelper {
input.put(entry.getKey(), value); input.put(entry.getKey(), value);
} }
RequestContext ctx = RequestContext.getCurrentContext(); RequestContext ctx = RequestContext.getCurrentContext();
if (!ctx.isChunkedRequestBody()) { if (shouldDebugBody(ctx)) {
// Prevent input stream from being read if it needs to go downstream
if (requestEntity != null) { if (requestEntity != null) {
debugRequestEntity(info, ctx.getRequest().getInputStream()); debugRequestEntity(info, ctx.getRequest().getInputStream());
} }
...@@ -241,6 +242,17 @@ public class ProxyRequestHelper { ...@@ -241,6 +242,17 @@ public class ProxyRequestHelper {
return info; return info;
} }
private boolean shouldDebugBody(RequestContext ctx) {
HttpServletRequest request = ctx.getRequest();
if (ctx.isChunkedRequestBody()) {
return false;
}
if (request == null || request.getContentType() == null) {
return true;
}
return !request.getContentType().toLowerCase().contains("multipart");
}
public void appendDebug(Map<String, Object> info, int status, public void appendDebug(Map<String, Object> info, int status,
MultiValueMap<String, String> headers) { MultiValueMap<String, String> headers) {
if (this.traces != null) { if (this.traces != null) {
...@@ -276,4 +288,3 @@ public class ProxyRequestHelper { ...@@ -276,4 +288,3 @@ public class ProxyRequestHelper {
} }
} }
spring-cloud-netflix-core/src/main/java/org/springframework/cloud/netflix/zuul/filters/pre/Servlet30WrapperFilter.java
...@@ -22,6 +22,7 @@ import javax.servlet.http.HttpServletRequest; ...@@ -22,6 +22,7 @@ import javax.servlet.http.HttpServletRequest;
import org.springframework.util.Assert; import org.springframework.util.Assert;
import org.springframework.util.ReflectionUtils; import org.springframework.util.ReflectionUtils;
import org.springframework.web.servlet.DispatcherServlet;
import com.netflix.zuul.ZuulFilter; import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext; import com.netflix.zuul.context.RequestContext;
...@@ -37,7 +38,8 @@ public class Servlet30WrapperFilter extends ZuulFilter { ...@@ -37,7 +38,8 @@ public class Servlet30WrapperFilter extends ZuulFilter {
public Servlet30WrapperFilter() { public Servlet30WrapperFilter() {
this.requestField = ReflectionUtils.findField(HttpServletRequestWrapper.class, this.requestField = ReflectionUtils.findField(HttpServletRequestWrapper.class,
"req", HttpServletRequest.class); "req", HttpServletRequest.class);
Assert.notNull(this.requestField, "HttpServletRequestWrapper.req field not found"); Assert.notNull(this.requestField,
"HttpServletRequestWrapper.req field not found");
this.requestField.setAccessible(true); this.requestField.setAccessible(true);
} }
...@@ -67,9 +69,18 @@ public class Servlet30WrapperFilter extends ZuulFilter { ...@@ -67,9 +69,18 @@ public class Servlet30WrapperFilter extends ZuulFilter {
if (request instanceof HttpServletRequestWrapper) { if (request instanceof HttpServletRequestWrapper) {
request = (HttpServletRequest) ReflectionUtils.getField(this.requestField, request = (HttpServletRequest) ReflectionUtils.getField(this.requestField,
request); request);
ctx.setRequest(new Servlet30RequestWrapper(request));
} }
else if (isDispatcherServletRequest(request)) {
// If it's going through the dispatcher we need to buffer the body
ctx.setRequest(new Servlet30RequestWrapper(request)); ctx.setRequest(new Servlet30RequestWrapper(request));
}
return null; return null;
} }
private boolean isDispatcherServletRequest(HttpServletRequest request) {
return request.getAttribute(
DispatcherServlet.WEB_APPLICATION_CONTEXT_ATTRIBUTE) != null;
}
} }
spring-cloud-netflix-core/src/main/java/org/springframework/cloud/netflix/zuul/filters/route/RibbonRoutingFilter.java
...@@ -23,8 +23,6 @@ import java.util.Map; ...@@ -23,8 +23,6 @@ import java.util.Map;
import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponse;
import lombok.extern.apachecommons.CommonsLog;
import org.springframework.cloud.netflix.zuul.filters.ProxyRequestHelper; import org.springframework.cloud.netflix.zuul.filters.ProxyRequestHelper;
import org.springframework.http.client.ClientHttpResponse; import org.springframework.http.client.ClientHttpResponse;
import org.springframework.util.MultiValueMap; import org.springframework.util.MultiValueMap;
...@@ -35,6 +33,8 @@ import com.netflix.zuul.ZuulFilter; ...@@ -35,6 +33,8 @@ import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext; import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException; import com.netflix.zuul.exception.ZuulException;
import lombok.extern.apachecommons.CommonsLog;
@CommonsLog @CommonsLog
public class RibbonRoutingFilter extends ZuulFilter { public class RibbonRoutingFilter extends ZuulFilter {
...@@ -64,8 +64,8 @@ public class RibbonRoutingFilter extends ZuulFilter { ...@@ -64,8 +64,8 @@ public class RibbonRoutingFilter extends ZuulFilter {
@Override @Override
public boolean shouldFilter() { public boolean shouldFilter() {
RequestContext ctx = RequestContext.getCurrentContext(); RequestContext ctx = RequestContext.getCurrentContext();
return (ctx.getRouteHost() == null && ctx.get("serviceId") != null && ctx return (ctx.getRouteHost() == null && ctx.get("serviceId") != null
.sendZuulResponse()); && ctx.sendZuulResponse());
} }
@Override @Override
...@@ -78,7 +78,8 @@ public class RibbonRoutingFilter extends ZuulFilter { ...@@ -78,7 +78,8 @@ public class RibbonRoutingFilter extends ZuulFilter {
return response; return response;
} }
catch (Exception ex) { catch (Exception ex) {
context.set("error.status_code", HttpServletResponse.SC_INTERNAL_SERVER_ERROR); context.set("error.status_code",
HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
context.set("error.exception", ex); context.set("error.exception", ex);
} }
return null; return null;
...@@ -93,6 +94,9 @@ public class RibbonRoutingFilter extends ZuulFilter { ...@@ -93,6 +94,9 @@ public class RibbonRoutingFilter extends ZuulFilter {
.buildZuulRequestQueryParams(request); .buildZuulRequestQueryParams(request);
String verb = getVerb(request); String verb = getVerb(request);
InputStream requestEntity = getRequestBody(request); InputStream requestEntity = getRequestBody(request);
if (request.getContentLength() < 0) {
context.setChunkedRequestBody();
}
String serviceId = (String) context.get("serviceId"); String serviceId = (String) context.get("serviceId");
Boolean retryable = (Boolean) context.get("retryable"); Boolean retryable = (Boolean) context.get("retryable");
...@@ -102,15 +106,15 @@ public class RibbonRoutingFilter extends ZuulFilter { ...@@ -102,15 +106,15 @@ public class RibbonRoutingFilter extends ZuulFilter {
// remove double slashes // remove double slashes
uri = uri.replace("//", "/"); uri = uri.replace("//", "/");
return new RibbonCommandContext(serviceId, verb, uri, retryable, return new RibbonCommandContext(serviceId, verb, uri, retryable, headers, params,
headers, params, requestEntity); requestEntity);
} }
private ClientHttpResponse forward(RibbonCommandContext context) throws Exception { private ClientHttpResponse forward(RibbonCommandContext context) throws Exception {
Map<String, Object> info = this.helper.debug(context.getVerb(), context.getUri(), Map<String, Object> info = this.helper.debug(context.getVerb(), context.getUri(),
context.getHeaders(), context.getParams(), context.getRequestEntity()); context.getHeaders(), context.getParams(), context.getRequestEntity());
RibbonCommand command = ribbonCommandFactory.create(context); RibbonCommand command = this.ribbonCommandFactory.create(context);
try { try {
ClientHttpResponse response = command.execute(); ClientHttpResponse response = command.execute();
this.helper.appendDebug(info, response.getStatusCode().value(), this.helper.appendDebug(info, response.getStatusCode().value(),
...@@ -124,11 +128,11 @@ public class RibbonRoutingFilter extends ZuulFilter { ...@@ -124,11 +128,11 @@ public class RibbonRoutingFilter extends ZuulFilter {
&& ex.getFallbackException().getCause() instanceof ClientException) { && ex.getFallbackException().getCause() instanceof ClientException) {
ClientException cause = (ClientException) ex.getFallbackException() ClientException cause = (ClientException) ex.getFallbackException()
.getCause(); .getCause();
throw new ZuulException(cause, "Forwarding error", 500, cause throw new ZuulException(cause, "Forwarding error", 500,
.getErrorType().toString()); cause.getErrorType().toString());
} }
throw new ZuulException(ex, "Forwarding error", 500, ex.getFailureType() throw new ZuulException(ex, "Forwarding error", 500,
.toString()); ex.getFailureType().toString());
} }
} }
...@@ -140,8 +144,8 @@ public class RibbonRoutingFilter extends ZuulFilter { ...@@ -140,8 +144,8 @@ public class RibbonRoutingFilter extends ZuulFilter {
return null; return null;
} }
try { try {
requestEntity = (InputStream) RequestContext.getCurrentContext().get( requestEntity = (InputStream) RequestContext.getCurrentContext()
"requestEntity"); .get("requestEntity");
if (requestEntity == null) { if (requestEntity == null) {
requestEntity = request.getInputStream(); requestEntity = request.getInputStream();
} }
...@@ -154,12 +158,14 @@ public class RibbonRoutingFilter extends ZuulFilter { ...@@ -154,12 +158,14 @@ public class RibbonRoutingFilter extends ZuulFilter {
private String getVerb(HttpServletRequest request) { private String getVerb(HttpServletRequest request) {
String method = request.getMethod(); String method = request.getMethod();
if (method == null) if (method == null) {
return "GET"; return "GET";
}
return method; return method;
} }
private void setResponse(ClientHttpResponse resp) throws ClientException, IOException { private void setResponse(ClientHttpResponse resp)
throws ClientException, IOException {
this.helper.setResponse(resp.getStatusCode().value(), this.helper.setResponse(resp.getStatusCode().value(),
resp.getBody() == null ? null : resp.getBody(), resp.getHeaders()); resp.getBody() == null ? null : resp.getBody(), resp.getHeaders());
} }
......
spring-cloud-netflix-core/src/main/java/org/springframework/cloud/netflix/zuul/filters/route/SimpleHostRoutingFilter.java
...@@ -29,6 +29,7 @@ import java.util.List; ...@@ -29,6 +29,7 @@ import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Timer; import java.util.Timer;
import java.util.TimerTask; import java.util.TimerTask;
import javax.annotation.PostConstruct; import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy; import javax.annotation.PreDestroy;
import javax.net.ssl.SSLContext; import javax.net.ssl.SSLContext;
...@@ -55,6 +56,7 @@ import org.apache.http.config.RegistryBuilder; ...@@ -55,6 +56,7 @@ import org.apache.http.config.RegistryBuilder;
import org.apache.http.conn.socket.ConnectionSocketFactory; import org.apache.http.conn.socket.ConnectionSocketFactory;
import org.apache.http.conn.socket.PlainConnectionSocketFactory; import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory; import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.InputStreamEntity; import org.apache.http.entity.InputStreamEntity;
import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.DefaultHttpRequestRetryHandler; import org.apache.http.impl.client.DefaultHttpRequestRetryHandler;
...@@ -80,12 +82,12 @@ import lombok.extern.apachecommons.CommonsLog; ...@@ -80,12 +82,12 @@ import lombok.extern.apachecommons.CommonsLog;
public class SimpleHostRoutingFilter extends ZuulFilter { public class SimpleHostRoutingFilter extends ZuulFilter {
private static final DynamicIntProperty SOCKET_TIMEOUT = DynamicPropertyFactory private static final DynamicIntProperty SOCKET_TIMEOUT = DynamicPropertyFactory
.getInstance().getIntProperty(ZuulConstants.ZUUL_HOST_SOCKET_TIMEOUT_MILLIS, .getInstance()
10000); .getIntProperty(ZuulConstants.ZUUL_HOST_SOCKET_TIMEOUT_MILLIS, 10000);
private static final DynamicIntProperty CONNECTION_TIMEOUT = DynamicPropertyFactory private static final DynamicIntProperty CONNECTION_TIMEOUT = DynamicPropertyFactory
.getInstance().getIntProperty(ZuulConstants.ZUUL_HOST_CONNECT_TIMEOUT_MILLIS, .getInstance()
2000); .getIntProperty(ZuulConstants.ZUUL_HOST_CONNECT_TIMEOUT_MILLIS, 2000);
private final Timer connectionManagerTimer = new Timer( private final Timer connectionManagerTimer = new Timer(
"SimpleHostRoutingFilter.connectionManagerTimer", true); "SimpleHostRoutingFilter.connectionManagerTimer", true);
...@@ -98,11 +100,12 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -98,11 +100,12 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
@Override @Override
public void run() { public void run() {
try { try {
httpClient.close(); SimpleHostRoutingFilter.this.httpClient.close();
} catch (IOException ex) { }
catch (IOException ex) {
log.error("error closing client", ex); log.error("error closing client", ex);
} }
httpClient = newClient(); SimpleHostRoutingFilter.this.httpClient = newClient();
} }
}; };
...@@ -117,22 +120,22 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -117,22 +120,22 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
@PostConstruct @PostConstruct
private void initialize() { private void initialize() {
this.httpClient = newClient(); this.httpClient = newClient();
SOCKET_TIMEOUT.addCallback(clientloader); SOCKET_TIMEOUT.addCallback(this.clientloader);
CONNECTION_TIMEOUT.addCallback(clientloader); CONNECTION_TIMEOUT.addCallback(this.clientloader);
connectionManagerTimer.schedule(new TimerTask() { this.connectionManagerTimer.schedule(new TimerTask() {
@Override @Override
public void run() { public void run() {
if (connectionManager == null) { if (SimpleHostRoutingFilter.this.connectionManager == null) {
return; return;
} }
connectionManager.closeExpiredConnections(); SimpleHostRoutingFilter.this.connectionManager.closeExpiredConnections();
} }
}, 30000, 5000); }, 30000, 5000);
} }
@PreDestroy @PreDestroy
public void stop() { public void stop() {
connectionManagerTimer.cancel(); this.connectionManagerTimer.cancel();
} }
@Override @Override
...@@ -161,16 +164,20 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -161,16 +164,20 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
.buildZuulRequestQueryParams(request); .buildZuulRequestQueryParams(request);
String verb = getVerb(request); String verb = getVerb(request);
InputStream requestEntity = getRequestBody(request); InputStream requestEntity = getRequestBody(request);
if (request.getContentLength() < 0) {
context.setChunkedRequestBody();
}
String uri = this.helper.buildZuulRequestURI(request); String uri = this.helper.buildZuulRequestURI(request);
try { try {
HttpResponse response = forward(httpClient, verb, uri, request, headers, HttpResponse response = forward(this.httpClient, verb, uri, request, headers,
params, requestEntity); params, requestEntity);
setResponse(response); setResponse(response);
} }
catch (Exception ex) { catch (Exception ex) {
context.set("error.status_code", HttpServletResponse.SC_INTERNAL_SERVER_ERROR); context.set("error.status_code",
HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
context.set("error.exception", ex); context.set("error.exception", ex);
} }
return null; return null;
...@@ -179,31 +186,37 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -179,31 +186,37 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
protected PoolingHttpClientConnectionManager newConnectionManager() { protected PoolingHttpClientConnectionManager newConnectionManager() {
try { try {
final SSLContext sslContext = SSLContext.getInstance("SSL"); final SSLContext sslContext = SSLContext.getInstance("SSL");
sslContext.init(null, new TrustManager[]{new X509TrustManager() { sslContext.init(null, new TrustManager[] { new X509TrustManager() {
@Override @Override
public void checkClientTrusted(X509Certificate[] x509Certificates, String s) throws CertificateException { public void checkClientTrusted(X509Certificate[] x509Certificates,
String s) throws CertificateException {
} }
@Override @Override
public void checkServerTrusted(X509Certificate[] x509Certificates, String s) throws CertificateException { public void checkServerTrusted(X509Certificate[] x509Certificates,
String s) throws CertificateException {
} }
@Override @Override
public X509Certificate[] getAcceptedIssuers() { public X509Certificate[] getAcceptedIssuers() {
return null; return null;
} }
}}, new SecureRandom()); } }, new SecureRandom());
final Registry<ConnectionSocketFactory> registry = RegistryBuilder.<ConnectionSocketFactory>create() final Registry<ConnectionSocketFactory> registry = RegistryBuilder
.<ConnectionSocketFactory> create()
.register("http", PlainConnectionSocketFactory.INSTANCE) .register("http", PlainConnectionSocketFactory.INSTANCE)
.register("https", new SSLConnectionSocketFactory(sslContext)) .register("https", new SSLConnectionSocketFactory(sslContext))
.build(); .build();
connectionManager = new PoolingHttpClientConnectionManager(registry); this.connectionManager = new PoolingHttpClientConnectionManager(registry);
connectionManager.setMaxTotal(Integer.parseInt(System.getProperty("zuul.max.host.connections", "200"))); this.connectionManager.setMaxTotal(Integer
connectionManager.setDefaultMaxPerRoute(Integer.parseInt(System.getProperty("zuul.max.host.connections", "20"))); .parseInt(System.getProperty("zuul.max.host.connections", "200")));
return connectionManager; this.connectionManager.setDefaultMaxPerRoute(Integer
} catch (Exception ex) { .parseInt(System.getProperty("zuul.max.host.connections", "20")));
return this.connectionManager;
}
catch (Exception ex) {
throw new RuntimeException(ex); throw new RuntimeException(ex);
} }
} }
...@@ -212,25 +225,26 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -212,25 +225,26 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
final RequestConfig requestConfig = RequestConfig.custom() final RequestConfig requestConfig = RequestConfig.custom()
.setSocketTimeout(SOCKET_TIMEOUT.get()) .setSocketTimeout(SOCKET_TIMEOUT.get())
.setConnectTimeout(CONNECTION_TIMEOUT.get()) .setConnectTimeout(CONNECTION_TIMEOUT.get())
.setCookieSpec(CookieSpecs.IGNORE_COOKIES) .setCookieSpec(CookieSpecs.IGNORE_COOKIES).build();
.build();
return HttpClients.custom() return HttpClients.custom().setConnectionManager(newConnectionManager())
.setConnectionManager(newConnectionManager())
.setDefaultRequestConfig(requestConfig) .setDefaultRequestConfig(requestConfig)
.setRetryHandler(new DefaultHttpRequestRetryHandler(0, false)) .setRetryHandler(new DefaultHttpRequestRetryHandler(0, false))
.setRedirectStrategy(new RedirectStrategy() { .setRedirectStrategy(new RedirectStrategy() {
@Override @Override
public boolean isRedirected(HttpRequest request, HttpResponse response, HttpContext context) throws ProtocolException { public boolean isRedirected(HttpRequest request,
HttpResponse response, HttpContext context)
throws ProtocolException {
return false; return false;
} }
@Override @Override
public HttpUriRequest getRedirect(HttpRequest request, HttpResponse response, HttpContext context) throws ProtocolException { public HttpUriRequest getRedirect(HttpRequest request,
HttpResponse response, HttpContext context)
throws ProtocolException {
return null; return null;
} }
}) }).build();
.build();
} }
private HttpResponse forward(HttpClient httpclient, String verb, String uri, private HttpResponse forward(HttpClient httpclient, String verb, String uri,
...@@ -243,24 +257,24 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -243,24 +257,24 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
HttpHost httpHost = getHttpHost(host); HttpHost httpHost = getHttpHost(host);
uri = StringUtils.cleanPath((host.getPath() + uri).replaceAll("/{2,}", "/")); uri = StringUtils.cleanPath((host.getPath() + uri).replaceAll("/{2,}", "/"));
HttpRequest httpRequest; HttpRequest httpRequest;
int contentLength = request.getContentLength();
InputStreamEntity entity = new InputStreamEntity(requestEntity, contentLength,
ContentType.create(request.getContentType()));
switch (verb.toUpperCase()) { switch (verb.toUpperCase()) {
case "POST": case "POST":
HttpPost httpPost = new HttpPost(uri + getQueryString()); HttpPost httpPost = new HttpPost(uri + getQueryString());
httpRequest = httpPost; httpRequest = httpPost;
httpPost.setEntity(new InputStreamEntity(requestEntity, request httpPost.setEntity(entity);
.getContentLength()));
break; break;
case "PUT": case "PUT":
HttpPut httpPut = new HttpPut(uri + getQueryString()); HttpPut httpPut = new HttpPut(uri + getQueryString());
httpRequest = httpPut; httpRequest = httpPut;
httpPut.setEntity(new InputStreamEntity(requestEntity, request httpPut.setEntity(entity);
.getContentLength()));
break; break;
case "PATCH": case "PATCH":
HttpPatch httpPatch = new HttpPatch(uri + getQueryString()); HttpPatch httpPatch = new HttpPatch(uri + getQueryString());
httpRequest = httpPatch; httpRequest = httpPatch;
httpPatch.setEntity(new InputStreamEntity(requestEntity, request httpPatch.setEntity(entity);
.getContentLength()));
break; break;
default: default:
httpRequest = new BasicHttpRequest(verb, uri + getQueryString()); httpRequest = new BasicHttpRequest(verb, uri + getQueryString());
...@@ -312,10 +326,11 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -312,10 +326,11 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
private String getQueryString() throws UnsupportedEncodingException { private String getQueryString() throws UnsupportedEncodingException {
HttpServletRequest request = RequestContext.getCurrentContext().getRequest(); HttpServletRequest request = RequestContext.getCurrentContext().getRequest();
MultiValueMap<String, String> params=helper.buildZuulRequestQueryParams(request); MultiValueMap<String, String> params = this.helper
StringBuilder query=new StringBuilder(); .buildZuulRequestQueryParams(request);
StringBuilder query = new StringBuilder();
for (Map.Entry<String, List<String>> entry : params.entrySet()) { for (Map.Entry<String, List<String>> entry : params.entrySet()) {
String key=URLEncoder.encode(entry.getKey(), "UTF-8"); String key = URLEncoder.encode(entry.getKey(), "UTF-8");
for (String value : entry.getValue()) { for (String value : entry.getValue()) {
query.append("&"); query.append("&");
query.append(key); query.append(key);
...@@ -323,7 +338,7 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -323,7 +338,7 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
query.append(URLEncoder.encode(value, "UTF-8")); query.append(URLEncoder.encode(value, "UTF-8"));
} }
} }
return (query.length()>0) ? "?" + query.substring(1) : ""; return (query.length() > 0) ? "?" + query.substring(1) : "";
} }
private HttpHost getHttpHost(URL host) { private HttpHost getHttpHost(URL host) {
...@@ -359,7 +374,7 @@ public class SimpleHostRoutingFilter extends ZuulFilter { ...@@ -359,7 +374,7 @@ public class SimpleHostRoutingFilter extends ZuulFilter {
* @param names * @param names
*/ */
protected void addIgnoredHeaders(String... names) { protected void addIgnoredHeaders(String... names) {
helper.addIgnoredHeaders(names); this.helper.addIgnoredHeaders(names);
} }
} }
spring-cloud-netflix-core/src/test/java/org/springframework/cloud/netflix/zuul/FormZuulServletProxyApplicationTests.java
...@@ -16,14 +16,10 @@ ...@@ -16,14 +16,10 @@
package org.springframework.cloud.netflix.zuul; package org.springframework.cloud.netflix.zuul;
import static org.junit.Assert.assertEquals;
import java.io.IOException; import java.io.IOException;
import java.io.InputStream; import java.io.InputStream;
import java.util.Map; import java.util.Map;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
...@@ -60,6 +56,10 @@ import com.netflix.loadbalancer.Server; ...@@ -60,6 +56,10 @@ import com.netflix.loadbalancer.Server;
import com.netflix.loadbalancer.ServerList; import com.netflix.loadbalancer.ServerList;
import com.netflix.zuul.ZuulFilter; import com.netflix.zuul.ZuulFilter;
import static org.junit.Assert.assertEquals;
import lombok.extern.slf4j.Slf4j;
@RunWith(SpringJUnit4ClassRunner.class) @RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = FormZuulServletProxyApplication.class) @SpringApplicationConfiguration(classes = FormZuulServletProxyApplication.class)
@WebAppConfiguration @WebAppConfiguration
...@@ -107,6 +107,8 @@ public class FormZuulServletProxyApplicationTests { ...@@ -107,6 +107,8 @@ public class FormZuulServletProxyApplicationTests {
form.set("foo", new HttpEntity<byte[]>("bar".getBytes(), part)); form.set("foo", new HttpEntity<byte[]>("bar".getBytes(), part));
HttpHeaders headers = new HttpHeaders(); HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.MULTIPART_FORM_DATA); headers.setContentType(MediaType.MULTIPART_FORM_DATA);
headers.set("Transfer-Encoding", "chunked");
headers.setContentLength(-1);
ResponseEntity<String> result = new TestRestTemplate().exchange( ResponseEntity<String> result = new TestRestTemplate().exchange(
"http://localhost:" + this.port + "/zuul/simple/file", HttpMethod.POST, "http://localhost:" + this.port + "/zuul/simple/file", HttpMethod.POST,
new HttpEntity<MultiValueMap<String, Object>>(form, headers), new HttpEntity<MultiValueMap<String, Object>>(form, headers),
...@@ -120,8 +122,8 @@ public class FormZuulServletProxyApplicationTests { ...@@ -120,8 +122,8 @@ public class FormZuulServletProxyApplicationTests {
MultiValueMap<String, String> form = new LinkedMultiValueMap<String, String>(); MultiValueMap<String, String> form = new LinkedMultiValueMap<String, String>();
form.set("foo", "bar"); form.set("foo", "bar");
HttpHeaders headers = new HttpHeaders(); HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType headers.setContentType(MediaType.valueOf(
.valueOf(MediaType.APPLICATION_FORM_URLENCODED_VALUE + "; charset=UTF-8")); MediaType.APPLICATION_FORM_URLENCODED_VALUE + "; charset=UTF-8"));
ResponseEntity<String> result = new TestRestTemplate().exchange( ResponseEntity<String> result = new TestRestTemplate().exchange(
"http://localhost:" + this.port + "/zuul/simple/form", HttpMethod.POST, "http://localhost:" + this.port + "/zuul/simple/form", HttpMethod.POST,
new HttpEntity<MultiValueMap<String, String>>(form, headers), new HttpEntity<MultiValueMap<String, String>>(form, headers),
...@@ -209,12 +211,12 @@ class FormZuulServletProxyApplication { ...@@ -209,12 +211,12 @@ class FormZuulServletProxyApplication {
} }
public static void main(String[] args) { public static void main(String[] args) {
new SpringApplicationBuilder(FormZuulProxyApplication.class).properties( new SpringApplicationBuilder(FormZuulProxyApplication.class)
"zuul.routes.simple:/zuul/simple/**", .properties("zuul.routes.simple:/zuul/simple/**",
"zuul.routes.direct.url:http://localhost:9999", "zuul.routes.direct.url:http://localhost:9999",
"zuul.routes.direct.path:/zuul/direct/**", "zuul.routes.direct.path:/zuul/direct/**",
"multipart.maxFileSize:4096MB", "multipart.maxRequestSize:4096MB").run( "multipart.maxFileSize:4096MB", "multipart.maxRequestSize:4096MB")
args); .run(args);
} }
} }
...@@ -228,7 +230,7 @@ class ServletFormRibbonClientConfiguration { ...@@ -228,7 +230,7 @@ class ServletFormRibbonClientConfiguration {
@Bean @Bean
public ServerList<Server> ribbonServerList() { public ServerList<Server> ribbonServerList() {
return new StaticServerList<>(new Server("localhost", port)); return new StaticServerList<>(new Server("localhost", this.port));
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment