HTTPCLIENT-951: Non-repeatable entity enclosing requests are not corr…

…ectly retried when 'expect-continue' handshake is active git-svn-id: https://svn.apache.org/repos/asf/httpcomponents/httpclient/branches/4.0.x@957015 13f79535-47bb-0310-9956-ffa450edef68
mojia · Jun 22, 2010 · ef067cb · ef067cb
1 parent ed18d9e
commit ef067cb
Show file tree

Hide file tree

Showing 4 changed files with 54 additions and 6 deletions.
diff --git a/RELEASE_NOTES.txt b/RELEASE_NOTES.txt
@@ -1,6 +1,10 @@
 Changes since 4.0.1
 -------------------
 
+* [HTTPCLIENT-951] Non-repeatable entity enclosing requests are not correctly
+  retried when 'expect-continue' handshake is active.
+  Contributed by Oleg Kalnichevski <olegk at apache.org>
+
 * [HTTPCLIENT-953] IllegalStateException thrown by RouteSpecificPool.
   Contributed by Guillaume <gueugaie at gmail.com>
 

diff --git a/httpclient/src/main/java/org/apache/http/impl/client/DefaultRequestDirector.java b/httpclient/src/main/java/org/apache/http/impl/client/DefaultRequestDirector.java
@@ -464,7 +464,7 @@ public HttpResponse execute(HttpHost target, HttpRequest request,
                     execCount++;
                     // Increment exec count for this particular request
                     wrapper.incrementExecCount();
-                    if (wrapper.getExecCount() > 1 && !wrapper.isRepeatable()) {
+                    if (!wrapper.isRepeatable()) {
                         this.log.debug("Cannot retry non-repeatable request");
                         if (retryReason != null) {
                             throw new NonRepeatableRequestException("Cannot retry request " +

diff --git a/httpclient/src/main/java/org/apache/http/impl/client/EntityEnclosingRequestWrapper.java b/httpclient/src/main/java/org/apache/http/impl/client/EntityEnclosingRequestWrapper.java
@@ -27,7 +27,12 @@
 
 package org.apache.http.impl.client;
 
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+
 import org.apache.http.annotation.NotThreadSafe;
+import org.apache.http.entity.HttpEntityWrapper;
 
 import org.apache.http.Header;
 import org.apache.http.HttpEntity;
@@ -51,19 +56,21 @@ public class EntityEnclosingRequestWrapper extends RequestWrapper
     implements HttpEntityEnclosingRequest {
 
     private HttpEntity entity;
+    private boolean consumed;
 
-    public EntityEnclosingRequestWrapper(final HttpEntityEnclosingRequest request) 
+    public EntityEnclosingRequestWrapper(final HttpEntityEnclosingRequest request)
         throws ProtocolException {
         super(request);
-        this.entity = request.getEntity();
+        setEntity(request.getEntity());
     }
 
     public HttpEntity getEntity() {
         return this.entity;
     }
 
     public void setEntity(final HttpEntity entity) {
-        this.entity = entity;
+        this.entity = entity != null ? new EntityWrapper(entity) : null;
+        this.consumed = false;
     }
 
     public boolean expectContinue() {
@@ -73,7 +80,33 @@ public boolean expectContinue() {
 
     @Override
     public boolean isRepeatable() {
-        return this.entity == null || this.entity.isRepeatable();
+        return this.entity == null || this.entity.isRepeatable() || !this.consumed;
+    }
+
+    class EntityWrapper extends HttpEntityWrapper {
+
+        EntityWrapper(final HttpEntity entity) {
+            super(entity);
+        }
+
+        @Override
+        public void consumeContent() throws IOException {
+            consumed = true;
+            super.consumeContent();
+        }
+
+        @Override
+        public InputStream getContent() throws IOException {
+            consumed = true;
+            return super.getContent();
+        }
+
+        @Override
+        public void writeTo(final OutputStream outstream) throws IOException {
+            consumed = true;
+            super.writeTo(outstream);
+        }
+
     }
 
 }
diff --git a/httpclient/src/test/java/org/apache/http/impl/client/TestDefaultClientRequestDirector.java b/httpclient/src/test/java/org/apache/http/impl/client/TestDefaultClientRequestDirector.java
@@ -658,12 +658,13 @@ public HttpResponse execute(
                 final HttpClientConnection conn, 
                 final HttpContext context) throws IOException, HttpException {
 
+            HttpResponse response = super.execute(request, conn, context);
             Object marker = context.getAttribute(MARKER);
             if (marker == null) {
                 context.setAttribute(MARKER, Boolean.TRUE);
                 throw new IOException(failureMsg);
             }
-            return super.execute(request, conn, context);
+            return response;
         }
 
     }
@@ -744,6 +745,16 @@ public void testNonRepeatableEntity() throws Exception {
         String failureMsg = "a message showing that this failed";
 
         FaultyHttpClient client = new FaultyHttpClient(failureMsg); 
+        client.setHttpRequestRetryHandler(new HttpRequestRetryHandler() {
+
+            public boolean retryRequest(
+                    final IOException exception,
+                    int executionCount,
+                    final HttpContext context) {
+                return true;
+            }
+
+        });
         HttpContext context = new BasicHttpContext();
 
         String s = "http://localhost:" + port;