private String buildFunctionOptions(FunctionMethod function) {
StringBuilder options = new StringBuilder();
addCommonOptions(options, function);
if (function.getCategory() != null) {
addOption(options, CATEGORY, function.getCategory());
}
if (!function.getDeterminism().equals(Determinism.DETERMINISTIC)) {
addOption(options, DETERMINISM, function.getDeterminism().name());
}
if (function.getInvocationClass() != null) {
addOption(options, JAVA_CLASS, function.getInvocationClass());
}
if (function.getInvocationMethod() != null) {
addOption(options, JAVA_METHOD, function.getInvocationMethod());
}
if (!function.getProperties().isEmpty()) {
for (String key : function.getProperties().keySet()) {
addOption(options, key, function.getProperty(key, false));
}
}
return options.toString();
}
public Object newInstance() {
try {
return invocationMethod.getDeclaringClass().newInstance();
} catch (InstantiationException e) {
throw new TeiidRuntimeException(
QueryPlugin.Event.TEIID30602,
QueryPlugin.Util.gs(
QueryPlugin.Event.TEIID30602, method.getName(), method.getInvocationClass()));
} catch (IllegalAccessException e) {
throw new TeiidRuntimeException(
QueryPlugin.Event.TEIID30602,
QueryPlugin.Util.gs(
QueryPlugin.Event.TEIID30602, method.getName(), method.getInvocationClass()));
}
}
@Test
public void testUDF() throws Exception {
String ddl =
"CREATE VIRTUAL FUNCTION SourceFunc(flag boolean, msg varchar) RETURNS varchar "
+ "OPTIONS(CATEGORY 'misc', DETERMINISM 'DETERMINISTIC', "
+ "\"NULL-ON-NULL\" 'true', JAVA_CLASS 'foo', JAVA_METHOD 'bar', RANDOM 'any', UUID 'x')";
Schema s = helpParse(ddl, "model").getSchema();
FunctionMethod fm = s.getFunction("x");
assertNotNull(fm);
assertEquals("string", fm.getOutputParameter().getType());
assertEquals(FunctionMethod.PushDown.CAN_PUSHDOWN, fm.getPushdown());
assertEquals(2, fm.getInputParameterCount());
assertEquals("flag", fm.getInputParameters().get(0).getName());
assertEquals("boolean", fm.getInputParameters().get(0).getType());
assertEquals("msg", fm.getInputParameters().get(1).getName());
assertEquals("string", fm.getInputParameters().get(1).getType());
assertFalse(fm.getInputParameters().get(1).isVarArg());
assertEquals(FunctionMethod.Determinism.DETERMINISTIC, fm.getDeterminism());
assertEquals("misc", fm.getCategory());
assertEquals(true, fm.isNullOnNull());
assertEquals("foo", fm.getInvocationClass());
assertEquals("bar", fm.getInvocationMethod());
assertEquals("any", fm.getProperties().get("RANDOM"));
}