Examples with QueryCallback org.ballerinalang.siddhi.core.query.output.callback.QueryCallback used on opensource projects

Example 76 with QueryCallback

use of org.ballerinalang.siddhi.core.query.output.callback.QueryCallback in project ballerina by ballerina-lang.

the class SequenceTestCase method testQuery25.

@Test
public void testQuery25() throws InterruptedException {
    log.info("testQuery25 - OUT 1");
    SiddhiManager siddhiManager = new SiddhiManager();
    String streams = "" + "define stream Stream1 (symbol string, price float, volume int); " + "define stream Stream2 (symbol string, price float, volume int); " + "define stream Stream3 (symbol string, price float, volume int); ";
    String query = "" + "@info(name = 'query1') " + "from e1=Stream1[price >20], e2=Stream2['IBM' == symbol] and " + "e3=Stream3['WSO2' == symbol]" + "select e1.price as price1, e2.price as price2, e3.price as price3 " + "insert into OutputStream ;";
    SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
    siddhiAppRuntime.addCallback("query1", new QueryCallback() {

        @Override
        public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
            EventPrinter.print(timeStamp, inEvents, removeEvents);
            if (inEvents != null) {
                inEventCount = inEventCount + inEvents.length;
                switch(inEventCount) {
                    case 1:
                        AssertJUnit.assertArrayEquals(new Object[] { 25.5f, 45.5f, 46.56f }, inEvents[0].getData());
                        break;
                    default:
                        AssertJUnit.assertEquals("Number of success events", 1, inEventCount);
                }
                eventArrived = true;
            }
            if (removeEvents != null) {
                removeEventCount = removeEventCount + removeEvents.length;
            }
            eventArrived = true;
        }
    });
    InputHandler stream1 = siddhiAppRuntime.getInputHandler("Stream1");
    InputHandler stream2 = siddhiAppRuntime.getInputHandler("Stream2");
    InputHandler stream3 = siddhiAppRuntime.getInputHandler("Stream3");
    siddhiAppRuntime.start();
    stream1.send(new Object[] { "IBM", 25.5f, 100 });
    Thread.sleep(100);
    stream2.send(new Object[] { "IBM", 45.5f, 100 });
    Thread.sleep(100);
    stream3.send(new Object[] { "WSO2", 46.56f, 100 });
    Thread.sleep(100);
    AssertJUnit.assertEquals("Number of success events", 1, inEventCount);
    AssertJUnit.assertEquals("Number of remove events", 0, removeEventCount);
    AssertJUnit.assertEquals("Event arrived", true, eventArrived);
    siddhiAppRuntime.shutdown();
}

Example 77 with QueryCallback

use of org.ballerinalang.siddhi.core.query.output.callback.QueryCallback in project ballerina by ballerina-lang.

the class SequenceTestCase method testQuery21.

@Test
public void testQuery21() throws InterruptedException {
    log.info("testSequence21 - OUT 1");
    SiddhiManager siddhiManager = new SiddhiManager();
    String streams = "" + "define stream Stream1 (symbol string, price float, volume int); " + "define stream Stream2 (symbol string, price float, volume int); ";
    String query = "" + "@info(name = 'query1') " + "from every e1=Stream1[price>20], " + "   e2=Stream1[((e2[last].price is null) and price>=e1.price) or ((not (e2[last].price is null)) and " + "price>=e2[last].price)]+, " + "   e3=Stream1[price<e2[last].price] " + "select e1.price as price1, e2[0].price as price2, e2[last-2].price as price3, e2[last-1].price as " + "price4, e2[last].price as price5, e3.price as price6, e2[last-20].price as price7 " + "insert into OutputStream ;";
    SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
    siddhiAppRuntime.addCallback("query1", new QueryCallback() {

        @Override
        public void receive(long timestamp, Event[] inEvents, Event[] removeEvents) {
            EventPrinter.print(timestamp, inEvents, removeEvents);
            if (inEvents != null) {
                for (Event event : inEvents) {
                    inEventCount++;
                    switch(inEventCount) {
                        case 1:
                            AssertJUnit.assertArrayEquals(new Object[] { 25.0f, 35.6f, 45.5f, 57.6f, 58.6f, 47.6f, null }, event.getData());
                            break;
                        default:
                            AssertJUnit.assertSame(1, inEventCount);
                    }
                }
                eventArrived = true;
            }
            if (removeEvents != null) {
                removeEventCount = removeEventCount + removeEvents.length;
            }
            eventArrived = true;
        }
    });
    InputHandler stream1 = siddhiAppRuntime.getInputHandler("Stream1");
    InputHandler stream2 = siddhiAppRuntime.getInputHandler("Stream2");
    siddhiAppRuntime.start();
    stream1.send(new Object[] { "WSO2", 29.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 25.0f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 35.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 45.5f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 57.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 58.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 47.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 45.6f, 100 });
    Thread.sleep(10000);
    AssertJUnit.assertEquals("Number of success events", 1, inEventCount);
    AssertJUnit.assertEquals("Number of remove events", 0, removeEventCount);
    AssertJUnit.assertEquals("Event arrived", true, eventArrived);
    siddhiAppRuntime.shutdown();
}

