Ejemplo n.º 1
0
  /**
   * Wait for the table to reach the desired status and returns the table description
   *
   * @param dynamo Dynamo client to use
   * @param tableName Table name to poll status of
   * @param desiredStatus Desired {@link TableStatus} to wait for. If null this method simply waits
   *     until DescribeTable returns something non-null (i.e. any status)
   * @param timeout Timeout in milliseconds to continue to poll for desired status
   * @param interval Time to wait in milliseconds between poll attempts
   * @return Null if DescribeTables never returns a result, otherwise the result of the last poll
   *     attempt (which may or may not have the desired state)
   * @throws InterruptedException
   * @throws {@link IllegalArgumentException} If timeout or interval is invalid
   */
  private static TableDescription waitForTableDescription(
      final AmazonDynamoDB dynamo,
      final String tableName,
      TableStatus desiredStatus,
      final int timeout,
      final int interval)
      throws InterruptedException, IllegalArgumentException {
    if (timeout < 0) {
      throw new IllegalArgumentException("Timeout must be >= 0");
    }
    if (interval <= 0 || interval >= timeout) {
      throw new IllegalArgumentException("Interval must be > 0 and < timeout");
    }
    long startTime = System.currentTimeMillis();
    long endTime = startTime + timeout;

    TableDescription table = null;
    while (System.currentTimeMillis() < endTime) {
      try {
        table = dynamo.describeTable(new DescribeTableRequest(tableName)).getTable();
        if (desiredStatus == null || table.getTableStatus().equals(desiredStatus.toString())) {
          return table;
        }
      } catch (ResourceNotFoundException rnfe) {
        // ResourceNotFound means the table doesn't exist yet,
        // so ignore this error and just keep polling.
      }

      Thread.sleep(interval);
    }
    return table;
  }
Ejemplo n.º 2
0
  /**
   * Waits up to a specified amount of time for a specified DynamoDB table to move into the <code>
   * ACTIVE</code> state. If the table does not exist or does not transition to the <code>ACTIVE
   * </code> state after this time, then an AmazonClientException is thrown.
   *
   * @param dynamo The DynamoDB client to use to make requests.
   * @param tableName The name of the table whose status is being checked.
   * @param timeout The maximum number of milliseconds to wait.
   * @param interval The poll interval in milliseconds.
   * @throws TableNeverTransitionedToStateException If the specified table does not exist or does
   *     not transition into the <code>ACTIVE</code> state before this method times out and stops
   *     polling.
   * @throws InterruptedException If the thread is interrupted while waiting for the table to
   *     transition into the <code>ACTIVE</code> state.
   */
  public static void waitUntilActive(
      final AmazonDynamoDB dynamo, final String tableName, final int timeout, final int interval)
      throws InterruptedException, TableNeverTransitionedToStateException {
    TableDescription table =
        waitForTableDescription(dynamo, tableName, TableStatus.ACTIVE, timeout, interval);

    if (table == null || !table.getTableStatus().equals(TableStatus.ACTIVE.toString())) {
      throw new TableNeverTransitionedToStateException(tableName, TableStatus.ACTIVE);
    }
  }
  @Test
  public void createsGlobalIndices() {
    List<TableDescription> tables = DynamoTestUtils.MAPPER.getTables(TestTask.class);
    TableDescription table = DynamoTestUtils.getTableByName(tables, "TestTask");
    assertEquals(3, table.getGlobalSecondaryIndexes().size());

    GlobalSecondaryIndexDescription index =
        findIndex(table.getGlobalSecondaryIndexes(), "guid-index");
    assertEquals("INCLUDE", index.getProjection().getProjectionType());
    assertEquals("guid", index.getKeySchema().get(0).getAttributeName());
    assertEquals(Long.valueOf(18), index.getProvisionedThroughput().getWriteCapacityUnits());
    assertEquals(Long.valueOf(20), index.getProvisionedThroughput().getReadCapacityUnits());

    index = findIndex(table.getGlobalSecondaryIndexes(), "healthCode-scheduledOn-index");
    assertEquals("ALL", index.getProjection().getProjectionType());
    assertEquals("healthCode", index.getKeySchema().get(0).getAttributeName());
    assertEquals("scheduledOn", index.getKeySchema().get(1).getAttributeName());
    assertEquals(Long.valueOf(18), index.getProvisionedThroughput().getWriteCapacityUnits());
    assertEquals(Long.valueOf(20), index.getProvisionedThroughput().getReadCapacityUnits());

    index = findIndex(table.getGlobalSecondaryIndexes(), "healthCode-expiresOn-index");
    assertEquals("expiresOn", index.getKeySchema().get(0).getAttributeName());
  }
 @Test
 public void testGetAnnotatedTables() {
   List<TableDescription> tables = DynamoTestUtils.MAPPER.getTables(DynamoTestUtils.PACKAGE);
   assertNotNull(tables);
   assertEquals(2, tables.size());
   Map<String, TableDescription> tableMap = new HashMap<String, TableDescription>();
   for (TableDescription table : tables) {
     tableMap.put(table.getTableName(), table);
   }
   String tableName =
       Environment.LOCAL.name().toLowerCase()
           + "-"
           + DynamoTestUtils.class.getSimpleName()
           + "-TestHealthDataRecord";
   TableDescription table = tableMap.get(tableName);
   assertNotNull(table);
   assertEquals(2, table.getKeySchema().size());
   assertEquals(5, table.getAttributeDefinitions().size());
   assertEquals(2, table.getLocalSecondaryIndexes().size());
   assertEquals(1, table.getGlobalSecondaryIndexes().size());
   assertEquals(30L, table.getProvisionedThroughput().getReadCapacityUnits().longValue());
   assertEquals(50L, table.getProvisionedThroughput().getWriteCapacityUnits().longValue());
 }