return new BatchWriteItemResult();
}
<<<<<<< HEAD
protected ScanResult scan(ScanRequest request) {
ScanResult result = new ScanResult();
List errors = new ScanRequestValidator().validate(request);
if (errors.size() > 0) {
throw createInternalServerException(errors);
}
result.setConsumedCapacityUnits(0.5);
result.setLastEvaluatedKey(new Key());
List> items = new ArrayList>();
for (String key : this.tables.get(request.getTableName()).getItems().keySet()) {
Map item = this.tables.get(request.getTableName()).getItem(key);
if (request.getScanFilter() != null) {
for (String k : request.getScanFilter().keySet()) {
Condition cond = request.getScanFilter().get(k);
if(cond.getComparisonOperator()==null){
throw new ResourceNotFoundException("There must be a comparisonOperator");
}
if (cond.getComparisonOperator().equals("EQ")) {
if (cond.getAttributeValueList().size() == 1) {
EQHandle(items, item, cond.getAttributeValueList().get(0), k);
}
else {
//TODO: throw Exception
}
}
if (cond.getComparisonOperator().equals("LE")) {
if (cond.getAttributeValueList().size() == 1) {
LEHandle(items, item, cond.getAttributeValueList().get(0), k);
}
else {
//TODO: throw Exception
}
}
if (cond.getComparisonOperator().equals("LT")) {
if (cond.getAttributeValueList().size() == 1) {
LTHandle(items, item, cond.getAttributeValueList().get(0), k);
}
else {
//TODO: throw Exception
}
}
if (cond.getComparisonOperator().equals("GE")) {
if (errors.size() > 0) {
if (cond.getAttributeValueList().size() == 1) {
GEHandle(items, item, cond.getAttributeValueList().get(0), k);
}
else {
//TODO: throw Exception
}
}
if (cond.getComparisonOperator().equals("GT")) {
if (cond.getAttributeValueList().size() == 1) {
GTHandle(items, item, cond.getAttributeValueList().get(0), k);
}
else {
//TODO: throw Exception
}
}
if (cond.getComparisonOperator().equals("BETWEEN")) {
if (cond.getAttributeValueList().size() == 2) {
BETWEENHandle(items, item, cond.getAttributeValueList().get(0), cond.getAttributeValueList().get(1), k);
}
else {
//TODO: throw Exception
}
}
if (cond.getComparisonOperator().equals("IN")) {
for (AttributeValue attrivuteValue : cond.getAttributeValueList()) {
EQHandle(items, item, attrivuteValue, k);
}
}
}
} else {
items.add(item);
}
}
if (request.getLimit() != null) {
items = items.subList(0, request.getLimit() - 1);
}
for (Map item : items) {
result.setLastEvaluatedKey(new Key(item.get(this.tables.get(request.getTableName()).getHashKeyName())));
}
if (request.getAttributesToGet() != null) {
List> copy = new ArrayList>();
for (Map item : items) {
Set keyz = new HashSet(item.keySet());
for(String sKey : keyz){
if(!request.getAttributesToGet().contains(sKey)){
item.remove(sKey);
}
}
copy.add(item);
}
items = copy;
}
if(items.size()==0){
result.setLastEvaluatedKey(new Key(new AttributeValue().withS(null)));
}
result.setItems(items);
result.setCount(items.size());
result.setScannedCount(items.size());
return result;
}
public QueryResult query(QueryRequest request) {
// Validate data coming in
QueryRequestValidator validator = new QueryRequestValidator();
List errors = validator.validate(request);
if (errors.size() != 0) {
throw createInternalServerException(errors);
}
String keyValue = getKeyValue(request.getHashKeyValue());
// Check existence of table
Table table = this.tables.get(request.getTableName());
if (table == null) {
throw new ResourceNotFoundException("The table '" + request.getTableName() + "' doesn't exist.");
}
Map item = table.getItem(keyValue);
QueryResult queryResult = new QueryResult();
List> list = null;
if(item==null) {
item = new HashMap();
}
list = new ArrayList>();
list.add(item);
=======
protected ScanResult scan(ScanRequest request) {
ScanResult result = new ScanResult();
List errors = new ScanRequestValidator().validate(request);
throw createInternalServerException(errors);
}
result.setConsumedCapacityUnits(0.5);
result.setLastEvaluatedKey(new Key());
List> items = new ArrayList>();
for (String key : this.tables.get(request.getTableName()).getItems().keySet()) {
Map item = this.tables.get(request.getTableName()).getItem(key);
if (request.getScanFilter() != null) {
for (String k : request.getScanFilter().keySet()) {
Condition cond = request.getScanFilter().get(k);
if (cond.getComparisonOperator() == null) {
throw new ResourceNotFoundException("There must be a comparisonOperator");
}
if (cond.getComparisonOperator().equals("EQ")) {
if (cond.getAttributeValueList().size() == 1) {
if (item.get(k).equals(cond.getAttributeValueList().get(0))) {
items.add(item);
}
} else {
if (item.get(k).equals(cond.getAttributeValueList())) {
items.add(item);
}
}
}
if (cond.getComparisonOperator().equals("LE")) {
if (cond.getAttributeValueList().size() == 1) {
if (getAttributeValueType(item.get(k)).equals(AttributeValueType.S) || getAttributeValueType(item.get(k)).equals(AttributeValueType.N)) {
String value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.S)) ? item.get(k).getS() : item.get(k).getN();
String comp = (getAttributeValueType(cond.getAttributeValueList().get(0)).equals(AttributeValueType.S)) ? cond.getAttributeValueList().get(0).getS() : cond.getAttributeValueList().get(0).getN();
if (value.compareTo(comp) >= 0) {
items.add(item);
}
} else {
//TODO to do
//List value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.SS))? item.get(k).getSS() : item.get(k).getNS();
}
} else {
//TODO to do
if (item.get(k).equals(cond.getAttributeValueList())) {
items.add(item);
}
}
}
if (cond.getComparisonOperator().equals("LT")) {
if (cond.getAttributeValueList().size() == 1) {
if (getAttributeValueType(item.get(k)).equals(AttributeValueType.S) || getAttributeValueType(item.get(k)).equals(AttributeValueType.N)) {
String value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.S)) ? item.get(k).getS() : item.get(k).getN();
String comp = (getAttributeValueType(cond.getAttributeValueList().get(0)).equals(AttributeValueType.S)) ? cond.getAttributeValueList().get(0).getS() : cond.getAttributeValueList().get(0).getN();
if (value.compareTo(comp) < 0) {
items.add(item);
}
} else {
//TODO to do
//List value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.SS))? item.get(k).getSS() : item.get(k).getNS();
}
} else {
//TODO to do
if (item.get(k).equals(cond.getAttributeValueList())) {
items.add(item);
}
}
}
if (cond.getComparisonOperator().equals("GE")) {
if (cond.getAttributeValueList().size() == 1) {
if (getAttributeValueType(item.get(k)).equals(AttributeValueType.S) || getAttributeValueType(item.get(k)).equals(AttributeValueType.N)) {
List> list = null;
String value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.S)) ? item.get(k).getS() : item.get(k).getN();
String comp = (getAttributeValueType(cond.getAttributeValueList().get(0)).equals(AttributeValueType.S)) ? cond.getAttributeValueList().get(0).getS() : cond.getAttributeValueList().get(0).getN();
if (value.compareTo(comp) <= 0) {
items.add(item);
}
} else {
//TODO to do
//List value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.SS))? item.get(k).getSS() : item.get(k).getNS();
}
} else {
//TODO to do
if (item.get(k).equals(cond.getAttributeValueList())) {
items.add(item);
}
}
}
if (cond.getComparisonOperator().equals("GT")) {
if (cond.getAttributeValueList().size() == 1) {
if (getAttributeValueType(item.get(k)).equals(AttributeValueType.S) || getAttributeValueType(item.get(k)).equals(AttributeValueType.N)) {
if (getAttributeValueType(item.get(k)).equals(AttributeValueType.S)) {
String value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.S)) ? item.get(k).getS() : item.get(k).getN();
String comp = (getAttributeValueType(cond.getAttributeValueList().get(0)).equals(AttributeValueType.S)) ? cond.getAttributeValueList().get(0).getS() : cond.getAttributeValueList().get(0).getN();
if (value.compareTo(comp) > 0) {
items.add(item);
}
} else {
String value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.S)) ? item.get(k).getS() : item.get(k).getN();
String comp = (getAttributeValueType(cond.getAttributeValueList().get(0)).equals(AttributeValueType.S)) ? cond.getAttributeValueList().get(0).getS() : cond.getAttributeValueList().get(0).getN();
if (Integer.parseInt(value) > Integer.parseInt(comp)) {
items.add(item);
}
}
} else {
//TODO to do
//List value = (getAttributeValueType(item.get(k)).equals(AttributeValueType.SS))? item.get(k).getSS() : item.get(k).getNS();
}
} else {
//TODO to do
if (item.get(k).equals(cond.getAttributeValueList())) {
items.add(item);
}
}
}
}
} else {
items.add(item);
}
}
if (request.getLimit() != null) {
items = items.subList(0, request.getLimit() - 1);
}
for (Map item : items) {
result.setLastEvaluatedKey(new Key(item.get(this.tables.get(request.getTableName()).getHashKeyName())));
}
if (request.getAttributesToGet() != null) {
List> copy = getItemWithAttributesToGet(items, request.getAttributesToGet());
items = copy;
}
if (items.size() == 0) {
result.setLastEvaluatedKey(new Key(new AttributeValue().withS(null)));
}
result.setItems(items);
result.setCount(items.size());
result.setScannedCount(items.size());
return result;
}
public QueryResult query(QueryRequest request) {
// Validate data coming in
QueryRequestValidator validator = new QueryRequestValidator();
List errors = validator.validate(request);
if (errors.size() != 0) {
throw createInternalServerException(errors);
}
String keyValue = getKeyValue(request.getHashKeyValue());
// Check existence of table
Table table = this.tables.get(request.getTableName());
if (table == null) {
throw new ResourceNotFoundException("The table '" + request.getTableName() + "' doesn't exist.");
}
Map item = table.getItem(keyValue);
QueryResult queryResult = new QueryResult();
if (item == null) {
item = new HashMap();
}
list = new ArrayList>();
list.add(item);
>>>>>>> 962598cede6f12d3739b8c81462a7ef5447953e7
queryResult.setItems(list);
queryResult.setCount(list.size());
queryResult.setConsumedCapacityUnits(0.5); |