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

Example 71 with QueryCallback

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

the class SequenceTestCase method testQuery12.

@Test
public void testQuery12() throws InterruptedException {
    log.info("testSequence12 - OUT 1");
    SiddhiManager siddhiManager = new SiddhiManager();
    String streams = "" + "define stream StockStream (symbol string, price float, volume int); " + "define stream TwitterStream (symbol string, count int); ";
    String query = "" + "@info(name = 'query1') " + "from every e1=StockStream[ price >= 50 and volume > 100 ], e2=TwitterStream[count > 10] " + "select e1.price as price, e1.symbol as symbol, e2.count as count " + "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[] { 76.6f, "IBM", 20 }, event.getData());
                            break;
                        default:
                            AssertJUnit.assertSame(1, inEventCount);
                    }
                }
                eventArrived = true;
            }
            if (removeEvents != null) {
                removeEventCount = removeEventCount + removeEvents.length;
            }
            eventArrived = true;
        }
    });
    InputHandler stockStream = siddhiAppRuntime.getInputHandler("StockStream");
    InputHandler twitterStream = siddhiAppRuntime.getInputHandler("TwitterStream");
    siddhiAppRuntime.start();
    stockStream.send(new Object[] { "IBM", 75.6f, 105 });
    stockStream.send(new Object[] { "GOOG", 51f, 101 });
    stockStream.send(new Object[] { "IBM", 76.6f, 111 });
    Thread.sleep(100);
    twitterStream.send(new Object[] { "IBM", 20 });
    stockStream.send(new Object[] { "WSO2", 45.6f, 100 });
    Thread.sleep(100);
    twitterStream.send(new Object[] { "GOOG", 20 });
    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 72 with QueryCallback

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

the class SequenceTestCase method testQuery16.

@Test
public void testQuery16() throws InterruptedException {
    log.info("testSequence16 - OUT 2");
    SiddhiManager siddhiManager = new SiddhiManager();
    String streams = "" + "define stream StockStream1 (symbol string, price float, volume int); " + "define stream StockStream2 (symbol string, price float, volume int); ";
    String query = "" + "@info(name = 'query1') " + "from every e1=StockStream1, e2=StockStream2[e1.symbol != 'AMBA']*, e3=StockStream2[volume <= 70] " + "select e3.symbol as symbol1, e2[0].symbol as symbol2, e3.volume as volume " + "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[] { "WSO2", "GOOG", 65 }, event.getData());
                            break;
                        case 2:
                            AssertJUnit.assertArrayEquals(new Object[] { "DOX", null, 25 }, event.getData());
                            break;
                        default:
                            AssertJUnit.assertSame(2, inEventCount);
                    }
                }
                eventArrived = true;
            }
            if (removeEvents != null) {
                removeEventCount = removeEventCount + removeEvents.length;
            }
            eventArrived = true;
        }
    });
    InputHandler stockStream1 = siddhiAppRuntime.getInputHandler("StockStream1");
    InputHandler stockStream2 = siddhiAppRuntime.getInputHandler("StockStream2");
    siddhiAppRuntime.start();
    stockStream1.send(new Object[] { "IBM", 75.6f, 105 });
    Thread.sleep(100);
    stockStream2.send(new Object[] { "GOOG", 21f, 81 });
    stockStream2.send(new Object[] { "WSO2", 176.6f, 65 });
    stockStream1.send(new Object[] { "BIRT", 21f, 81 });
    stockStream1.send(new Object[] { "AMBA", 126.6f, 165 });
    stockStream2.send(new Object[] { "DDD", 23f, 181 });
    stockStream2.send(new Object[] { "BIRT", 21f, 86 });
    stockStream2.send(new Object[] { "BIRT", 21f, 82 });
    stockStream2.send(new Object[] { "WSO2", 176.6f, 60 });
    stockStream1.send(new Object[] { "AMBA", 126.6f, 165 });
    stockStream2.send(new Object[] { "DOX", 16.2f, 25 });
    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();
}

Example 73 with QueryCallback

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

the class SequenceTestCase method testQuery24.

@Test
public void testQuery24() throws InterruptedException {
    log.info("testSequence24 - 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 and (not (e2[last-1].price is null)) and price>=e2[last-1]" + ".price+5)  or ((e2[last-1].price is null) and price>=e1.price+5 )]+, " + "   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[] { 43.6f, 57.7f, null, 57.7f, 58.7f, 45.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", 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", 41.5f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 42.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "WSO2", 43.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 57.7f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 58.7f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "IBM", 45.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 74 with QueryCallback

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

the class SequenceTestCase method testQuery30.

@Test
public void testQuery30() throws InterruptedException {
    log.info("testSequence30 - OUT 2");
    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=Stream2[price>e1.price] " + "select e1.symbol as symbol1, e2.symbol as symbol2 " + "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;
                eventArrived = true;
                switch(inEventCount) {
                    case 1:
                        AssertJUnit.assertArrayEquals(new Object[] { "WSO2", "IBM" }, inEvents[0].getData());
                        break;
                    case 2:
                        AssertJUnit.assertArrayEquals(new Object[] { "MICROSOFT", "GOOGLE" }, inEvents[0].getData());
                        break;
                }
            }
            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", 55.6f, 100 });
    Thread.sleep(100);
    stream2.send(new Object[] { "IBM", 55.7f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "ORACLE", 55.6f, 100 });
    Thread.sleep(100);
    stream1.send(new Object[] { "MICROSOFT", 55.8f, 100 });
    Thread.sleep(100);
    stream2.send(new Object[] { "GOOGLE", 55.9f, 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();
}

Example 75 with QueryCallback

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

the class SequenceTestCase method testQuery7.

@Test
public void testQuery7() throws InterruptedException {
    log.info("testSequence7 - OUT 2");
    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=Stream2[price>e1.price] or e3=Stream2[symbol=='IBM'] " + "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) {
                for (Event event : inEvents) {
                    inEventCount++;
                    switch(inEventCount) {
                        case 1:
                            AssertJUnit.assertArrayEquals(new Object[] { 55.6f, 55.7f, null }, event.getData());
                            break;
                        case 2:
                            AssertJUnit.assertArrayEquals(new Object[] { 55.7f, 57.6f, null }, 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();
    stream2.send(new Object[] { "WSO2", 59.6f, 100 });
    Thread.sleep(100);
    stream2.send(new Object[] { "WSO2", 55.6f, 100 });
    Thread.sleep(100);
    stream2.send(new Object[] { "IBM", 55.7f, 100 });
    Thread.sleep(100);
    stream2.send(new Object[] { "WSO2", 57.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();
}