From ddfb8e58a5a729cabb5edf954a113ee8127cd7cb Mon Sep 17 00:00:00 2001 From: Andrew Gaul Date: Fri, 10 May 2013 15:39:48 -0700 Subject: [PATCH] JCLOUDS-27: Allow repeatable Payload with InputSupplier input This allows HTTP retries to work. Also remove duplicated calls to ByteStreams.slice. --- .../org/jclouds/io/internal/BasePayloadSlicer.java | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/core/src/main/java/org/jclouds/io/internal/BasePayloadSlicer.java b/core/src/main/java/org/jclouds/io/internal/BasePayloadSlicer.java index 5af50660b1..a7f814ccd6 100644 --- a/core/src/main/java/org/jclouds/io/internal/BasePayloadSlicer.java +++ b/core/src/main/java/org/jclouds/io/internal/BasePayloadSlicer.java @@ -36,6 +36,7 @@ import org.jclouds.io.payloads.InputStreamSupplierPayload; import com.google.common.base.Throwables; import com.google.common.io.ByteStreams; import com.google.common.io.Files; +import com.google.common.io.InputSupplier; /** * @@ -58,14 +59,16 @@ public class BasePayloadSlicer implements PayloadSlicer { returnVal = doSlice((String) input.getRawContent(), offset, length); } else if (input.getRawContent() instanceof byte[]) { returnVal = doSlice((byte[]) input.getRawContent(), offset, length); + } else if (input.getRawContent() instanceof InputStream) { + returnVal = doSlice((InputStream) input.getRawContent(), offset, length); } else { - returnVal = doSlice(input.getInput(), offset, length); + returnVal = doSlice(input, offset, length); } return copyMetadataAndSetLength(input, returnVal, length); } protected Payload doSlice(Payload content, long offset, long length) { - return new InputStreamSupplierPayload(ByteStreams.slice(content, offset, length)); + return doSlice((InputSupplier) content, offset, length); } protected Payload doSlice(String content, long offset, long length) { @@ -73,7 +76,7 @@ public class BasePayloadSlicer implements PayloadSlicer { } protected Payload doSlice(File content, long offset, long length) { - return new InputStreamSupplierPayload(ByteStreams.slice(Files.newInputStreamSupplier(content), offset, length)); + return doSlice(Files.newInputStreamSupplier(content), offset, length); } protected Payload doSlice(InputStream content, long offset, long length) { @@ -85,6 +88,10 @@ public class BasePayloadSlicer implements PayloadSlicer { return new InputStreamPayload(ByteStreams.limit(content, length)); } + protected Payload doSlice(InputSupplier content, long offset, long length) { + return new InputStreamSupplierPayload(ByteStreams.slice(content, offset, length)); + } + protected Payload doSlice(byte[] content, long offset, long length) { Payload returnVal; checkArgument(offset <= Integer.MAX_VALUE, "offset is too big for an array");