Selects.java

package railo.runtime.sql;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import railo.runtime.exp.DatabaseException;
import railo.runtime.sql.exp.Column;
import railo.runtime.sql.exp.Expression;
import railo.runtime.sql.exp.op.Operation;
import railo.runtime.sql.exp.value.ValueNumber;

public class Selects {

	private List<Column> orderbys=new ArrayList<Column>();
	private List<Select> selects=new ArrayList<Select>();
	


	public void addOrderByExpression(Column exp) {
		this.orderbys.add(exp);
	}

	/**
	 * @return the orderbys
	 */
	public Column[] getOrderbys() {
		if(orderbys==null) return new Column[0];
		return orderbys.toArray(new Column[orderbys.size()]);
	}

	public void addSelect(Select select) {
		selects.add(select);
	}
	
	public Select[] getSelects() {
		if(selects==null) return new Select[0];
		return selects.toArray(new Select[selects.size()]);
	}
	

	@Override

	public String toString() {
		return _toString(this);
	}
	public static String _toString(Selects __selects) {
		Select[] _selects = __selects.getSelects();
		Select s;
		StringBuffer sb=new StringBuffer();
		
		for(int y=0;y<_selects.length;y++) {
			s = _selects[y];
			
			if(y>0) {
				if(s.isUnionDistinct())sb.append("union distinct\n\n");
				else sb.append("union\n\n");
			}
			
			sb.append("select\n\t");

			if(s.isDistinct()) sb.append("distinct\n\t");
			ValueNumber top = s.getTop();
			if(top!=null) sb.append("top "+top.getString()+"\n\t");
			// select
			Expression[] sels = s.getSelects();
			Expression exp;
			boolean first=true;
			for(int i=0;i<sels.length;i++) {
				if(!first)sb.append("\t,");
				exp=sels[i];
				sb.append(exp.toString(false)+"\n");
				first=false;
			}
	
			// from
			sb.append("from\n\t");
			Column[] forms = s.getFroms();
			first=true;
			for(int i=0;i<forms.length;i++) {
				if(!first)sb.append("\t,");
				exp=forms[i];
				sb.append(exp.toString(false)+"\n");
				first=false;
			}
	
			// where
			if(s.getWhere()!=null){
				sb.append("where \n\t");
				sb.append(s.getWhere().toString(true));
				sb.append("\n");
			}
	
			// group by
			Column[] gbs = s.getGroupbys();
			if(gbs.length>0) {
				sb.append("group by\n\t");
				first=true;
				for(int i=0;i<gbs.length;i++) {
					if(!first)sb.append("\t,");
					exp=gbs[i];
					sb.append(exp.toString(false)+"\n");
					first=false;
				}
			}
	
			
			// having
			Operation having = s.getHaving();
			if(having!=null){
				sb.append("having \n\t");
				sb.append(having.toString(true));
				sb.append("\n");
			}
			
		}

		// order by
		if(__selects.orderbys!=null && __selects.orderbys.size()>0) {
			sb.append("order by\n\t");
			Iterator<Column> it = __selects.orderbys.iterator();
			Expression exp;
			boolean first = true;
			while(it.hasNext()) {
				if(!first)sb.append("\t,");
				exp=it.next();
				sb.append(exp.toString(false)+" "+(exp.isDirectionBackward()?"DESC":"ASC")+"\n");
				first=false;
			}
		}
		return sb.toString();
	}

	public Column[] getTables() {
		Iterator<Select> it = selects.iterator();
		Select s;
		ArrayList<Column> rtn=new ArrayList<Column>();
		Column[] froms;
		while(it.hasNext()) {
			s=it.next();
			froms = s.getFroms();
			for(int i=0;i<froms.length;i++) {
				rtn.add(froms[i]);
			}
		}
		return rtn.toArray(new Column[rtn.size()]);
	}

	public boolean isDistinct() {
		Select s;
		int len=selects.size();
		if(len==1) {
			s=selects.get(0);
			return s.isDistinct();
		}
		for(int i=1;i<len;i++) {
			s=selects.get(i);
			if(!s.isUnionDistinct()) return false;
		}
		return true;
	}
	public Column[] getDistincts() throws DatabaseException {
		List<Column> columns=new ArrayList<Column>();
		Select s;
		int len=selects.size();
		if(len==1) {
			s=selects.get(0);
			Expression[] _selects = s.getSelects();
			for(int i=0;i<_selects.length;i++){
				if(_selects[i] instanceof Column) {
					columns.add((Column) _selects[i]);
				}	
			}
			return columns.toArray(new Column[columns.size()]);
		}
		throw new DatabaseException("not supported for Union distinct",null,null,null);
	}
}