@Test(expected = DuplicateDefinitionException.class)
public void testDefineFunctionsWithSameFunctionID() throws InterruptedException {
log.info("testDefineFunctionsWithSameFunctionID");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
String concatFunc1 =
"define function concat[Scala] return string {\n"
+ " var concatenatedString = \"\"\n"
+ " for(i <- 0 until data.length) {\n"
+ " concatenatedString += data(i).toString\n"
+ " }\n"
+ " concatenatedString\n"
+ "};";
String concatFunc2 =
"define function concat[JavaScript] return string {\n"
+ " var str1 = data[0];\n"
+ " var str2 = data[1];\n"
+ " var str3 = data[2];\n"
+ " var res = str1.concat(str2,str3);\n"
+ " return res;\n"
+ "};\n";
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(concatFunc1 + concatFunc2);
executionPlanRuntime.shutdown();
}
@Test(expected = ExecutionPlanCreationException.class)
public void testJavaScriptCompilationFailure() throws InterruptedException {
log.info("testJavaScriptCompilationFailure");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
String concatFunc =
"define function concatJ[JavaScript] return string {\n"
+ " var str1 = data[0;\n"
+ " var str2 = data[1];\n"
+ " var str3 = data[2];\n"
+ " var res = str1.concat(str2,str3);\n"
+ " return res;\n"
+ "};";
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(concatFunc);
executionPlanRuntime.shutdown();
}
@Test
public void testEvalJavaScriptConcat() throws InterruptedException {
log.info("testEvalJavaScriptConcat");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
String concatFunc =
"define function concatJ[JavaScript] return string {\n"
+ " var str1 = data[0];\n"
+ " var str2 = data[1];\n"
+ " var str3 = data[2];\n"
+ " var res = str1.concat(str2,str3);\n"
+ " return res;\n"
+ "};";
String cseEventStream =
"@config(async = 'true')define stream cseEventStream (symbol string, price float, volume long);";
String query =
("@info(name = 'query1') from cseEventStream select price , concatJ(symbol,' ',price) as concatStr "
+ "group by volume insert into mailOutput;");
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(concatFunc + cseEventStream + query);
executionPlanRuntime.addCallback(
"query1",
new QueryCallback() {
@Override
public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
EventPrinter.print(timeStamp, inEvents, removeEvents);
isReceived[0] = true;
value[0] = inEvents[inEvents.length - 1].getData(1);
}
});
isReceived[0] = false;
value[0] = null;
InputHandler inputHandler = executionPlanRuntime.getInputHandler("cseEventStream");
executionPlanRuntime.start();
inputHandler.send(new Object[] {"WSO2", 50f, 60f, 60l, 6});
Thread.sleep(100);
if (isReceived[0]) {
Assert.assertEquals("WSO2 50", value[0]);
} else {
throw new RuntimeException("The event has not been received");
}
executionPlanRuntime.shutdown();
}
@Test
public void testReturnType() throws InterruptedException {
log.info("testReturnType");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
String toFloatSFunc =
"define function toFloatS[Scala] return float {\n"
+ " data(0).asInstanceOf[String].toFloat\n"
+ "};\n";
String cseEventStream =
"@config(async = 'true')define stream cseEventStream (symbol string, price string, volume long);\n";
String query1 =
("@info(name = 'query1') from cseEventStream select price , toFloatS(price) as priceF insert into mailto1;\n");
String query2 =
("@info(name = 'query2') from mailto1 select priceF/2 as newPrice insert into mailto2;\n");
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(toFloatSFunc + cseEventStream + query1 + query2);
executionPlanRuntime.addCallback(
"query2",
new QueryCallback() {
@Override
public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
EventPrinter.print(timeStamp, inEvents, removeEvents);
isReceived[0] = true;
value[0] = inEvents[inEvents.length - 1].getData(0);
}
});
isReceived[0] = false;
value[0] = null;
InputHandler inputHandler = executionPlanRuntime.getInputHandler("cseEventStream");
executionPlanRuntime.start();
inputHandler.send(new Object[] {"WSO2", "50.0", 60f, 60l, 6});
Thread.sleep(100);
if (isReceived[0]) {
Assert.assertEquals(25.0f, value[0]);
} else {
throw new RuntimeException("The event has not been received");
}
executionPlanRuntime.shutdown();
}
@Test(expected = ExecutionPlanValidationException.class)
public void testUseUndefinedFunction() throws InterruptedException {
log.info("testUseUndefinedFunction");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
// siddhiManager.defineFunction(concatFunc);
String cseEventStream =
"@config(async = 'true')define stream cseEventStream (symbol string, price float, volume long);";
String query =
("@info(name = 'query1') from cseEventStream select price , undefinedFunc(symbol,' ',price) as concatStr "
+ "group by volume insert into mailOutput;");
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(cseEventStream + query);
executionPlanRuntime.addCallback(
"query1",
new QueryCallback() {
@Override
public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
EventPrinter.print(timeStamp, inEvents, removeEvents);
isReceived[0] = true;
value[0] = inEvents[inEvents.length - 1].getData(1);
}
});
isReceived[0] = false;
value[0] = null;
InputHandler inputHandler = executionPlanRuntime.getInputHandler("cseEventStream");
executionPlanRuntime.start();
inputHandler.send(new Object[] {"IBM", 700f, 100l});
Thread.sleep(100);
if (isReceived[0]) {
Assert.assertEquals("IBM 700.0", value[0]);
} else {
throw new RuntimeException("The event has not been received");
}
executionPlanRuntime.shutdown();
}
@Test(expected = ExecutionPlanValidationException.class)
public void testMissingReturnType() {
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
ExecutionPlan.executionPlan("test")
.defineFunction(
(new FunctionDefinition()
.id("concat")
.language("Scala")
.body(
"var concatenatedString = \"\"\n"
+ "for(i <- 0 until data.length){\n"
+ " concatenatedString += data(i).toString\n"
+ "}\n"
+ "concatenatedString")));
}
@Test(expected = SiddhiParserException.class)
public void testMissingReturnTypeWhileParsing() throws InterruptedException {
log.info("testDefineManyFunctionsAndCallThemRandom");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
String concatFunc1 =
"define function concat[Scala] {\n"
+ " var concatenatedString = \"\"\n"
+ " for(i <- 0 until data.length) {\n"
+ " concatenatedString += data(i).toString\n"
+ " }\n"
+ " concatenatedString\n"
+ "};";
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(concatFunc1);
executionPlanRuntime.shutdown();
}
@Test(expected = ExecutionPlanCreationException.class)
public void testScalaCompilationFailure() throws InterruptedException {
log.info("testScalaCompilationFailure");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
String concatFunc =
"define function concat[Scala] return string {\n"
+ " for(i <- 0 until data.length){\n"
+ " concatenatedString += data(i).toString\n"
+ " }\n"
+ " concatenatedString\n"
+ "}";
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(concatFunc);
executionPlanRuntime.shutdown();
}
@Test
public void testDefineManyFunctionsAndCallThemRandom() throws InterruptedException {
log.info("testDefineManyFunctionsAndCallThemRandom");
SiddhiManager siddhiManager = new SiddhiManager();
siddhiManager.setExtension(
"evalscript:javascript", org.wso2.siddhi.extension.evalscript.EvalJavaScript.class);
siddhiManager.setExtension(
"evalscript:scala", org.wso2.siddhi.extension.evalscript.EvalScala.class);
String concatSFunc =
"define function concatS[Scala] return string {\n"
+ " var concatenatedString = \"\"\n"
+ " for(i <- 0 until data.length){\n"
+ " concatenatedString += data(i).toString\n"
+ " }\n"
+ " concatenatedString\n"
+ "};\n";
String concatJFunc =
"define function concatJ[JavaScript] return string {\n"
+ " var str1 = data[0].toString();\n"
+ " var str2 = data[1].toString();\n"
+ " var str3 = data[2].toString();\n"
+ " var res = str1.concat(str2,str3);\n"
+ " return res;\n"
+ "};\n";
String toFloatSFunc =
"define function toFloatS[Scala] return float {\n"
+ " data(0).asInstanceOf[Long].toFloat\n"
+ "};\n";
String toStringJFunc =
"define function toStringJ[JavaScript] return string {\n"
+ " return data[0].toString();\n"
+ "};\n";
String cseEventStream =
"@config(async = 'true')define stream cseEventStream (symbol string, price float, volume long);\n";
String query1 =
("@info(name = 'query1') from cseEventStream select price , toStringJ(price) as concatStr insert into mailto1;\n");
String query2 =
("@info(name = 'query2') from cseEventStream select price , toFloatS(volume) as concatStr insert into mailto2;\n");
String query3 =
("@info(name = 'query3') from cseEventStream select price , concatJ(symbol,' ',price) as concatStr insert into mailto3;\n");
String query4 =
("@info(name = 'query4') from cseEventStream select price , concatS(symbol,' ',price) as concatStr insert into mailto4;\n");
ExecutionPlanRuntime executionPlanRuntime =
siddhiManager.createExecutionPlanRuntime(
concatSFunc
+ concatJFunc
+ toFloatSFunc
+ toStringJFunc
+ cseEventStream
+ query1
+ query2
+ query3
+ query4);
executionPlanRuntime.addCallback(
"query1",
new QueryCallback() {
@Override
public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
EventPrinter.print(timeStamp, inEvents, removeEvents);
isReceived[0] = true;
value[0] = inEvents[inEvents.length - 1].getData(1);
}
});
executionPlanRuntime.addCallback(
"query2",
new QueryCallback() {
@Override
public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
EventPrinter.print(timeStamp, inEvents, removeEvents);
isReceived[1] = true;
value[1] = inEvents[inEvents.length - 1].getData(1);
}
});
executionPlanRuntime.addCallback(
"query3",
new QueryCallback() {
@Override
public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
EventPrinter.print(timeStamp, inEvents, removeEvents);
isReceived[2] = true;
value[2] = inEvents[inEvents.length - 1].getData(1);
}
});
executionPlanRuntime.addCallback(
"query4",
new QueryCallback() {
@Override
public void receive(long timeStamp, Event[] inEvents, Event[] removeEvents) {
EventPrinter.print(timeStamp, inEvents, removeEvents);
isReceived[3] = true;
value[3] = inEvents[inEvents.length - 1].getData(1);
}
});
for (int i = 0; i < isReceived.length; i++) {
isReceived[i] = false;
}
for (int i = 0; i < value.length; i++) {
value[i] = null;
}
InputHandler inputHandler = executionPlanRuntime.getInputHandler("cseEventStream");
executionPlanRuntime.start();
inputHandler.send(new Object[] {"WSO2", 50f, 6l});
Thread.sleep(100);
if (isReceived[0]) {
Assert.assertEquals("50", value[0]);
} else {
throw new RuntimeException("The event has not been received");
}
if (isReceived[1]) {
Assert.assertEquals(6f, value[1]);
} else {
throw new RuntimeException("The event has not been received");
}
if (isReceived[2]) {
Assert.assertEquals("WSO2 50", value[2]);
} else {
throw new RuntimeException("The event has not been received");
}
if (isReceived[3]) {
Assert.assertEquals("WSO2 50.0", value[3]);
} else {
throw new RuntimeException("The event has not been received");
}
executionPlanRuntime.shutdown();
}
