public QueryResult unmarshall(JsonUnmarshallerContext context) throws Exception {
QueryResult queryResult = new QueryResult();
int originalDepth = context.getCurrentDepth();
String currentParentElement = context.getCurrentParentElement();
int targetDepth = originalDepth + 1;
JsonToken token = context.currentToken;
if (token == null) token = context.nextToken();
if (token == VALUE_NULL) return null;
while (true) {
if (token == null) break;
if (token == FIELD_NAME || token == START_OBJECT) {
if (context.testExpression("Items", targetDepth)) {
context.nextToken();
queryResult.setItems(
new ListUnmarshaller<java.util.Map<String, AttributeValue>>(
new MapUnmarshaller<String, AttributeValue>(
StringJsonUnmarshaller.getInstance(),
AttributeValueJsonUnmarshaller.getInstance()))
.unmarshall(context));
}
if (context.testExpression("Count", targetDepth)) {
context.nextToken();
queryResult.setCount(IntegerJsonUnmarshaller.getInstance().unmarshall(context));
}
if (context.testExpression("ScannedCount", targetDepth)) {
context.nextToken();
queryResult.setScannedCount(IntegerJsonUnmarshaller.getInstance().unmarshall(context));
}
if (context.testExpression("LastEvaluatedKey", targetDepth)) {
context.nextToken();
queryResult.setLastEvaluatedKey(
new MapUnmarshaller<String, AttributeValue>(
StringJsonUnmarshaller.getInstance(),
AttributeValueJsonUnmarshaller.getInstance())
.unmarshall(context));
}
if (context.testExpression("ConsumedCapacity", targetDepth)) {
context.nextToken();
queryResult.setConsumedCapacity(
ConsumedCapacityJsonUnmarshaller.getInstance().unmarshall(context));
}
} else if (token == END_ARRAY || token == END_OBJECT) {
if (context.getLastParsedParentElement() == null
|| context.getLastParsedParentElement().equals(currentParentElement)) {
if (context.getCurrentDepth() <= originalDepth) break;
}
}
token = context.nextToken();
}
return queryResult;
}
public GetItemResult unmarshall(JsonUnmarshallerContext context) throws Exception {
GetItemResult getItemResult = new GetItemResult();
int originalDepth = context.getCurrentDepth();
String currentParentElement = context.getCurrentParentElement();
int targetDepth = originalDepth + 1;
JsonToken token = context.currentToken;
if (token == null) token = context.nextToken();
if (token == VALUE_NULL) return null;
while (true) {
if (token == null) break;
if (token == FIELD_NAME || token == START_OBJECT) {
if (context.testExpression("Item", targetDepth)) {
context.nextToken();
getItemResult.setItem(
new MapUnmarshaller<String, AttributeValue>(
StringJsonUnmarshaller.getInstance(),
AttributeValueJsonUnmarshaller.getInstance())
.unmarshall(context));
}
if (context.testExpression("ConsumedCapacity", targetDepth)) {
context.nextToken();
getItemResult.setConsumedCapacity(
ConsumedCapacityJsonUnmarshaller.getInstance().unmarshall(context));
}
} else if (token == END_ARRAY || token == END_OBJECT) {
if (context.getLastParsedParentElement() == null
|| context.getLastParsedParentElement().equals(currentParentElement)) {
if (context.getCurrentDepth() <= originalDepth) break;
}
}
token = context.nextToken();
}
return getItemResult;
}
