use of org.apache.rya.streams.api.interactor.AddQuery in project incubator-rya by apache.
the class AddQueryCommand method execute.
@Override
public void execute(final String[] args) throws ArgumentsException, ExecutionException {
requireNonNull(args);
// Parse the command line arguments.
final AddParameters params = new AddParameters();
try {
new JCommander(params, args);
} catch (final ParameterException e) {
throw new ArgumentsException("Could not add a new query because of invalid command line parameters.", e);
}
// Create the Kafka backed QueryChangeLog.
final String bootstrapServers = params.kafkaIP + ":" + params.kafkaPort;
final String topic = KafkaTopics.queryChangeLogTopic(params.ryaInstance);
final QueryChangeLog queryChangeLog = KafkaQueryChangeLogFactory.make(bootstrapServers, topic);
// The AddQuery command doesn't use the scheduled service feature.
final Scheduler scheduler = Scheduler.newFixedRateSchedule(0L, 5, TimeUnit.SECONDS);
final QueryRepository queryRepo = new InMemoryQueryRepository(queryChangeLog, scheduler);
// Execute the add query command.
try {
final AddQuery addQuery = new DefaultAddQuery(queryRepo);
try {
final Boolean isActive = Boolean.parseBoolean(params.isActive);
final Boolean isInsert = Boolean.parseBoolean(params.isInsert);
// If the query's results are meant to be written back to Rya, make sure it creates statements.
if (isInsert) {
final boolean isConstructQuery = QueryInvestigator.isConstruct(params.query);
final boolean isInsertQuery = QueryInvestigator.isInsertWhere(params.query);
if (isConstructQuery) {
System.out.println("WARNING: CONSTRUCT is part of the SPARQL Query API, so they do not normally\n" + "get written back to the triple store. Consider using an INSERT, which is\n" + "part of the SPARQL Update API, in the future.");
}
if (!(isConstructQuery || isInsertQuery)) {
throw new ArgumentsException("Only CONSTRUCT queries and INSERT updates may be inserted back to the triple store.");
}
}
final StreamsQuery query = addQuery.addQuery(params.query, isActive, isInsert);
System.out.println("Added query: " + query.getSparql());
} catch (final RyaStreamsException e) {
throw new ExecutionException("Unable to add the query to Rya Streams.", e);
}
} catch (final MalformedQueryException e) {
throw new ArgumentsException("Could not parse the provided query.", e);
}
}
use of org.apache.rya.streams.api.interactor.AddQuery in project incubator-rya by apache.
the class DefaultAddQueryTest method addQuery_validSparql.
@Test
public void addQuery_validSparql() throws Exception {
// Valid SPARQL.
final String sparql = "SELECT * WHERE { ?person <urn:worksAt> ?business }";
// Setup the interactor.
final QueryRepository repo = mock(QueryRepository.class);
final AddQuery addQuery = new DefaultAddQuery(repo);
// Add the query.
addQuery.addQuery(sparql, true, true);
// Verify the call was forwarded to the repository.
verify(repo, times(1)).add(eq(sparql), eq(true), eq(true));
}
use of org.apache.rya.streams.api.interactor.AddQuery in project incubator-rya by apache.
the class DefaultAddQueryTest method addQuery_invalidSparql.
@Test(expected = RyaStreamsException.class)
public void addQuery_invalidSparql() throws Exception {
// Inalid SPARQL.
final String sparql = "This is not sparql.";
// Setup the interactor.
final QueryRepository repo = mock(QueryRepository.class);
final AddQuery addQuery = new DefaultAddQuery(repo);
// Add the query.
addQuery.addQuery(sparql, true, true);
}
Aggregations