Example 78 with QueryCallback

use of org.ballerinalang.siddhi.core.query.output.callback.QueryCallback in project ballerina by ballerina-lang.

the class SequenceTestCase method testQuery10.

@Test
public void testQuery10() throws InterruptedException {
    log.info("testSequence10 - OUT 1");
    SiddhiManager siddhiManager = new SiddhiManager();
    String streams = "" + "define stream Stream1 (symbol string, price float, volume int); " + "define stream Stream2 (symbol string, price float, volume int); ";
    String query = "" + "@info(name = 'query1') " + "from every e1=Stream2[price>20]+, e2=Stream1[price>e1[0].price] " + "select e1[0].price as price1, e1[1].price as price2, e2.price as price3 " + "insert into OutputStream ;";
    SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
    siddhiAppRuntime.addCallback("query1", new QueryCallback() {

        @Override
        public void receive(long timestamp, Event[] inEvents, Event[] removeEvents) {
            EventPrinter.print(timestamp, inEvents, removeEvents);
            if (inEvents != null) {
                for (Event event : inEvents) {
                    inEventCount++;
                    switch(inEventCount) {
                        case 1:
                            AssertJUnit.assertArrayEquals(new Object[] { 55.6f, null, 57.6f }, event.getData());
                            break;
                        default:
                            AssertJUnit.assertSame(1, inEventCount);
                    }
                }
                eventArrived = true;
            }
            if (removeEvents != null) {
                removeEventCount = removeEventCount + removeEvents.length;
            }
            eventArrived = true;
        }
    });
    InputHandler stream1 = siddhiAppRuntime.getInputHandler("Stream1");
    InputHandler stream2 = siddhiAppRuntime.getInputHandler("Stream2");
    siddhiAppRuntime.start();
    stream1.send(new Object[] { "WSO2", 59.6f, 100 });
    Thread.sleep(100);
    stream2.send(new Object[] { "WSO2", 55.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 57.6f, 100 });
    Thread.sleep(100);
    AssertJUnit.assertEquals("Number of success events", 1, inEventCount);
    AssertJUnit.assertEquals("Number of remove events", 0, removeEventCount);
    AssertJUnit.assertEquals("Event arrived", true, eventArrived);
    siddhiAppRuntime.shutdown();
}

Example 79 with QueryCallback

use of org.ballerinalang.siddhi.core.query.output.callback.QueryCallback in project ballerina by ballerina-lang.

the class SequenceTestCase method testQuery22.

