001package gudusoft.gsqlparser.nodes.hive;
002
003import gudusoft.gsqlparser.nodes.*;
004
005import java.util.ArrayList;
006
007
008public class THiveFromQuerySqlNode extends TParseTreeNode {
009    private ArrayList<TParseTreeNode> hiveBodyList;
010    private TFromTableList fromTableList;
011
012    public void setCteList(TCTEList cteList) {
013        this.cteList = cteList;
014    }
015
016    public TCTEList getCteList() {
017
018        return cteList;
019    }
020
021    private TCTEList cteList;
022
023    public void init(Object arg1){
024        fromTableList = (TFromTableList)arg1;
025    }
026
027    public ArrayList<TParseTreeNode> getHiveBodyList() {
028        return hiveBodyList;
029    }
030
031    public TFromTableList getFromTableList() {
032        return fromTableList;
033    }
034
035    public void init(Object arg1,Object arg2){
036        init(arg1);
037        hiveBodyList = (ArrayList<TParseTreeNode>)arg2;
038
039    }
040
041    public TSelectSqlNode convertToSelect(){
042        if (hiveBodyList.size() > 1) return null;
043        if (!(hiveBodyList.get(0) instanceof TSelectSqlNode)) return null;
044        TSelectSqlNode ret = (TSelectSqlNode)hiveBodyList.get(0);
045        ret.setFromTableList(this.getFromTableList());
046        ret.setStartToken(this.getStartToken());
047        return ret;
048    }
049
050}