@Test
public void testSpecific() throws Exception {
// Construct Dao
Dao<CompositeRecord> dao =
SpecificAvroDao.buildCompositeDao(
tablePool,
tableName,
Arrays.asList(subRecord1String, subRecord2String),
CompositeRecord.class);
// Construct records
SubRecord1 subRecord1 =
SubRecord1.newBuilder()
.setKeyPart1("1")
.setKeyPart2("1")
.setField1("field1_1")
.setField2("field1_2")
.build();
SubRecord2 subRecord2 =
SubRecord2.newBuilder()
.setKeyPart1("1")
.setKeyPart2("1")
.setField1("field2_1")
.setField2("field2_2")
.build();
CompositeRecord compositeRecord =
CompositeRecord.newBuilder().setSubRecord1(subRecord1).setSubRecord2(subRecord2).build();
// Test put and get
dao.put(compositeRecord);
PartitionKey key = dao.getPartitionStrategy().partitionKey("1", "1");
CompositeRecord returnedCompositeRecord = dao.get(key);
assertEquals("field1_1", returnedCompositeRecord.getSubRecord1().getField1());
assertEquals("field1_2", returnedCompositeRecord.getSubRecord1().getField2());
assertEquals("field2_1", returnedCompositeRecord.getSubRecord2().getField1());
assertEquals("field2_2", returnedCompositeRecord.getSubRecord2().getField2());
// Test OCC
assertFalse(dao.put(compositeRecord));
assertTrue(dao.put(returnedCompositeRecord));
// Test null field
subRecord1 =
SubRecord1.newBuilder()
.setKeyPart1("1")
.setKeyPart2("2")
.setField1("field1_1")
.setField2("field1_2")
.build();
compositeRecord = CompositeRecord.newBuilder().setSubRecord1(subRecord1).build();
dao.put(compositeRecord);
key = dao.getPartitionStrategy().partitionKey("1", "2");
compositeRecord = dao.get(key);
assertEquals(null, compositeRecord.getSubRecord2());
}
/**
* Update the married status of a new user record.
*
* <p>This method demonstrates updating both a UserProfileModel and a UserActionsModel with a
* single HBase request using the composite dao. It performs a get/update/put operation, which is
* protected by the check_conflict field on UserProfileModel from colliding with another
* get/update/put operation.
*
* @param firstName The first name of the user we are updating
* @param lastName The last name of the user we are updating
* @param married True if this person is married. Otherwise false.
*/
public void updateUserProfile(String firstName, String lastName, boolean married) {
// Get the timestamp we'll use to set the value of the profile_updated
// action.
long ts = System.currentTimeMillis();
// Construct the key we'll use to fetch the user.
PartitionKey key =
userProfileActionsDao.getPartitionStrategy().partitionKey(lastName, firstName);
// Get the profile and actions entity from the composite dao.
UserProfileActionsModel profileActionsModel = userProfileActionsDao.get(key);
// Updating the married status is hairy since our avro compiler isn't setup
// to compile setters for fields. We have to construct a clone through the
// builder.
UserProfileActionsModel updatedProfileActionsModel =
UserProfileActionsModel.newBuilder(profileActionsModel)
.setUserProfileModel(
UserProfileModel.newBuilder(profileActionsModel.getUserProfileModel())
.setMarried(married)
.build())
.build();
// Since maps are mutable, we can update the actions map without having to
// go through the builder like above.
updatedProfileActionsModel
.getUserActionsModel()
.getActions()
.put("profile_updated", Long.toString(ts));
if (!userProfileActionsDao.put(updatedProfileActionsModel)) {
// If put returns false, a write conflict occurred where someone else
// updated the row between the times we did the get and put.
System.out.println("Updating the user profile failed due to a write conflict");
}
}
/**
* Print the user profiles and actions for all users with the provided last name
*
* <p>This method demonstrates how to open a scanner with a start key. It's using the composite
* dao, so the records it returns will be a composite of both the profile model and actions model.
*
* @param lastName The last name of users to scan.
*/
public void printUserProfileActionsForLastName(String lastName) {
// Create a partial key that will allow us to start the scanner from the
// first user record that has last name equal to the one provided.
PartitionKey startKey = userProfileActionsDao.getPartitionStrategy().partitionKey("lastName");
// Get the scanner with the start key. Null for stopKey in the getScanner
// method indicates that the scanner will scan to the end of the table. Our
// loop will break out when it encounters a record without the last name.
EntityScanner<UserProfileActionsModel> scanner =
userProfileActionsDao.getScanner(startKey, null);
scanner.open();
try {
// scan until we find a last name not equal to the one provided
for (UserProfileActionsModel entity : scanner) {
if (!entity.getUserProfileModel().getLastName().equals(lastName)) {
// last name of row different, break out of the scan.
break;
}
System.out.println(entity.toString());
}
} finally {
// scanners need to be closed.
scanner.close();
}
}