Search in sources :

Example 21 with Expected

use of com.amazonaws.services.dynamodbv2.document.Expected in project thunder by RohanNagar.

the class UsersDao method delete.

/**
 * Delete a PilotUser in the data store.
 *
 * @param email The email of the user to delete.
 * @return The PilotUser object that was deleted or {@code null} if the delete failed.
 * @throws DatabaseException If the user is not found or if the database is down.
 */
public PilotUser delete(String email) {
    checkNotNull(email);
    // Get the item that will be deleted to return it
    Item item;
    try {
        item = table.getItem("email", email);
    } catch (AmazonClientException e) {
        LOG.error("The database is currently unresponsive.", e);
        throw new DatabaseException("The database is currently unavailable.", DatabaseError.DATABASE_DOWN);
    }
    DeleteItemSpec deleteItemSpec = new DeleteItemSpec().withPrimaryKey("email", email).withExpected(new Expected("email").exists());
    try {
        table.deleteItem(deleteItemSpec);
    } catch (ConditionalCheckFailedException e) {
        LOG.warn("The email {} was not found in the database.", email, e);
        throw new DatabaseException("The user to delete was not found.", DatabaseError.USER_NOT_FOUND);
    } catch (AmazonClientException e) {
        LOG.error("The database is currently unresponsive.", e);
        throw new DatabaseException("The database is currently unavailable.", DatabaseError.DATABASE_DOWN);
    }
    return fromJson(mapper, item.getJSON("document"));
}
Also used : Item(com.amazonaws.services.dynamodbv2.document.Item) DeleteItemSpec(com.amazonaws.services.dynamodbv2.document.spec.DeleteItemSpec) Expected(com.amazonaws.services.dynamodbv2.document.Expected) AmazonClientException(com.amazonaws.AmazonClientException) ConditionalCheckFailedException(com.amazonaws.services.dynamodbv2.model.ConditionalCheckFailedException)

Example 22 with Expected

use of com.amazonaws.services.dynamodbv2.document.Expected in project thunder by RohanNagar.

the class UsersDao method update.

/**
 * Update a PilotUser in the data store.
 *
 * @param existingEmail The existing email of the user.
 *                     This must not be {@code null} if the email is to be changed.
 * @param user The user object to update. Must have the same email as the one to update.
 * @return The PilotUser object that was updated or {@code null} if the updated failed.
 * @throws DatabaseException If the user is not found, the database is down, or the update fails.
 */
public PilotUser update(@Nullable String existingEmail, PilotUser user) {
    checkNotNull(user);
    // Different emails means we need to delete and insert
    if (existingEmail != null && !existingEmail.equals(user.getEmail().getAddress())) {
        LOG.info("User to update has new email. The user will be deleted and then reinserted.");
        delete(existingEmail);
        return insert(user);
    }
    // Compute the new data
    long now = Instant.now().toEpochMilli();
    String newVersion = UUID.randomUUID().toString();
    String document = toJson(mapper, user);
    // Get the old version
    Item item;
    try {
        item = table.getItem("email", user.getEmail().getAddress());
    } catch (AmazonClientException e) {
        LOG.error("The database is currently unresponsive.", e);
        throw new DatabaseException("The database is currently unavailable.", DatabaseError.DATABASE_DOWN);
    }
    if (item == null) {
        LOG.warn("The user {} was not found in the database.", user.getEmail().getAddress());
        throw new DatabaseException("The user was not found.", DatabaseError.USER_NOT_FOUND);
    }
    String oldVersion = item.getString("version");
    Item newItem = item.withString("version", newVersion).withLong("update_time", now).withJSON("document", document);
    try {
        table.putItem(newItem, new Expected("version").eq(oldVersion));
    } catch (ConditionalCheckFailedException e) {
        LOG.error("The user was updated while this update was in progress." + " Aborting to avoid race condition.", e);
        throw new DatabaseException("The user to update is at an unexpected stage.", DatabaseError.CONFLICT);
    } catch (AmazonServiceException e) {
        LOG.error("The database rejected the update request.", e);
        throw new DatabaseException("The database rejected the update request.", DatabaseError.REQUEST_REJECTED);
    } catch (AmazonClientException e) {
        LOG.error("The database is currently unresponsive.", e);
        throw new DatabaseException("The database is currently unavailable.", DatabaseError.DATABASE_DOWN);
    }
    return user;
}
Also used : Item(com.amazonaws.services.dynamodbv2.document.Item) Expected(com.amazonaws.services.dynamodbv2.document.Expected) AmazonClientException(com.amazonaws.AmazonClientException) AmazonServiceException(com.amazonaws.AmazonServiceException) ConditionalCheckFailedException(com.amazonaws.services.dynamodbv2.model.ConditionalCheckFailedException)

Aggregations

AttributeValue (com.amazonaws.services.dynamodbv2.model.AttributeValue)14 Test (org.junit.Test)14 ByteArrayInputStream (java.io.ByteArrayInputStream)10 ByteArrayOutputStream (java.io.ByteArrayOutputStream)10 Expected (com.amazonaws.services.dynamodbv2.document.Expected)5 ConditionalCheckFailedException (com.amazonaws.services.dynamodbv2.model.ConditionalCheckFailedException)5 AmazonClientException (com.amazonaws.AmazonClientException)4 UpdateItemSpec (com.amazonaws.services.dynamodbv2.document.spec.UpdateItemSpec)4 HashMap (java.util.HashMap)4 Item (com.amazonaws.services.dynamodbv2.document.Item)3 ExpectedAttributeValue (com.amazonaws.services.dynamodbv2.model.ExpectedAttributeValue)3 AmazonServiceException (com.amazonaws.AmazonServiceException)2 AttributeUpdate (com.amazonaws.services.dynamodbv2.document.AttributeUpdate)2 X509CertRecord (com.yahoo.athenz.common.server.cert.X509CertRecord)2 DescribeAutoScalingGroupsRequest (com.amazonaws.services.autoscaling.model.DescribeAutoScalingGroupsRequest)1 DescribeAutoScalingGroupsResult (com.amazonaws.services.autoscaling.model.DescribeAutoScalingGroupsResult)1 ListMetricsRequest (com.amazonaws.services.cloudwatch.model.ListMetricsRequest)1 ListMetricsResult (com.amazonaws.services.cloudwatch.model.ListMetricsResult)1 PutMetricDataRequest (com.amazonaws.services.cloudwatch.model.PutMetricDataRequest)1 PutMetricDataResult (com.amazonaws.services.cloudwatch.model.PutMetricDataResult)1