Examples with Item com.amazonaws.services.dynamodbv2.document.Item used on opensource projects

Example 41 with Item

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

the class UsersDaoTest method testSuccessfulDelete.

@Test
public void testSuccessfulDelete() {
    when(table.getItem(anyString(), anyString())).thenReturn(item);
    PilotUser result = usersDao.delete("email");
    verify(table, times(1)).getItem(anyString(), anyString());
    verify(table, times(1)).deleteItem(any(DeleteItemSpec.class));
    assertEquals(user, result);
}

Example 42 with Item

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

the class UsersDaoTest method testSuccessfulUpdate.

@Test
public void testSuccessfulUpdate() {
    when(table.getItem(anyString(), anyString())).thenReturn(item);
    PilotUser result = usersDao.update(null, user);
    verify(table, times(1)).getItem(anyString(), anyString());
    verify(table, times(1)).putItem(any(Item.class), any(Expected.class));
    assertEquals(user, result);
}

Example 43 with Item

use of com.amazonaws.services.dynamodbv2.document.Item 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"));
}

Example 44 with Item

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

the class UsersDao method findByEmail.

/**
 * Find a PilotUser from the data store.
 *
 * @param email The email of the user to find.
 * @return The requested PilotUser or {@code null} if it does not exist.
 * @throws DatabaseException If the user does not exist or if the database is down.
 */
public PilotUser findByEmail(String email) {
    checkNotNull(email);
    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);
    }
    if (item == null) {
        LOG.warn("The email {} was not found in the database.", email);
        throw new DatabaseException("The user was not found.", DatabaseError.USER_NOT_FOUND);
    }
    return fromJson(mapper, item.getJSON("document"));
}

Example 45 with Item

use of com.amazonaws.services.dynamodbv2.document.Item 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;
}