How to use

S3ObjectResponseHandler

in

com.amazonaws.services.s3.internal

@Override
public SelectObjectContentResult selectObjectContent(SelectObjectContentRequest selectRequest) throws AmazonServiceException, SdkClientException {
  selectRequest = beforeClientExecution(selectRequest);
  rejectNull(selectRequest, "The request parameter must be specified");
  rejectNull(selectRequest.getBucketName(), "The bucket name parameter must be specified when selecting object content.");
  rejectNull(selectRequest.getKey(), "The key parameter must be specified when selecting object content.");
  Request<SelectObjectContentRequest> request = createRequest(selectRequest.getBucketName(), selectRequest.getKey(), selectRequest, HttpMethodName.POST);
  request.addParameter("select", null);
  request.addParameter("select-type", "2");
  populateSSE_C(request, selectRequest.getSSECustomerKey());
  setContent(request, RequestXmlFactory.convertToXmlByteArray(selectRequest), ContentType.APPLICATION_XML.toString(), true);
  S3Object result = invoke(request, new S3ObjectResponseHandler(), selectRequest.getBucketName(), selectRequest.getKey());
  // Hold a reference to this client while the InputStream is still
  // around - otherwise a finalizer in the HttpClient may reset the
  // underlying TCP connection out from under us.
  SdkFilterInputStream resultStream = new ServiceClientHolderInputStream(result.getObjectContent(), this);
  return new SelectObjectContentResult().withPayload(new SelectObjectContentEventStream(resultStream));
}

/**
 * @see com.amazonaws.http.HttpResponseHandler#handle(com.amazonaws.http.HttpResponse)
 */
public AmazonWebServiceResponse<S3Object> handle(HttpResponse response) throws Exception {
  /*
   * TODO: It'd be nice to set the bucket name and key here, but the information isn't easy to
   * pull out of the response/request currently.
   */
  S3Object object = new S3Object();
  AmazonWebServiceResponse<S3Object> awsResponse = parseResponseMetadata(response);
  if (response.getHeaders().get(Headers.REDIRECT_LOCATION) != null) {
    object.setRedirectLocation(response.getHeaders().get(Headers.REDIRECT_LOCATION));
  }
  // If the requester is charged when downloading a object from an
  // Requester Pays bucket, then this header is set.
  if (response.getHeaders().get(Headers.REQUESTER_CHARGED_HEADER) != null) {
    object.setRequesterCharged(true);
  }
  if (response.getHeaders().get(Headers.S3_TAGGING_COUNT) != null) {
    object.setTaggingCount(Integer.parseInt(response.getHeaders().get(Headers.S3_TAGGING_COUNT)));
  }
  ObjectMetadata metadata = object.getObjectMetadata();
  populateObjectMetadata(response, metadata);
  object.setObjectContent(new S3ObjectInputStream(response.getContent(), response.getHttpRequest()));
  awsResponse.setResult(object);
  return awsResponse;
}

/**
 * @see com.amazonaws.http.HttpResponseHandler#handle(com.amazonaws.http.HttpResponse)
 */
public AmazonWebServiceResponse<S3Object> handle(HttpResponse response) throws Exception {
  /*
   * TODO: It'd be nice to set the bucket name and key here, but the information isn't easy to
   * pull out of the response/request currently.
   */
  S3Object object = new S3Object();
  AmazonWebServiceResponse<S3Object> awsResponse = parseResponseMetadata(response);
  if (response.getHeaders().get(Headers.REDIRECT_LOCATION) != null) {
    object.setRedirectLocation(response.getHeaders().get(Headers.REDIRECT_LOCATION));
  }
  // If the requester is charged when downloading a object from an
  // Requester Pays bucket, then this header is set.
  if (response.getHeaders().get(Headers.REQUESTER_CHARGED_HEADER) != null) {
    object.setRequesterCharged(true);
  }
  if (response.getHeaders().get(Headers.S3_TAGGING_COUNT) != null) {
    object.setTaggingCount(Integer.parseInt(response.getHeaders().get(Headers.S3_TAGGING_COUNT)));
  }
  ObjectMetadata metadata = object.getObjectMetadata();
  populateObjectMetadata(response, metadata);
  object.setObjectContent(new S3ObjectInputStream(abortableIs(response), response.getHttpRequest()));
  awsResponse.setResult(object);
  return awsResponse;
}

@Test
public void testNeedsConnectionLeftOpen() {
  S3ObjectResponseHandler handler = new S3ObjectResponseHandler();
  assertTrue(handler.needsConnectionLeftOpen());
}

