use of org.apache.rya.indexing.entity.model.Entity in project incubator-rya by apache.
the class MongoEntityStorageIT method update_differentSubjects.
@Test(expected = EntityStorageException.class)
public void update_differentSubjects() throws Exception {
// Two objects that do not have the same Subjects.
final Entity old = Entity.builder().setSubject(new RyaURI("urn:SSN/111-11-1111")).setExplicitType(new RyaURI("urn:person")).build();
final Entity updated = Entity.builder().setSubject(new RyaURI("urn:SSN/222-22-2222")).setExplicitType(new RyaURI("urn:person")).build();
// The update will fail.
final EntityStorage storage = new MongoEntityStorage(super.getMongoClient(), RYA_INSTANCE_NAME);
storage.update(old, updated);
}
use of org.apache.rya.indexing.entity.model.Entity in project incubator-rya by apache.
the class BaseEntityIndexer method deleteStatement.
@Override
public void deleteStatement(final RyaStatement statement) throws IOException {
requireNonNull(statement);
final EntityStorage entities = this.entities.get();
final TypeStorage types = this.types.get();
checkState(entities != null, "Must set this indexers configuration before storing statements.");
checkState(types != null, "Must set this indexers configuration before storing statements.");
try {
new EntityUpdater(entities).update(statement.getSubject(), old -> {
// If there is no Entity for the subject of the statement, then do nothing.
if (!old.isPresent()) {
return Optional.empty();
}
final Entity oldEntity = old.get();
// Increment the version of the Entity.
final Entity.Builder updated = Entity.builder(oldEntity);
updated.setVersion(oldEntity.getVersion() + 1);
if (TYPE_URI.equals(statement.getPredicate())) {
// If the Type ID already isn't in the list of explicit types, then do nothing.
final RyaURI typeId = new RyaURI(statement.getObject().getData());
if (!oldEntity.getExplicitTypeIds().contains(typeId)) {
return Optional.empty();
}
// Otherwise remove it from the list.
updated.unsetExplicitType(typeId);
} else {
// If the deleted property appears within the old entity's properties, then remove it.
final RyaURI deletedPropertyName = statement.getPredicate();
boolean propertyWasPresent = false;
for (final RyaURI typeId : oldEntity.getProperties().keySet()) {
for (final RyaURI propertyName : oldEntity.getProperties().get(typeId).keySet()) {
if (deletedPropertyName.equals(propertyName)) {
propertyWasPresent = true;
updated.unsetProperty(typeId, deletedPropertyName);
}
}
}
// If no properties were removed, then do nothing.
if (!propertyWasPresent) {
return Optional.empty();
}
}
return Optional.of(updated.build());
});
} catch (final IndexingException e) {
throw new IOException("Failed to update the Entity index.", e);
}
}
use of org.apache.rya.indexing.entity.model.Entity in project incubator-rya by apache.
the class BaseEntityIndexer method updateEntity.
/**
* Updates a {@link Entity} to reflect new {@link RyaStatement}s.
*
* @param subject - The Subject of the {@link Entity} the statements are for. (not null)
* @param statements - Statements that the {@link Entity} will be updated with. (not null)
* @throws IndexingException
*/
private void updateEntity(final RyaURI subject, final Collection<RyaStatement> statements) throws IndexingException {
requireNonNull(subject);
requireNonNull(statements);
final EntityStorage entities = this.entities.get();
final TypeStorage types = this.types.get();
checkState(entities != null, "Must set this indexers configuration before storing statements.");
checkState(types != null, "Must set this indexers configuration before storing statements.");
new EntityUpdater(entities).update(subject, old -> {
// Create a builder with the updated Version.
final Entity.Builder updated;
if (!old.isPresent()) {
updated = Entity.builder().setSubject(subject).setVersion(0);
} else {
final int updatedVersion = old.get().getVersion() + 1;
updated = Entity.builder(old.get()).setVersion(updatedVersion);
}
// Update the entity based on the Statements.
for (final RyaStatement statement : statements) {
// The Statement is setting an Explicit Type ID for the Entity.
if (Objects.equal(TYPE_URI, statement.getPredicate())) {
final RyaURI typeId = new RyaURI(statement.getObject().getData());
updated.setExplicitType(typeId);
} else // The Statement is adding a Property to the Entity.
{
final RyaURI propertyName = statement.getPredicate();
final RyaType propertyValue = statement.getObject();
try (final ConvertingCursor<Type> typesIt = types.search(propertyName)) {
// Set the Property for each type that includes the Statement's predicate.
while (typesIt.hasNext()) {
final RyaURI typeId = typesIt.next().getId();
updated.setProperty(typeId, new Property(propertyName, propertyValue));
}
} catch (final TypeStorageException | IOException e) {
throw new RuntimeException("Failed to fetch Types that include the property name '" + statement.getPredicate().getData() + "'.", e);
}
}
}
return Optional.of(updated.build());
});
}
use of org.apache.rya.indexing.entity.model.Entity in project incubator-rya by apache.
the class EntityDocumentConverterTest method to_and_from_document.
@Test
public void to_and_from_document() throws DocumentConverterException {
// Convert an Entity into a Document.
final Entity entity = Entity.builder().setSubject(new RyaURI("urn:alice")).setExplicitType(new RyaURI("urn:person")).setProperty(new RyaURI("urn:person"), new Property(new RyaURI("urn:age"), new RyaType(XMLSchema.INT, "30"))).setProperty(new RyaURI("urn:person"), new Property(new RyaURI("urn:eye"), new RyaType("blue"))).setProperty(new RyaURI("urn:employee"), new Property(new RyaURI("urn:hours"), new RyaType(XMLSchema.INT, "40"))).setProperty(new RyaURI("urn:employee"), new Property(new RyaURI("urn:employer"), new RyaType("Burger Joint"))).build();
final Document document = new EntityDocumentConverter().toDocument(entity);
// Convert the Document back into an Entity.
final Entity converted = new EntityDocumentConverter().fromDocument(document);
// Ensure the original matches the round trip converted Entity.
assertEquals(entity, converted);
}
use of org.apache.rya.indexing.entity.model.Entity in project incubator-rya by apache.
the class MongoDbSmartUriIT method testSerializeDeserialize.
@Test
public void testSerializeDeserialize() throws SmartUriException, URISyntaxException {
final URI smartUri = SmartUriAdapter.serializeUriEntity(BOB_ENTITY);
final Entity resultEntity = SmartUriAdapter.deserializeUriEntity(smartUri);
assertEquals(BOB_ENTITY.getSubject(), resultEntity.getSubject());
}
Aggregations