001package gudusoft.gsqlparser.nodes.functions;
002
003import java.util.ArrayList;
004
005import gudusoft.gsqlparser.EFunctionType;
006import gudusoft.gsqlparser.ESqlClause;
007import gudusoft.gsqlparser.TCustomSqlStatement;
008
009import gudusoft.gsqlparser.nodes.TObjectName;
010import gudusoft.gsqlparser.nodes.TParseTreeVisitor;
011import gudusoft.gsqlparser.nodes.TTableFunction;
012
013public class TMLPredictFunction  extends TTableFunction{
014    private ArrayList<TPredictArgument> arguments;
015
016    public ArrayList<TPredictArgument> getArguments() {
017        return arguments;
018    }
019
020    public void init(Object arg1, Object arg2){
021        functionType = EFunctionType.ml_predict_t;
022        functionName = ((TObjectName) arg1);
023        arguments = (ArrayList<TPredictArgument>) arg2;
024    }
025
026    public void doParse(TCustomSqlStatement psql, ESqlClause plocation){
027        for (TPredictArgument argument : arguments) {
028            argument.doParse(psql, plocation);
029        }
030    }
031
032    public void accept(TParseTreeVisitor v){
033        v.preVisit(this);
034        v.postVisit(this);
035    }
036
037    public void acceptChildren(TParseTreeVisitor v) {
038        v.preVisit(this);
039        v.postVisit(this);
040    }
041
042}