com.amazonaws.ResponseMetadata Java Examples

public <RequestT extends AmazonWebServiceRequest, ResultT extends AmazonWebServiceResult<ResponseMetadata>>
    ResultT
    injectCredentialsAndInvoke(final RequestT request, final Function<RequestT, ResultT> requestFunction) {

    request.setRequestCredentialsProvider(v1CredentialsProvider);

    try {
        ResultT respose = requestFunction.apply(request);
        logRequestMetadata(request, respose);
        return respose;
    } catch (final Throwable e) {
        loggerProxy.log(String.format("Failed to execute remote function: {%s}", e.getMessage()));
        throw e;
    } finally {
        request.setRequestCredentialsProvider(null);
    }
}
 

static void tagSpanWithRequestId(Span span, Response response) {
  String requestId = null;
  if (response != null) {
    if (response.getAwsResponse() instanceof AmazonWebServiceResult<?>) {
      ResponseMetadata metadata =
          ((AmazonWebServiceResult<?>) response.getAwsResponse()).getSdkResponseMetadata();
      if (null != metadata) {
        requestId = metadata.getRequestId();
      }
    } else if (response.getHttpResponse() != null) {
      if (response.getHttpResponse().getHeader("x-amz-request-id") != null) {
        requestId = response.getHttpResponse().getHeader("x-amz-request-id");
      }
    }
  }
  if (requestId != null) {
    span.tag("aws.request_id", requestId);
  }
}
 

private <RequestT extends AmazonWebServiceRequest, ResultT extends AmazonWebServiceResult<ResponseMetadata>>
    void
    logRequestMetadata(final RequestT request, final ResultT response) {
    try {
        String requestName = request.getClass().getSimpleName();
        String requestId = (response == null || response.getSdkResponseMetadata() == null)
            ? ""
            : response.getSdkResponseMetadata().getRequestId();
        loggerProxy
            .log(String.format("{\"apiRequest\": {\"requestId\": \"%s\", \"requestName\": \"%s\"}}", requestId, requestName));
    } catch (final Exception e) {
        loggerProxy.log(e.getMessage());
    }
}
 

private void populateAndEndSubsegment(Subsegment currentSubsegment, Request<?> request, Response<?> response) {
    if (null != response) {
        String requestId = null;
        if (response.getAwsResponse() instanceof AmazonWebServiceResult<?>) {
            // Not all services return responses extending AmazonWebServiceResult (e.g. S3)
            ResponseMetadata metadata = ((AmazonWebServiceResult<?>) response.getAwsResponse()).getSdkResponseMetadata();
            if (null != metadata) {
                requestId = metadata.getRequestId();
                if (null != requestId) {
                    currentSubsegment.putAws(REQUEST_ID_SUBSEGMENT_KEY, requestId);
                }
            }
        } else if (null != response.getHttpResponse()) { // S3 does not follow request id header convention
            if (null != response.getHttpResponse().getHeader(S3_REQUEST_ID_HEADER_KEY)) {
                currentSubsegment.putAws(REQUEST_ID_SUBSEGMENT_KEY,
                                         response.getHttpResponse().getHeader(S3_REQUEST_ID_HEADER_KEY));
            }
            if (null != response.getHttpResponse().getHeader(EntityHeaderKeys.AWS.EXTENDED_REQUEST_ID_HEADER)) {
                currentSubsegment.putAws(EntityDataKeys.AWS.EXTENDED_REQUEST_ID_KEY,
                                         response.getHttpResponse().getHeader(
                                             EntityHeaderKeys.AWS.EXTENDED_REQUEST_ID_HEADER));
            }
        }
        currentSubsegment.putAllAws(extractResponseParameters(request, response.getAwsResponse()));
        currentSubsegment.putAllHttp(extractHttpResponseInformation(response.getHttpResponse()));
    }

    finalizeSubsegment(request);
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest request) {
  Map<String, String> metadata = new HashMap<String, String>();
  ResponseMetadata responseMetadata = new ResponseMetadata(metadata);

  return responseMetadata;
}
 

private PublishResult buildFullPublishResult() {
  PublishResult publishResult = new PublishResult().withMessageId(UUID.randomUUID().toString());
  publishResult.setSdkResponseMetadata(
      new ResponseMetadata(
          ImmutableMap.of(ResponseMetadata.AWS_REQUEST_ID, UUID.randomUUID().toString())));
  HttpResponse httpResponse = new HttpResponse(null, null);
  httpResponse.setStatusCode(200);
  httpResponse.addHeader("Content-Type", "application/json");
  publishResult.setSdkHttpMetadata(SdkHttpMetadata.from(httpResponse));
  return publishResult;
}
 

@Test
public void test_getCachedResponseMetadata() throws Exception {
  ResponseMetadata result = dynamoDb.getCachedResponseMetadata(null);

  assertThat(result, notNullValue());
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest amazonWebServiceRequest)
{
    return null;
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest request) {
    return null;
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest request) {
    return client.getResponseMetadataForRequest(request);
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(
        AmazonWebServiceRequest arg0) {
    throw new UnsupportedOperationException("Use the underlying client instance instead");
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest request) {
    return getBackend().getCachedResponseMetadata(request);
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(
        AmazonWebServiceRequest arg0) {
    throw new UnsupportedOperationException("Use the underlying client instance instead");
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(final AmazonWebServiceRequest arg0) {
  throw new UnsupportedOperationException();
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest amazonWebServiceRequest) {
    return null;
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest amazonWebServiceRequest)
{
    return null;
}
 

@Override
public ResponseMetadata getCachedResponseMetadata(AmazonWebServiceRequest arg0) {
    throw new java.lang.UnsupportedOperationException();
}
 

private ResponseMetadata buildResponseMetadata() {
  return new ResponseMetadata(
      ImmutableMap.of(ResponseMetadata.AWS_REQUEST_ID, UUID.randomUUID().toString()));
}
 

@Test
public void testResponseMetadataDecodeEncodeEquals() throws Exception {
  ResponseMetadata value = buildResponseMetadata();
  ResponseMetadata clone = CoderUtils.clone(AwsCoders.responseMetadata(), value);
  assertThat(clone.getRequestId(), equalTo(value.getRequestId()));
}
 

@Override
public ResponseMetadata decode(InputStream inStream) throws CoderException, IOException {
  return new ResponseMetadata(METADATA_ENCODER.decode(inStream));
}
 

@Override
public void encode(ResponseMetadata value, OutputStream outStream)
    throws CoderException, IOException {
  METADATA_ENCODER.encode(
      ImmutableMap.of(ResponseMetadata.AWS_REQUEST_ID, value.getRequestId()), outStream);
}
 

private PublishResultCoder(
    Coder<ResponseMetadata> responseMetadataEncoder,
    Coder<SdkHttpMetadata> sdkHttpMetadataCoder) {
  this.responseMetadataEncoder = responseMetadataEncoder;
  this.sdkHttpMetadataCoder = sdkHttpMetadataCoder;
}