Ensure function types and names are registered in memory

This commit is contained in:
Dave Syer
2019-02-21 17:39:24 +00:00
parent aaf268ea40
commit 155d76b1a5
3 changed files with 31 additions and 2 deletions

View File

@@ -281,7 +281,7 @@ public abstract class AbstractComposableFunctionRegistry implements FunctionRegi
this.types.computeIfAbsent(name, str -> functionType);
}
private void addName(Object function, String name) {
protected void addName(Object function, String name) {
this.names.put(function, name);
}

View File

@@ -82,6 +82,7 @@ public class InMemoryFunctionCatalog extends AbstractComposableFunctionRegistry
for (String name : functionRegistration.getNames()) {
addType(name, functionRegistration.getType());
addName(functionRegistration.getTarget(), name);
if (functionRegistration.getTarget() instanceof Function) {
this.addFunction(name, (Function<?, ?>) functionRegistration.getTarget());
}