use of edu.uci.ics.textdb.api.dataflow.ISink in project textdb by TextDB.
the class Engine method evaluate.
public void evaluate(Plan plan) throws TextDBException {
ISink root = plan.getRoot();
root.open();
root.processTuples();
root.close();
}
use of edu.uci.ics.textdb.api.dataflow.ISink in project textdb by TextDB.
the class NewQueryPlanResource method executeQueryPlan.
/**
* This is the edu.uci.ics.textdb.web.request handler for the execution of a Query Plan.
* @param logicalPlanJson, the json representation of the logical plan
* @return - Generic TextdbWebResponse object
*/
@POST
@Path("/execute")
public // TODO: investigate how to use LogicalPlan directly
TextdbWebResponse executeQueryPlan(String logicalPlanJson) {
try {
LogicalPlan logicalPlan = new ObjectMapper().readValue(logicalPlanJson, LogicalPlan.class);
Plan plan = logicalPlan.buildQueryPlan();
ISink sink = plan.getRoot();
// send response back to frontend
if (sink instanceof TupleSink) {
TupleSink tupleSink = (TupleSink) sink;
tupleSink.open();
List<Tuple> results = tupleSink.collectAllTuples();
tupleSink.close();
ArrayNode arrayNode = new ObjectMapper().createArrayNode();
for (Tuple tuple : results) {
arrayNode.add(tuple.getReadableJson());
}
return new TextdbWebResponse(0, new ObjectMapper().writeValueAsString(arrayNode));
} else {
// execute the plan and return success message
Engine.getEngine().evaluate(plan);
ObjectNode objectNode = new ObjectMapper().createObjectNode();
objectNode.put("status", "plan sucessfully executed");
return new TextdbWebResponse(0, new ObjectMapper().writeValueAsString(objectNode));
}
} catch (IOException | RuntimeException e) {
// TODO remove RuntimeException after the exception refactor
e.printStackTrace();
throw new TextdbWebException(e.getMessage());
}
}
use of edu.uci.ics.textdb.api.dataflow.ISink in project textdb by TextDB.
the class LogicalPlanTest method testLogicalPlan2.
/*
* Test a valid operator graph.
* -> RegexMatcher -->
* KeywordSource --< >-- Join --> TupleSink
* -> NlpEntityOperator -->
*
*/
@Test
public void testLogicalPlan2() throws Exception {
LogicalPlan logicalPlan = getLogicalPlan2();
Plan queryPlan = logicalPlan.buildQueryPlan();
ISink tupleSink = queryPlan.getRoot();
Assert.assertTrue(tupleSink instanceof TupleSink);
IOperator join = ((TupleSink) tupleSink).getInputOperator();
Assert.assertTrue(join instanceof Join);
IOperator joinInput1 = ((Join) join).getInnerInputOperator();
Assert.assertTrue(joinInput1 instanceof RegexMatcher);
IOperator joinInput2 = ((Join) join).getOuterInputOperator();
Assert.assertTrue(joinInput2 instanceof NlpEntityOperator);
IOperator connectorOut1 = ((RegexMatcher) joinInput1).getInputOperator();
Assert.assertTrue(connectorOut1 instanceof ConnectorOutputOperator);
IOperator connectorOut2 = ((NlpEntityOperator) joinInput2).getInputOperator();
Assert.assertTrue(connectorOut2 instanceof ConnectorOutputOperator);
HashSet<Integer> connectorIndices = new HashSet<>();
connectorIndices.add(((ConnectorOutputOperator) connectorOut1).getOutputIndex());
connectorIndices.add(((ConnectorOutputOperator) connectorOut2).getOutputIndex());
Assert.assertEquals(connectorIndices.size(), 2);
OneToNBroadcastConnector connector1 = ((ConnectorOutputOperator) connectorOut1).getOwnerConnector();
OneToNBroadcastConnector connector2 = ((ConnectorOutputOperator) connectorOut2).getOwnerConnector();
Assert.assertSame(connector1, connector2);
IOperator keywordSource = connector1.getInputOperator();
Assert.assertTrue(keywordSource instanceof KeywordMatcherSourceOperator);
}
Aggregations