/*
* Copyright 2011-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is divalibuted
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package com.amazonaws.protocol.json.internal;
import com.amazonaws.AmazonWebServiceRequest;
import com.amazonaws.DefaultRequest;
import com.amazonaws.Request;
import com.amazonaws.annotation.SdkInternalApi;
import com.amazonaws.protocol.MarshallingInfo;
import com.amazonaws.protocol.MarshallingType;
import com.amazonaws.protocol.OperationInfo;
import com.amazonaws.protocol.ProtocolMarshaller;
import com.amazonaws.protocol.ProtocolRequestMarshaller;
import com.amazonaws.protocol.json.StructuredJsonGenerator;
import com.amazonaws.util.BinaryUtils;
import com.amazonaws.util.UriResourcePathUtils;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.nio.ByteBuffer;
/**
* Implementation of {@link ProtocolMarshaller} for JSON based services. This includes JSON-RPC and REST-JSON.
*
* @param <OrigRequest> Type of the original request object.
*/
@SdkInternalApi
public class JsonProtocolMarshaller<OrigRequest> implements ProtocolRequestMarshaller<OrigRequest> {
private static final MarshallerRegistry marshallerRegistry = createMarshallerRegistry();
private final StructuredJsonGenerator jsonGenerator;
private final Request<OrigRequest> request;
private final String contentType;
private final boolean hasExplicitPayloadMember;
private final JsonMarshallerContext marshallerContext;
public JsonProtocolMarshaller(StructuredJsonGenerator jsonGenerator,
String contentType,
OperationInfo operationInfo,
OrigRequest originalRequest) {
this.jsonGenerator = jsonGenerator;
this.contentType = contentType;
this.hasExplicitPayloadMember = operationInfo.hasExplicitPayloadMember();
this.request = fillBasicRequestParams(operationInfo, originalRequest);
this.marshallerContext = JsonMarshallerContext.builder()
.jsonGenerator(jsonGenerator)
.marshallerRegistry(marshallerRegistry)
.protocolHandler(this)
.request(request)
.build();
}
private Request<OrigRequest> fillBasicRequestParams(OperationInfo operationInfo, OrigRequest originalRequest) {
Request<OrigRequest> request = createRequest(operationInfo, originalRequest);
request.setHttpMethod(operationInfo.httpMethodName());
request.setResourcePath(UriResourcePathUtils.addStaticQueryParamtersToRequest(request, operationInfo.requestUri()));
if (operationInfo.operationIdentifier() != null) {
request.addHeader("X-Amz-Target", operationInfo.operationIdentifier());
}
return request;
}
private DefaultRequest<OrigRequest> createRequest(OperationInfo operationInfo, OrigRequest originalRequest) {
if (originalRequest instanceof AmazonWebServiceRequest) {
return new DefaultRequest<OrigRequest>((AmazonWebServiceRequest) originalRequest, operationInfo.serviceName());
} else {
return new DefaultRequest<OrigRequest>(operationInfo.serviceName());
}
}
private static MarshallerRegistry createMarshallerRegistry() {
return MarshallerRegistry.builder()
.payloadMarshaller(MarshallingType.STRING, SimpleTypeJsonMarshallers.STRING)
.payloadMarshaller(MarshallingType.JSON_VALUE, SimpleTypeJsonMarshallers.STRING)
.payloadMarshaller(MarshallingType.INTEGER, SimpleTypeJsonMarshallers.INTEGER)
.payloadMarshaller(MarshallingType.LONG, SimpleTypeJsonMarshallers.LONG)
.payloadMarshaller(MarshallingType.DOUBLE, SimpleTypeJsonMarshallers.DOUBLE)
.payloadMarshaller(MarshallingType.FLOAT, SimpleTypeJsonMarshallers.FLOAT)
.payloadMarshaller(MarshallingType.BIG_DECIMAL, SimpleTypeJsonMarshallers.BIG_DECIMAL)
.payloadMarshaller(MarshallingType.BOOLEAN, SimpleTypeJsonMarshallers.BOOLEAN)
.payloadMarshaller(MarshallingType.DATE, SimpleTypeJsonMarshallers.DATE)
.payloadMarshaller(MarshallingType.BYTE_BUFFER, SimpleTypeJsonMarshallers.BYTE_BUFFER)
.payloadMarshaller(MarshallingType.STRUCTURED, SimpleTypeJsonMarshallers.STRUCTURED)
.payloadMarshaller(MarshallingType.LIST, SimpleTypeJsonMarshallers.LIST)
.payloadMarshaller(MarshallingType.MAP, SimpleTypeJsonMarshallers.MAP)
.payloadMarshaller(MarshallingType.NULL, SimpleTypeJsonMarshallers.NULL)
.headerMarshaller(MarshallingType.STRING, HeaderMarshallers.STRING)
.headerMarshaller(MarshallingType.JSON_VALUE, HeaderMarshallers.JSON_VALUE)
.headerMarshaller(MarshallingType.INTEGER, HeaderMarshallers.INTEGER)
.headerMarshaller(MarshallingType.LONG, HeaderMarshallers.LONG)
.headerMarshaller(MarshallingType.DOUBLE, HeaderMarshallers.DOUBLE)
.headerMarshaller(MarshallingType.FLOAT, HeaderMarshallers.FLOAT)
.headerMarshaller(MarshallingType.BOOLEAN, HeaderMarshallers.BOOLEAN)
.headerMarshaller(MarshallingType.DATE, HeaderMarshallers.DATE)
.headerMarshaller(MarshallingType.NULL, JsonMarshaller.NULL)
.queryParamMarshaller(MarshallingType.STRING, QueryParamMarshallers.STRING)
.queryParamMarshaller(MarshallingType.INTEGER, QueryParamMarshallers.INTEGER)
.queryParamMarshaller(MarshallingType.LONG, QueryParamMarshallers.LONG)
.queryParamMarshaller(MarshallingType.DOUBLE, QueryParamMarshallers.DOUBLE)
.queryParamMarshaller(MarshallingType.FLOAT, QueryParamMarshallers.FLOAT)
.queryParamMarshaller(MarshallingType.BOOLEAN, QueryParamMarshallers.BOOLEAN)
.queryParamMarshaller(MarshallingType.DATE, QueryParamMarshallers.DATE)
.queryParamMarshaller(MarshallingType.LIST, QueryParamMarshallers.LIST)
.queryParamMarshaller(MarshallingType.MAP, QueryParamMarshallers.MAP)
.queryParamMarshaller(MarshallingType.NULL, JsonMarshaller.NULL)
.pathParamMarshaller(MarshallingType.STRING, SimpleTypePathMarshallers.STRING)
.pathParamMarshaller(MarshallingType.INTEGER, SimpleTypePathMarshallers.INTEGER)
.pathParamMarshaller(MarshallingType.LONG, SimpleTypePathMarshallers.LONG)
.pathParamMarshaller(MarshallingType.NULL, SimpleTypePathMarshallers.NULL)
.greedyPathParamMarshaller(MarshallingType.STRING, SimpleTypePathMarshallers.GREEDY_STRING)
.greedyPathParamMarshaller(MarshallingType.NULL, SimpleTypePathMarshallers.NULL)
.build();
}
/**
* If there is not an explicit payload member then we need to start the implicit JSON request object. All
* members bound to the payload will be added as fields to this object.
*/
@Override
public void startMarshalling() {
if (!hasExplicitPayloadMember) {
jsonGenerator.writeStartObject();
}
}
@Override
public <V> void marshall(V val, MarshallingInfo<V> marshallingInfo) {
doMarshall(resolveValue(val, marshallingInfo), marshallingInfo);
}
/**
* @return The original value if non-null, or if value is null and a {@link com.amazonaws.protocol.DefaultValueSupplier} is
* present return the default value. Otherwise return null.
*/
private <V> V resolveValue(V val, MarshallingInfo<V> marshallingInfo) {
return val == null && marshallingInfo.defaultValueSupplier() != null ? marshallingInfo.defaultValueSupplier().get() : val;
}
private <V> void doMarshall(V val, MarshallingInfo<V> marshallingInfo) {
if (marshallingInfo.isBinary()) {
marshallBinaryPayload(val);
} else {
marshallerRegistry.getMarshaller(marshallingInfo.marshallLocation(), marshallingInfo.marshallingType(), val)
.marshall(val, marshallerContext, marshallingInfo.marshallLocationName());
}
}
/**
* Binary data should be placed as is, directly into the content.
*/
private void marshallBinaryPayload(Object val) {
if (val instanceof ByteBuffer) {
request.setContent(BinaryUtils.toStream((ByteBuffer) val));
} else if (val instanceof InputStream) {
request.setContent((InputStream) val);
}
}
@Override
public Request<OrigRequest> finishMarshalling() {
// Content may already be set if the payload is binary data.
if (request.getContent() == null) {
// End the implicit request object if needed.
if (!hasExplicitPayloadMember) {
jsonGenerator.writeEndObject();
}
byte[] content = jsonGenerator.getBytes();
request.setContent(new ByteArrayInputStream(content));
if (content.length > 0) {
request.addHeader("Content-Length", Integer.toString(content.length));
}
}
if (!request.getHeaders().containsKey("Content-Type")) {
request.addHeader("Content-Type", contentType);
}
return request;
}
}