/** * Decorates the given <code>updateItemRequest</code> with attributes required for geo spatial * querying. * * @param attributeValueMap the items that needs to be decorated with geo attributes * @param latitude the latitude that needs to be attached with the item * @param longitude the longitude that needs to be attached with the item * @param configs the collection of configurations to be used for decorating the request with geo * attributes */ public void updateAttributeValues( Map<String, AttributeValue> attributeValueMap, double latitude, double longitude, List<GeoConfig> configs) { if (configs == null) { throw new IllegalArgumentException("Geo configs should not be null"); } for (GeoConfig config : configs) { // Fail-fast if any of the preconditions fail checkConfigParams( config.getGeoIndexName(), config.getGeoHashKeyColumn(), config.getGeoHashColumn(), config.getGeoHashKeyLength()); long geohash = s2Manager.generateGeohash(latitude, longitude); long geoHashKey = s2Manager.generateHashKey(geohash, config.getGeoHashKeyLength()); // Decorate the request with the geohash AttributeValue geoHashValue = new AttributeValue().withN(Long.toString(geohash)); attributeValueMap.put(config.getGeoHashColumn(), geoHashValue); AttributeValue geoHashKeyValue; if (config.getHashKeyDecorator().isPresent() && config.getCompositeHashKeyColumn().isPresent()) { AttributeValue compositeHashKeyValue = attributeValueMap.get(config.getCompositeHashKeyColumn().get()); if (compositeHashKeyValue == null) { continue; } String compositeColumnValue = compositeHashKeyValue.getS(); String hashKey = config.getHashKeyDecorator().get().decorate(compositeColumnValue, geoHashKey); // Decorate the request with the composite geoHashKey (type String) geoHashKeyValue = new AttributeValue().withS(String.valueOf(hashKey)); } else { // Decorate the request with the geoHashKey (type Number) geoHashKeyValue = new AttributeValue().withN(String.valueOf(geoHashKey)); } attributeValueMap.put(config.getGeoHashKeyColumn(), geoHashKeyValue); } }
public Request<BatchGetItemRequest> marshall(BatchGetItemRequest batchGetItemRequest) { if (batchGetItemRequest == null) { throw new AmazonClientException("Invalid argument passed to marshall(...)"); } Request<BatchGetItemRequest> request = new DefaultRequest<BatchGetItemRequest>(batchGetItemRequest, "AmazonDynamoDB"); String target = "DynamoDB_20111205.BatchGetItem"; request.addHeader("X-Amz-Target", target); request.addHeader("Content-Type", "application/x-amz-json-1.0"); request.setHttpMethod(HttpMethodName.POST); String uriResourcePath = ""; uriResourcePath = uriResourcePath.replaceAll("//", "/"); if (uriResourcePath.contains("?")) { String queryString = uriResourcePath.substring(uriResourcePath.indexOf("?") + 1); uriResourcePath = uriResourcePath.substring(0, uriResourcePath.indexOf("?")); for (String s : queryString.split("[;&]")) { String[] nameValuePair = s.split("="); if (nameValuePair.length == 2) { request.addParameter(nameValuePair[0], nameValuePair[1]); } else { request.addParameter(s, null); } } } request.setResourcePath(uriResourcePath); try { StringWriter stringWriter = new StringWriter(); JSONWriter jsonWriter = new JSONWriter(stringWriter); jsonWriter.object(); if (batchGetItemRequest.getRequestItems() != null) { jsonWriter.key("RequestItems"); jsonWriter.object(); for (Map.Entry<String, KeysAndAttributes> requestItemsListValue : batchGetItemRequest.getRequestItems().entrySet()) { if (requestItemsListValue.getValue() != null) { jsonWriter.key(requestItemsListValue.getKey()); jsonWriter.object(); java.util.List<Key> keysList = requestItemsListValue.getValue().getKeys(); if (keysList != null && keysList.size() > 0) { jsonWriter.key("Keys"); jsonWriter.array(); for (Key keysListValue : keysList) { if (keysListValue != null) { jsonWriter.object(); AttributeValue hashKeyElement = keysListValue.getHashKeyElement(); if (hashKeyElement != null) { jsonWriter.key("HashKeyElement"); jsonWriter.object(); if (hashKeyElement.getS() != null) { jsonWriter.key("S").value(hashKeyElement.getS()); } if (hashKeyElement.getN() != null) { jsonWriter.key("N").value(hashKeyElement.getN()); } if (hashKeyElement.getB() != null) { jsonWriter.key("B").value(hashKeyElement.getB()); } java.util.List<String> sSList = hashKeyElement.getSS(); if (sSList != null && sSList.size() > 0) { jsonWriter.key("SS"); jsonWriter.array(); for (String sSListValue : sSList) { if (sSListValue != null) { jsonWriter.value(sSListValue); } } jsonWriter.endArray(); } java.util.List<String> nSList = hashKeyElement.getNS(); if (nSList != null && nSList.size() > 0) { jsonWriter.key("NS"); jsonWriter.array(); for (String nSListValue : nSList) { if (nSListValue != null) { jsonWriter.value(nSListValue); } } jsonWriter.endArray(); } java.util.List<java.nio.ByteBuffer> bSList = hashKeyElement.getBS(); if (bSList != null && bSList.size() > 0) { jsonWriter.key("BS"); jsonWriter.array(); for (java.nio.ByteBuffer bSListValue : bSList) { if (bSListValue != null) { jsonWriter.value(bSListValue); } } jsonWriter.endArray(); } jsonWriter.endObject(); } AttributeValue rangeKeyElement = keysListValue.getRangeKeyElement(); if (rangeKeyElement != null) { jsonWriter.key("RangeKeyElement"); jsonWriter.object(); if (rangeKeyElement.getS() != null) { jsonWriter.key("S").value(rangeKeyElement.getS()); } if (rangeKeyElement.getN() != null) { jsonWriter.key("N").value(rangeKeyElement.getN()); } if (rangeKeyElement.getB() != null) { jsonWriter.key("B").value(rangeKeyElement.getB()); } java.util.List<String> sSList = rangeKeyElement.getSS(); if (sSList != null && sSList.size() > 0) { jsonWriter.key("SS"); jsonWriter.array(); for (String sSListValue : sSList) { if (sSListValue != null) { jsonWriter.value(sSListValue); } } jsonWriter.endArray(); } java.util.List<String> nSList = rangeKeyElement.getNS(); if (nSList != null && nSList.size() > 0) { jsonWriter.key("NS"); jsonWriter.array(); for (String nSListValue : nSList) { if (nSListValue != null) { jsonWriter.value(nSListValue); } } jsonWriter.endArray(); } java.util.List<java.nio.ByteBuffer> bSList = rangeKeyElement.getBS(); if (bSList != null && bSList.size() > 0) { jsonWriter.key("BS"); jsonWriter.array(); for (java.nio.ByteBuffer bSListValue : bSList) { if (bSListValue != null) { jsonWriter.value(bSListValue); } } jsonWriter.endArray(); } jsonWriter.endObject(); } jsonWriter.endObject(); } } jsonWriter.endArray(); } java.util.List<String> attributesToGetList = requestItemsListValue.getValue().getAttributesToGet(); if (attributesToGetList != null && attributesToGetList.size() > 0) { jsonWriter.key("AttributesToGet"); jsonWriter.array(); for (String attributesToGetListValue : attributesToGetList) { if (attributesToGetListValue != null) { jsonWriter.value(attributesToGetListValue); } } jsonWriter.endArray(); } if (requestItemsListValue.getValue().isConsistentRead() != null) { jsonWriter .key("ConsistentRead") .value(requestItemsListValue.getValue().isConsistentRead()); } jsonWriter.endObject(); } } jsonWriter.endObject(); } jsonWriter.endObject(); String snippet = stringWriter.toString(); byte[] content = snippet.getBytes("UTF-8"); request.setContent(new StringInputStream(snippet)); request.addHeader("Content-Length", Integer.toString(content.length)); } catch (Throwable t) { throw new AmazonClientException("Unable to marshall request to JSON: " + t.getMessage(), t); } return request; }