@Test
public void testQuery22() throws InterruptedException {
    log.info("testSequence22 - OUT 1");
    SiddhiManager siddhiManager = new SiddhiManager();
    String streams = "" + "define stream Stream1 (symbol string, price float, volume int); " + "define stream Stream2 (symbol string, price float, volume int); ";
    String query = "" + "@info(name = 'query1') " + "from every e1=Stream1[price>20], " + "   e2=Stream1[((e2[last].price is null) and price>=e1.price) or ((not (e2[last].price is null)) and " + "price>=e2[last].price)]+, " + "   e3=Stream1[price<e2[last].price and price>e2[last-1].price] " + "select e1.price as price1, e2[0].price as price2, e2[last-2].price as price3, e2[last-1].price as " + "price4, e2[last].price as price5, e3.price as price6, e2[last-20].price as price7 " + "insert into OutputStream ;";
    SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
    siddhiAppRuntime.addCallback("query1", new QueryCallback() {

        @Override
        public void receive(long timestamp, Event[] inEvents, Event[] removeEvents) {
            EventPrinter.print(timestamp, inEvents, removeEvents);
            if (inEvents != null) {
                for (Event event : inEvents) {
                    inEventCount++;
                    switch(inEventCount) {
                        case 1:
                            AssertJUnit.assertArrayEquals(new Object[] { 25.0f, 35.6f, 45.5f, 57.6f, 58.6f, 57.7f, null }, event.getData());
                            break;
                        default:
                            AssertJUnit.assertSame(1, inEventCount);
                    }
                }
                eventArrived = true;
            }
            if (removeEvents != null) {
                removeEventCount = removeEventCount + removeEvents.length;
            }
            eventArrived = true;
        }
    });
    InputHandler stream1 = siddhiAppRuntime.getInputHandler("Stream1");
    InputHandler stream2 = siddhiAppRuntime.getInputHandler("Stream2");
    siddhiAppRuntime.start();
    stream1.send(new Object[] { "WSO2", 29.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 25.0f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 35.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 45.5f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 57.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 58.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 57.7f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 45.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 60.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 61.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 59.7f, 100 });
    Thread.sleep(100);
    AssertJUnit.assertEquals("Number of success events", 1, inEventCount);
    AssertJUnit.assertEquals("Number of remove events", 0, removeEventCount);
    AssertJUnit.assertEquals("Event arrived", true, eventArrived);
    siddhiAppRuntime.shutdown();
}

Example 80 with QueryCallback

use of org.ballerinalang.siddhi.core.query.output.callback.QueryCallback in project ballerina by ballerina-lang.

the class SequenceTestCase method testQuery23.

@Test
public void testQuery23() throws InterruptedException {
    log.info("testSequence23 - OUT 1");
    SiddhiManager siddhiManager = new SiddhiManager();
    String streams = "" + "define stream Stream1 (symbol string, price float, volume int); " + "define stream Stream2 (symbol string, price float, volume int); ";
    String query = "" + "@info(name = 'query1') " + "from every e1=Stream1[price>20], " + "   e2=Stream1[price>=e2[last].price or price>=e1.price ]+, " + "   e3=Stream1[price<e2[last].price]" + "select e1.price as price1, e2[0].price as price2, e2[last-2].price as price3, e2[last-1].price as " + "price4, e2[last].price as price5, e3.price as price6 " + "insert into OutputStream ;";
    SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
    siddhiAppRuntime.addCallback("query1", new QueryCallback() {

        @Override
        public void receive(long timestamp, Event[] inEvents, Event[] removeEvents) {
            EventPrinter.print(timestamp, inEvents, removeEvents);
            if (inEvents != null) {
                for (Event event : inEvents) {
                    inEventCount++;
                    switch(inEventCount) {
                        case 1:
                            AssertJUnit.assertArrayEquals(new Object[] { 25.0f, 35.6f, null, null, 35.6f, 29.5f }, event.getData());
                            break;
                        case 2:
                            AssertJUnit.assertArrayEquals(new Object[] { 29.5f, 57.6f, null, 57.6f, 58.6f, 57.7f }, event.getData());
                            break;
                        default:
                            AssertJUnit.assertSame(2, inEventCount);
                    }
                }
                eventArrived = true;
            }
            if (removeEvents != null) {
                removeEventCount = removeEventCount + removeEvents.length;
            }
            eventArrived = true;
        }
    });
    InputHandler stream1 = siddhiAppRuntime.getInputHandler("Stream1");
    InputHandler stream2 = siddhiAppRuntime.getInputHandler("Stream2");
    siddhiAppRuntime.start();
    stream1.send(new Object[] { "WSO2", 29.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 25.0f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 35.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 29.5f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 57.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 58.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 57.7f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 45.6f, 100 });
    Thread.sleep(100);
    AssertJUnit.assertEquals("Number of success events", 2, inEventCount);
    AssertJUnit.assertEquals("Number of remove events", 0, removeEventCount);
    AssertJUnit.assertEquals("Event arrived", true, eventArrived);
    siddhiAppRuntime.shutdown();
}