Example #1
0
 private Map<String, Object> getAllPartitionIndexValues(EntityConfig config, Object entity) {
   Map<String, Object> partitionIndexMap = new HashMap<String, Object>();
   for (EntityIndexConfig indexConfig : getPartitionIndexConfigs(config))
     partitionIndexMap.put(
         indexConfig.getIndexName(),
         hive.directory()
             .getPrimaryIndexKeyOfResourceId(
                 config.getResourceName(),
                 // indexConfig.getIndexName(),
                 config.getId(entity)));
   return partitionIndexMap;
 }
Example #2
0
 @SuppressWarnings("unchecked")
 private Map<String, Collection<Object>> getAllSecondaryIndexValues(
     EntityConfig config, Object entity) {
   Map<String, Collection<Object>> secondaryIndexMap = new HashMap<String, Collection<Object>>();
   for (EntityIndexConfig indexConfig : getSecondaryIndexConfigs(config))
     secondaryIndexMap.put(
         indexConfig.getIndexName(),
         hive.directory()
             .getSecondaryIndexKeysWithResourceId(
                 config.getResourceName(), indexConfig.getIndexName(), config.getId(entity)));
   return secondaryIndexMap;
 }
Example #3
0
 private void conditionallyInsertDelegatedResourceIndexes(EntityConfig config, Object entity)
     throws HiveLockableException {
   for (EntityIndexConfig entityIndexConfig : config.getEntityIndexConfigs())
     if (entityIndexConfig.getIndexType().equals(IndexType.Delegates)) {
       final EntityIndexConfigDelegator delegateEntityConfig =
           ((EntityIndexConfigDelegator) entityIndexConfig);
       for (Object value : entityIndexConfig.getIndexValues(entity))
         if (!hive.directory()
             .doesResourceIdExist(
                 delegateEntityConfig.getDelegateEntityConfig().getResourceName(), value))
           insert(
               delegateEntityConfig.getDelegateEntityConfig(),
               delegateEntityConfig.stubEntityInstance(value, config.getPrimaryIndexKey(entity)));
     }
 }
Example #4
0
  public void update(EntityConfig config, Object entity) throws HiveLockableException {
    Map<String, Collection<Object>> secondaryIndexValues =
        getAllSecondaryIndexValues(config, entity);
    Map<String, Object> partitionIndexValues = getAllPartitionIndexValues(config, entity);
    Map<String, Collection<Object>> toDelete = Maps.newHashMap();
    Map<String, Collection<Object>> toInsert = Maps.newHashMap();

    conditionallyInsertDelegatedResourceIndexes(config, entity);

    for (EntityIndexConfig indexConfig : getPartitionIndexConfigs(config)) {
      if (!partitionIndexValues
          .get(indexConfig.getIndexName())
          .equals(indexConfig.getIndexValues(entity).iterator().next())) {
        if (!hive.directory()
            .doesPrimaryIndexKeyExist(indexConfig.getIndexValues(entity).iterator().next())) {
          hive.directory()
              .insertPrimaryIndexKey(indexConfig.getIndexValues(entity).iterator().next());
        }
        hive.directory()
            .updatePrimaryIndexKeyOfResourceId(
                config.getResourceName(),
                config.getId(entity),
                indexConfig.getIndexValues(entity).iterator().next());
      }
    }

    // Delete and insert secondary index values that have changed
    for (EntityIndexConfig indexConfig : getSecondaryIndexConfigs(config)) {
      Pair<Collection<Object>, Collection<Object>> diff =
          Collect.diff(
              secondaryIndexValues.get(indexConfig.getIndexName()),
              indexConfig.getIndexValues(entity));
      toDelete.put(indexConfig.getIndexName(), diff.getKey());
      toInsert.put(indexConfig.getIndexName(), diff.getValue());
    }

    hive.directory()
        .insertSecondaryIndexKeys(config.getResourceName(), toInsert, config.getId(entity));
    hive.directory()
        .deleteSecondaryIndexKeys(config.getResourceName(), toDelete, config.getId(entity));
  }