@Test
public void testHandle() throws Exception {
  ByteArrayInputStream bais = new ByteArrayInputStream("content".getBytes(StringUtils.UTF8));
  HttpResponse response = new HttpResponse.Builder().content(bais)
      .header(Headers.REDIRECT_LOCATION, "redirect")
      .header(Headers.REQUESTER_CHARGED_HEADER, "true").build();
  S3ObjectResponseHandler handler = new S3ObjectResponseHandler();
  AmazonWebServiceResponse<S3Object> object = handler.handle(response);
  S3Object content = object.getResult();
  assertEquals(content.getRedirectLocation(), "redirect");
  assertTrue(content.isRequesterCharged());
  S3ObjectInputStream is = content.getObjectContent();
  int curr = -1;
  ByteArrayOutputStream baos = new ByteArrayOutputStream();
  while ((curr = is.read()) != -1) {
    baos.write(curr);
  }
  assertArrayEquals(baos.toByteArray(), "content".getBytes(StringUtils.UTF8));
}

private S3AbortableInputStream abortableIs(HttpResponse response) {
  return new S3AbortableInputStream(response.getContent(), response.getHttpRequest(), getContentLength(response));
}

/**
 * @see com.amazonaws.http.HttpResponseHandler#handle(com.amazonaws.http.HttpResponse)
 */
@Override
public AmazonWebServiceResponse<S3Object> handle(HttpResponse response) throws Exception {
  /*
   * TODO: It'd be nice to set the bucket name and key here, but the
   * information isn't easy to pull out of the response/request currently.
   */
  final S3Object object = new S3Object();
  final AmazonWebServiceResponse<S3Object> awsResponse = parseResponseMetadata(response);
  if (response.getHeaders().get(Headers.REDIRECT_LOCATION) != null) {
    object.setRedirectLocation(response.getHeaders().get(Headers.REDIRECT_LOCATION));
  }
  // If the requester is charged when downloading a object from an
  // Requester Pays bucket, then this header is set.
  if (response.getHeaders().get(Headers.REQUESTER_CHARGED_HEADER) != null) {
    object.setRequesterCharged(true);
  }
  if (response.getHeaders().get(Headers.S3_TAGGING_COUNT) != null) {
    object.setTaggingCount(Integer.parseInt(response.getHeaders().get(Headers.S3_TAGGING_COUNT)));
  }
  final ObjectMetadata metadata = object.getObjectMetadata();
  populateObjectMetadata(response, metadata);
  object.setObjectContent(new S3ObjectInputStream(response.getContent()));
  awsResponse.setResult(object);
  return awsResponse;
}

S3Object s3Object = invoke(request, new S3ObjectResponseHandler(),
    getObjectRequest.getBucketName(), getObjectRequest.getKey());

/**
 * @see com.amazonaws.http.HttpResponseHandler#handle(com.amazonaws.http.HttpResponse)
 */
@Override
public AmazonWebServiceResponse<S3Object> handle(HttpResponse response) throws Exception {
  /*
   * TODO: It'd be nice to set the bucket name and key here, but the
   * information isn't easy to pull out of the response/request currently.
   */
  final S3Object object = new S3Object();
  final AmazonWebServiceResponse<S3Object> awsResponse = parseResponseMetadata(response);
  if (response.getHeaders().get(Headers.REDIRECT_LOCATION) != null) {
    object.setRedirectLocation(response.getHeaders().get(Headers.REDIRECT_LOCATION));
  }
  // If the requester is charged when downloading a object from an
  // Requester Pays bucket, then this header is set.
  if (response.getHeaders().get(Headers.REQUESTER_CHARGED_HEADER) != null) {
    object.setRequesterCharged(true);
  }
  if (response.getHeaders().get(Headers.S3_TAGGING_COUNT) != null) {
    object.setTaggingCount(Integer.parseInt(response.getHeaders().get(Headers.S3_TAGGING_COUNT)));
  }
  final ObjectMetadata metadata = object.getObjectMetadata();
  populateObjectMetadata(response, metadata);
  object.setObjectContent(new S3ObjectInputStream(response.getContent()));
  awsResponse.setResult(object);
  return awsResponse;
}

final S3Object s3Object = invoke(request, new S3ObjectResponseHandler(),
    getObjectRequest.getBucketName(), getObjectRequest.getKey());

S3Object s3Object = invoke(request, new S3ObjectResponseHandler(),
    getObjectRequest.getBucketName(), getObjectRequest.getKey());

final S3Object s3Object = invoke(request, new S3ObjectResponseHandler(),
    getObjectRequest.getBucketName(), getObjectRequest.getKey());