Menu
Amazon DynamoDB
Developer Guide (API Version 2012-08-10)

TryDaxTests.java

The TryDaxTests.java file contains methods that perform read operations against a test table in DynamoDB. These methods are not concerned with how they access the data (using either the DynamoDB client or the DAX client), so there is no need to modify the application logic.

You can modify the program in several ways. For example:

  • Modify the queryTest method so that it uses a different KeyConditionExpression.

  • Add a ScanFilter to the scanTest method, so that only some of the items are returned to you.

Note

To run this program, you must include both the DAX Java client and the AWS SDK for Java in your classpath. See Step 4a: (Java) Run the Sample Application for an example of setting your CLASSPATH variable.

Copy
import java.util.HashMap; import java.util.Iterator; import com.amazonaws.services.dynamodbv2.document.DynamoDB; import com.amazonaws.services.dynamodbv2.document.Item; import com.amazonaws.services.dynamodbv2.document.ItemCollection; import com.amazonaws.services.dynamodbv2.document.QueryOutcome; import com.amazonaws.services.dynamodbv2.document.ScanOutcome; import com.amazonaws.services.dynamodbv2.document.Table; import com.amazonaws.services.dynamodbv2.document.spec.QuerySpec; public class TryDaxTests { void getItemTest(String tableName, DynamoDB client, int pk, int sk, int iterations) { long startTime, endTime; System.out.println("GetItem test - partition key " + pk + " and sort keys 1-" + sk); Table table = client.getTable(tableName); for (int i = 0; i < iterations; i++) { startTime = System.nanoTime(); try { for (Integer ipk = 1; ipk <= pk; ipk++) { for (Integer isk = 1; isk <= sk; isk++) { table.getItem("pk", ipk, "sk", isk); } } } catch (Exception e) { System.err.println("Unable to get item:"); e.printStackTrace(); } endTime = System.nanoTime(); printTime(startTime, endTime, pk * sk); } } void queryTest(String tableName, DynamoDB client, int pk, int sk1, int sk2, int iterations) { long startTime, endTime; System.out.println("Query test - partition key " + pk + " and sort keys between " + sk1 + " and " + sk2); Table table = client.getTable(tableName); HashMap<String, Object> valueMap = new HashMap<String, Object>(); valueMap.put(":pkval", pk); valueMap.put(":skval1", sk1); valueMap.put(":skval2", sk2); QuerySpec spec = new QuerySpec() .withKeyConditionExpression("pk = :pkval and sk between :skval1 and :skval2") .withValueMap(valueMap); for (int i = 0; i < iterations; i++) { startTime = System.nanoTime(); ItemCollection<QueryOutcome> items = table.query(spec); try { Iterator<Item> iter = items.iterator(); while (iter.hasNext()) { iter.next(); } } catch (Exception e) { System.err.println("Unable to query table:"); e.printStackTrace(); } endTime = System.nanoTime(); printTime(startTime, endTime, iterations); } } void scanTest(String tableName, DynamoDB client, int iterations) { long startTime, endTime; System.out.println("Scan test - all items in the table"); Table table = client.getTable(tableName); for (int i = 0; i < iterations; i++) { startTime = System.nanoTime(); ItemCollection<ScanOutcome> items = table.scan(); try { Iterator<Item> iter = items.iterator(); while (iter.hasNext()) { iter.next(); } } catch (Exception e) { System.err.println("Unable to scan table:"); e.printStackTrace(); } endTime = System.nanoTime(); printTime(startTime, endTime, iterations); } } public void printTime(long startTime, long endTime, int iterations) { System.out.format("\tTotal time: %.3f ms - ", (endTime - startTime) / (1000000.0)); System.out.format("Avg time: %.3f ms\n", (endTime - startTime) / (iterations * 1000000.0)); } }