Search in sources :

Example 1 with HavingCompareValue

use of org.apache.apex.malhar.contrib.misc.streamquery.condition.HavingCompareValue in project apex-malhar by apache.

the class HavingOperatorTest method testSqlGroupBy.

@SuppressWarnings({ "rawtypes", "unchecked" })
@Test
public void testSqlGroupBy() throws Exception {
    // create operator
    GroupByHavingOperator oper = new GroupByHavingOperator();
    oper.addColumnGroupByIndex(new ColumnIndex("b", null));
    FunctionIndex sum = new SumFunction("c", null);
    oper.addAggregateIndex(sum);
    // create having condition
    HavingCondition having = new HavingCompareValue<Double>(sum, 6.0, 0);
    oper.addHavingCondition(having);
    EqualValueCondition condition = new EqualValueCondition();
    condition.addEqualValue("a", 1);
    oper.setCondition(condition);
    CollectorTestSink sink = new CollectorTestSink();
    oper.outport.setSink(sink);
    oper.setup(null);
    oper.beginWindow(1);
    HashMap<String, Object> tuple = new HashMap<String, Object>();
    tuple.put("a", 1);
    tuple.put("b", 1);
    tuple.put("c", 2);
    oper.inport.process(tuple);
    tuple = new HashMap<String, Object>();
    tuple.put("a", 1);
    tuple.put("b", 1);
    tuple.put("c", 4);
    oper.inport.process(tuple);
    tuple = new HashMap<String, Object>();
    tuple.put("a", 1);
    tuple.put("b", 2);
    tuple.put("c", 6);
    oper.inport.process(tuple);
    tuple = new HashMap<String, Object>();
    tuple.put("a", 1);
    tuple.put("b", 2);
    tuple.put("c", 7);
    oper.inport.process(tuple);
    oper.endWindow();
    oper.teardown();
    LOG.debug("{}", sink.collectedTuples);
}
Also used : SumFunction(org.apache.apex.malhar.contrib.misc.streamquery.function.SumFunction) ColumnIndex(org.apache.apex.malhar.lib.streamquery.index.ColumnIndex) EqualValueCondition(org.apache.apex.malhar.contrib.misc.streamquery.condition.EqualValueCondition) FunctionIndex(org.apache.apex.malhar.contrib.misc.streamquery.function.FunctionIndex) HashMap(java.util.HashMap) HavingCondition(org.apache.apex.malhar.contrib.misc.streamquery.condition.HavingCondition) HavingCompareValue(org.apache.apex.malhar.contrib.misc.streamquery.condition.HavingCompareValue) CollectorTestSink(org.apache.apex.malhar.lib.testbench.CollectorTestSink) Test(org.junit.Test)

Aggregations

HashMap (java.util.HashMap)1 EqualValueCondition (org.apache.apex.malhar.contrib.misc.streamquery.condition.EqualValueCondition)1 HavingCompareValue (org.apache.apex.malhar.contrib.misc.streamquery.condition.HavingCompareValue)1 HavingCondition (org.apache.apex.malhar.contrib.misc.streamquery.condition.HavingCondition)1 FunctionIndex (org.apache.apex.malhar.contrib.misc.streamquery.function.FunctionIndex)1 SumFunction (org.apache.apex.malhar.contrib.misc.streamquery.function.SumFunction)1 ColumnIndex (org.apache.apex.malhar.lib.streamquery.index.ColumnIndex)1 CollectorTestSink (org.apache.apex.malhar.lib.testbench.CollectorTestSink)1 Test (org.junit.Test)1