use of org.apache.rya.api.domain.RyaStatement.RyaStatementBuilder in project incubator-rya by apache.
the class MongoDBRyaDAO2IT method testReconstructDao.
@Test
public void testReconstructDao() throws RyaDAOException, IOException {
MongoDBRyaDAO dao = new MongoDBRyaDAO();
try {
dao.setConf(conf);
dao.init();
final RyaStatementBuilder builder = new RyaStatementBuilder();
builder.setPredicate(new RyaURI("http://temp.com"));
builder.setSubject(new RyaURI("http://subject.com"));
builder.setObject(new RyaURI("http://object.com"));
builder.setColumnVisibility(new DocumentVisibility("B").flatten());
final MongoDatabase db = conf.getMongoClient().getDatabase(conf.get(MongoDBRdfConfiguration.MONGO_DB_NAME));
final MongoCollection<Document> coll = db.getCollection(conf.getTriplesCollectionName());
dao.add(builder.build());
assertEquals(coll.count(), 1);
final Document dbo = coll.find().first();
assertTrue(dbo.containsKey(DOCUMENT_VISIBILITY));
assertTrue(dbo.containsKey(TIMESTAMP));
} finally {
dao.destroy();
}
// Test reinitializing the same instance
try {
dao.init();
} finally {
dao.destroy();
}
// Reconstruct new DAO and try again
dao = new MongoDBRyaDAO();
try {
dao.setConf(conf);
dao.init();
final RyaStatementBuilder builder = new RyaStatementBuilder();
builder.setPredicate(new RyaURI("http://temp.com"));
builder.setSubject(new RyaURI("http://subject.com"));
builder.setObject(new RyaURI("http://object.com"));
builder.setColumnVisibility(new DocumentVisibility("B").flatten());
final MongoDatabase db = conf.getMongoClient().getDatabase(conf.get(MongoDBRdfConfiguration.MONGO_DB_NAME));
final MongoCollection<Document> coll = db.getCollection(conf.getTriplesCollectionName());
dao.add(builder.build());
assertEquals(coll.count(), 1);
final Document dbo = coll.find().first();
assertTrue(dbo.containsKey(DOCUMENT_VISIBILITY));
assertTrue(dbo.containsKey(TIMESTAMP));
} finally {
dao.destroy();
}
}
use of org.apache.rya.api.domain.RyaStatement.RyaStatementBuilder in project incubator-rya by apache.
the class MongoDBRyaBatchWriterIT method statement.
private static RyaStatement statement(final int v) {
final RyaStatementBuilder builder = new RyaStatementBuilder();
builder.setPredicate(ryaURI(v));
builder.setSubject(ryaURI(v));
builder.setObject(ryaURI(v));
return builder.build();
}
use of org.apache.rya.api.domain.RyaStatement.RyaStatementBuilder in project incubator-rya by apache.
the class MongoDBRyaDAOIT method testDeleteWildcard.
@Test
public void testDeleteWildcard() throws RyaDAOException {
final MongoDBRyaDAO dao = new MongoDBRyaDAO();
try {
dao.setConf(conf);
dao.init();
final RyaStatementBuilder builder = new RyaStatementBuilder();
builder.setPredicate(new RyaURI("http://temp.com"));
builder.setColumnVisibility(new DocumentVisibility("A").flatten());
dao.delete(builder.build(), conf);
} finally {
dao.destroy();
}
}
use of org.apache.rya.api.domain.RyaStatement.RyaStatementBuilder in project incubator-rya by apache.
the class PipelineQueryIT method insert.
private void insert(Resource subject, URI predicate, Value object, int derivationLevel) throws RyaDAOException {
final RyaStatementBuilder builder = new RyaStatementBuilder();
builder.setSubject(RdfToRyaConversions.convertResource(subject));
builder.setPredicate(RdfToRyaConversions.convertURI(predicate));
builder.setObject(RdfToRyaConversions.convertValue(object));
final RyaStatement rstmt = builder.build();
if (derivationLevel > 0) {
DBObject obj = new SimpleMongoDBStorageStrategy().serialize(builder.build());
obj.put("derivation_level", derivationLevel);
getRyaDbCollection().insert(obj);
} else {
dao.add(rstmt);
}
}
Aggregations