use of org.ballerinalang.siddhi.core.SiddhiManager in project ballerina by ballerina-lang.
the class SequenceTestCase method testQuery4.
@Test
public void testQuery4() throws InterruptedException {
log.info("testSequence4 - 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, 55.7f, 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);
stream2.send(new Object[] { "IBM", 55.7f, 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();
}
use of org.ballerinalang.siddhi.core.SiddhiManager in project ballerina by ballerina-lang.
the class SequenceTestCase method testTimeBatchAndSequence.
@Test
public void testTimeBatchAndSequence() throws Exception {
log.info("testTimeBatchAndSequence OUT 1");
SiddhiManager siddhiManager = new SiddhiManager();
String streams = "" + "define stream received_reclamations (timestamp long, product_id string, defect_category string);";
String query = "" + "@info(name = 'query1') " + "from received_reclamations#window.timeBatch(1 sec) " + "select product_id, defect_category, count(product_id) as num " + "group by product_id, defect_category " + "insert into reclamation_averages;" + "" + "@info(name = 'query2') " + "from a=reclamation_averages[num > 1], b=reclamation_averages[num > a.num and product_id == a" + ".product_id and defect_category == a.defect_category] " + "select a.product_id, a.defect_category, a.num as oldNum, b.num as newNum " + "insert into increased_reclamations;";
SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
siddhiAppRuntime.addCallback("increased_reclamations", new StreamCallback() {
@Override
public void receive(Event[] events) {
EventPrinter.print(events);
if (events != null) {
for (Event event : events) {
inEventCount++;
switch(inEventCount) {
case 1:
String product = (String) event.getData()[0];
String defectCategory = (String) event.getData()[1];
long oldNum = (Long) event.getData()[2];
long newNum = (Long) event.getData()[3];
AssertJUnit.assertTrue(product.equals("abc"));
AssertJUnit.assertTrue(defectCategory.equals("123"));
AssertJUnit.assertTrue(oldNum < newNum);
break;
default:
AssertJUnit.assertSame(1, inEventCount);
}
}
}
eventArrived = true;
}
});
InputHandler i1 = siddhiAppRuntime.getInputHandler("received_reclamations");
siddhiAppRuntime.start();
for (int i = 0; i < 5; i++) {
i1.send(new Object[] { System.currentTimeMillis(), "abc", "123" });
Thread.sleep(100);
}
Thread.sleep(500);
for (int i = 0; i < 8; i++) {
i1.send(new Object[] { System.currentTimeMillis(), "abc", "123" });
Thread.sleep(100);
}
Thread.sleep(1000);
siddhiAppRuntime.shutdown();
AssertJUnit.assertEquals("Event arrived", true, eventArrived);
}
use of org.ballerinalang.siddhi.core.SiddhiManager in project ballerina by ballerina-lang.
the class AbsentSequenceTestCase method testQueryAbsent13.
@Test
public void testQueryAbsent13() throws InterruptedException {
log.info("Test the query e1, not e2 for 1 sec, e3 with e1, e2(condition failed) e3");
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>10], not Stream2[price>20] for 1 sec, e3=Stream3[price>30] " + "select e1.symbol as symbol1, e3.symbol as symbol3 " + "insert into OutputStream ;";
SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
TestUtil.TestCallback callback = TestUtil.addQueryCallback(siddhiAppRuntime, "query1", new Object[] { "WSO2", "GOOGLE" });
InputHandler stream1 = siddhiAppRuntime.getInputHandler("Stream1");
InputHandler stream2 = siddhiAppRuntime.getInputHandler("Stream2");
InputHandler stream3 = siddhiAppRuntime.getInputHandler("Stream3");
siddhiAppRuntime.start();
stream1.send(new Object[] { "WSO2", 15.6f, 100 });
Thread.sleep(100);
stream2.send(new Object[] { "IBM", 8.7f, 100 });
Thread.sleep(1100);
stream3.send(new Object[] { "GOOGLE", 55.7f, 100 });
Thread.sleep(100);
callback.throwAssertionErrors();
AssertJUnit.assertEquals("Number of success events", 0, callback.getInEventCount());
AssertJUnit.assertEquals("Number of remove events", 0, callback.getRemoveEventCount());
AssertJUnit.assertFalse("Event arrived", callback.isEventArrived());
siddhiAppRuntime.shutdown();
}
use of org.ballerinalang.siddhi.core.SiddhiManager in project ballerina by ballerina-lang.
the class AbsentSequenceTestCase method testQueryAbsent3.
@Test
public void testQueryAbsent3() throws InterruptedException {
log.info("Test the query e1, not e2 sending e2 for 1 sec");
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 e1=Stream1[price>20], not Stream2[price>e1.price] for 1 sec " + "select e1.symbol as symbol1 " + "insert into OutputStream ;";
SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
TestUtil.TestCallback callback = TestUtil.addQueryCallback(siddhiAppRuntime, "query1");
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", 58.7f, 100 });
Thread.sleep(1000);
callback.throwAssertionErrors();
AssertJUnit.assertEquals("Number of success events", 0, callback.getInEventCount());
AssertJUnit.assertEquals("Number of remove events", 0, callback.getRemoveEventCount());
AssertJUnit.assertFalse("Event arrived", callback.isEventArrived());
siddhiAppRuntime.shutdown();
}
use of org.ballerinalang.siddhi.core.SiddhiManager in project ballerina by ballerina-lang.
the class AbsentSequenceTestCase method testQueryAbsent24.
@Test
public void testQueryAbsent24() throws InterruptedException {
log.info("Test the query not e1 for 1 sec, e2, not e3 for 1 sec, e4 with e2 after 1 sec e4");
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); " + "define stream Stream4 (symbol string, price float, volume int); ";
String query = "" + "@info(name = 'query1') " + "from not Stream1[price>10] for 1 sec, e2=Stream2[price>20], not Stream3[price>30] for 1 " + "sec, e4=Stream4[price>40] " + "select e2.symbol as symbol2, e4.symbol as symbol4 " + "insert into OutputStream ;";
SiddhiAppRuntime siddhiAppRuntime = siddhiManager.createSiddhiAppRuntime(streams + query);
TestUtil.TestCallback callback = TestUtil.addQueryCallback(siddhiAppRuntime, "query1", new Object[] { "IBM", "ORACLE" });
InputHandler stream2 = siddhiAppRuntime.getInputHandler("Stream2");
InputHandler stream4 = siddhiAppRuntime.getInputHandler("Stream4");
siddhiAppRuntime.start();
Thread.sleep(1100);
stream2.send(new Object[] { "IBM", 28.7f, 100 });
Thread.sleep(1100);
stream4.send(new Object[] { "ORACLE", 44.7f, 100 });
Thread.sleep(100);
callback.throwAssertionErrors();
AssertJUnit.assertEquals("Number of success events", 1, callback.getInEventCount());
AssertJUnit.assertEquals("Number of remove events", 0, callback.getRemoveEventCount());
AssertJUnit.assertTrue("Event arrived", callback.isEventArrived());
siddhiAppRuntime.shutdown();
}
Aggregations