Issue 62: always set Content-Length

git-svn-id: http://jclouds.googlecode.com/svn/trunk@1424 3d8758e0-26b5-11de-8745-db77d3ebf521
This commit is contained in:
adrian.f.cole 2009-06-11 23:20:37 +00:00
parent cd1c3ae900
commit e773803ee2
4 changed files with 189 additions and 185 deletions

View File

@ -34,6 +34,7 @@ import java.net.MalformedURLException;
import java.net.URL; import java.net.URL;
import org.apache.commons.io.IOUtils; import org.apache.commons.io.IOUtils;
import org.jclouds.http.HttpConstants;
import org.jclouds.http.HttpFutureCommand; import org.jclouds.http.HttpFutureCommand;
import org.jclouds.http.HttpFutureCommandClient; import org.jclouds.http.HttpFutureCommandClient;
import org.jclouds.http.HttpRequest; import org.jclouds.http.HttpRequest;
@ -50,8 +51,7 @@ import com.google.inject.Inject;
public class JavaUrlHttpFutureCommandClient extends BaseHttpFutureCommandClient { public class JavaUrlHttpFutureCommandClient extends BaseHttpFutureCommandClient {
@Inject @Inject
public JavaUrlHttpFutureCommandClient(URL target) public JavaUrlHttpFutureCommandClient(URL target) throws MalformedURLException {
throws MalformedURLException {
super(target); super(target);
} }
@ -66,9 +66,7 @@ public class JavaUrlHttpFutureCommandClient extends BaseHttpFutureCommandClient
} }
logger.trace("%1$s - converting request %2$s", target, request); logger.trace("%1$s - converting request %2$s", target, request);
connection = openJavaConnection(request); connection = openJavaConnection(request);
logger logger.trace("%1$s - submitting request %2$s", target, connection);
.trace("%1$s - submitting request %2$s", target,
connection);
response = getResponse(connection); response = getResponse(connection);
logger.trace("%1$s - received response %2$s", target, response); logger.trace("%1$s - received response %2$s", target, response);
if (response.getStatusCode() >= 500 && httpRetryHandler.retryRequest(command, response)) if (response.getStatusCode() >= 500 && httpRetryHandler.retryRequest(command, response))
@ -86,8 +84,7 @@ public class JavaUrlHttpFutureCommandClient extends BaseHttpFutureCommandClient
} }
} }
protected HttpResponse getResponse(HttpURLConnection connection) protected HttpResponse getResponse(HttpURLConnection connection) throws IOException {
throws IOException {
HttpResponse response = new HttpResponse(); HttpResponse response = new HttpResponse();
InputStream in; InputStream in;
try { try {
@ -100,16 +97,14 @@ public class JavaUrlHttpFutureCommandClient extends BaseHttpFutureCommandClient
} }
response.setStatusCode(connection.getResponseCode()); response.setStatusCode(connection.getResponseCode());
for (String header : connection.getHeaderFields().keySet()) { for (String header : connection.getHeaderFields().keySet()) {
response.getHeaders().putAll(header, response.getHeaders().putAll(header, connection.getHeaderFields().get(header));
connection.getHeaderFields().get(header));
} }
response.setMessage(connection.getResponseMessage()); response.setMessage(connection.getResponseMessage());
return response; return response;
} }
protected HttpURLConnection openJavaConnection(HttpRequest request) protected HttpURLConnection openJavaConnection(HttpRequest request) throws IOException {
throws IOException {
URL url = new URL(target, request.getUri()); URL url = new URL(target, request.getUri());
HttpURLConnection connection = (HttpURLConnection) url.openConnection(); HttpURLConnection connection = (HttpURLConnection) url.openConnection();
connection.setDoOutput(true); connection.setDoOutput(true);
@ -130,19 +125,19 @@ public class JavaUrlHttpFutureCommandClient extends BaseHttpFutureCommandClient
} else if (request.getPayload() instanceof InputStream) { } else if (request.getPayload() instanceof InputStream) {
IOUtils.copy((InputStream) request.getPayload(), out); IOUtils.copy((InputStream) request.getPayload(), out);
} else if (request.getPayload() instanceof File) { } else if (request.getPayload() instanceof File) {
IOUtils.copy(new FileInputStream((File) request IOUtils.copy(new FileInputStream((File) request.getPayload()), out);
.getPayload()), out);
} else if (request.getPayload() instanceof byte[]) { } else if (request.getPayload() instanceof byte[]) {
IOUtils.write((byte[]) request.getPayload(), out); IOUtils.write((byte[]) request.getPayload(), out);
} else { } else {
throw new UnsupportedOperationException( throw new UnsupportedOperationException("Content not supported "
"Content not supported "
+ request.getPayload().getClass()); + request.getPayload().getClass());
} }
} finally { } finally {
IOUtils.closeQuietly(out); IOUtils.closeQuietly(out);
} }
} else {
connection.setRequestProperty(HttpConstants.CONTENT_LENGTH, "0");
} }
return connection; return connection;
} }

