use of com.google.zetasql.resolvedast.ResolvedNode in project beam by apache.
the class TVFScanConverter method getInputs.
@Override
public List<ResolvedNode> getInputs(ResolvedTVFScan zetaNode) {
List<ResolvedNode> inputs = new ArrayList();
if (zetaNode.getTvf() != null && context.getUserDefinedTableValuedFunctions().containsKey(zetaNode.getTvf().getNamePath())) {
inputs.add(context.getUserDefinedTableValuedFunctions().get(zetaNode.getTvf().getNamePath()));
}
for (ResolvedFunctionArgument argument : zetaNode.getArgumentList()) {
if (argument.getScan() != null) {
inputs.add(argument.getScan());
}
}
// Extract ResolvedArguments for solving ResolvedArgumentRef in later conversion.
if (zetaNode.getTvf() instanceof FixedOutputSchemaTVF) {
FileDescriptorSetsBuilder temp = new FileDescriptorSetsBuilder();
// TODO: migrate to public Java API to retrieve FunctionSignature.
TableValuedFunctionProto tableValuedFunctionProto = zetaNode.getTvf().serialize(temp);
for (int i = 0; i < tableValuedFunctionProto.getSignature().getArgumentList().size(); i++) {
String argumentName = tableValuedFunctionProto.getSignature().getArgument(i).getOptions().getArgumentName();
if (zetaNode.getArgumentList().get(i).nodeKind() == ResolvedNodeKind.RESOLVED_FUNCTION_ARGUMENT) {
ResolvedFunctionArgument resolvedTVFArgument = zetaNode.getArgumentList().get(i);
if (resolvedTVFArgument.getExpr().nodeKind() == ResolvedNodeKind.RESOLVED_LITERAL) {
ResolvedLiteral literal = (ResolvedLiteral) resolvedTVFArgument.getExpr();
context.addToFunctionArgumentRefMapping(argumentName, getExpressionConverter().convertResolvedLiteral(literal));
}
}
}
}
return inputs;
}
Aggregations