View File

@ -65,13 +65,12 @@ public abstract class BaseJettyTest {
@BeforeTest @BeforeTest
@Parameters( { "test-jetty-port" }) @Parameters( { "test-jetty-port" })
public void setUpJetty(@Optional("8123") final int testPort) public void setUpJetty(@Optional("8123") final int testPort) throws Exception {
throws Exception {
Handler handler = new AbstractHandler() { Handler handler = new AbstractHandler() {
public void handle(String target, HttpServletRequest request, public void handle(String target, HttpServletRequest request,
HttpServletResponse response, int dispatch) HttpServletResponse response, int dispatch) throws IOException, ServletException {
throws IOException, ServletException { failIfNoContentLength(request, response);
if (request.getHeader("test") != null) { if (request.getHeader("test") != null) {
response.setContentType("text/plain"); response.setContentType("text/plain");
response.setStatus(HttpServletResponse.SC_OK); response.setStatus(HttpServletResponse.SC_OK);
@ -95,8 +94,7 @@ public abstract class BaseJettyTest {
properties.put(HttpConstants.PROPERTY_HTTP_PORT, testPort + ""); properties.put(HttpConstants.PROPERTY_HTTP_PORT, testPort + "");
properties.put(HttpConstants.PROPERTY_HTTP_SECURE, "false"); properties.put(HttpConstants.PROPERTY_HTTP_SECURE, "false");
addConnectionProperties(properties); addConnectionProperties(properties);
final List<HttpRequestFilter> filters = new ArrayList<HttpRequestFilter>( final List<HttpRequestFilter> filters = new ArrayList<HttpRequestFilter>(1);
1);
filters.add(new HttpRequestFilter() { filters.add(new HttpRequestFilter() {
public void filter(HttpRequest request) throws HttpException { public void filter(HttpRequest request) throws HttpException {
if (request.getHeaders().containsKey("filterme")) { if (request.getHeaders().containsKey("filterme")) {
@ -109,8 +107,8 @@ public abstract class BaseJettyTest {
protected void configure() { protected void configure() {
Names.bindProperties(binder(), properties); Names.bindProperties(binder(), properties);
} }
}, new JDKLoggingModule(), new HttpCommandsModule(), }, new JDKLoggingModule(), new HttpCommandsModule(), createClientModule(),
createClientModule(), new AbstractModule() { new AbstractModule() {
@Override @Override
protected void configure() { protected void configure() {
bind(new TypeLiteral<List<HttpRequestFilter>>() { bind(new TypeLiteral<List<HttpRequestFilter>>() {
@ -129,7 +127,6 @@ public abstract class BaseJettyTest {
server.stop(); server.stop();
} }
protected abstract void addConnectionProperties(Properties props); protected abstract void addConnectionProperties(Properties props);
protected abstract Module createClientModule(); protected abstract Module createClientModule();
@ -142,8 +139,8 @@ public abstract class BaseJettyTest {
* @return * @return
* @throws IOException * @throws IOException
*/ */
protected boolean failOnRequest(HttpServletRequest request, protected boolean failOnRequest(HttpServletRequest request, HttpServletResponse response)
HttpServletResponse response) throws IOException { throws IOException {
if (cycle.incrementAndGet() % 10 == 0) { if (cycle.incrementAndGet() % 10 == 0) {
response.sendError(500); response.sendError(500);
((Request) request).setHandled(true); ((Request) request).setHandled(true);
@ -152,4 +149,14 @@ public abstract class BaseJettyTest {
return false; return false;
} }
protected boolean failIfNoContentLength(HttpServletRequest request, HttpServletResponse response)
throws IOException {
if (request.getHeader(HttpConstants.CONTENT_LENGTH) == null) {
response.sendError(500);
((Request) request).setHandled(true);
return true;
}
return false;
}
} }

View File

@ -179,6 +179,8 @@ public class URLFetchServiceClient extends BaseHttpFutureCommandClient {
if (request.getPayload() != null) { if (request.getPayload() != null) {
changeRequestContentToBytes(request); changeRequestContentToBytes(request);
gaeRequest.setPayload((byte[]) request.getPayload()); gaeRequest.setPayload((byte[]) request.getPayload());
} else {
gaeRequest.addHeader(new HTTPHeader(HttpConstants.CONTENT_LENGTH, "0"));
} }
return gaeRequest; return gaeRequest;
} }

View File

@ -126,7 +126,7 @@ public class URLFetchServiceClientTest {
HttpRequest request = new HttpRequest("GET", "foo"); HttpRequest request = new HttpRequest("GET", "foo");
HTTPRequest gaeRequest = client.convert(request); HTTPRequest gaeRequest = client.convert(request);
assert gaeRequest.getPayload() == null; assert gaeRequest.getPayload() == null;
assertEquals(gaeRequest.getHeaders().size(), 0); assertEquals(gaeRequest.getHeaders().size(), 1);//content length
} }
@Test @Test