QuerySetColumn.java

/**
 * Implements the CFML Function queryaddcolumn
 */
package railo.runtime.functions.query;

import railo.runtime.PageContext;
import railo.runtime.exp.PageException;
import railo.runtime.functions.BIF;
import railo.runtime.op.Caster;
import railo.runtime.type.Array;
import railo.runtime.type.ArrayImpl;
import railo.runtime.type.Collection;
import railo.runtime.type.KeyImpl;
import railo.runtime.type.Query;
import railo.runtime.type.QueryColumn;

public final class QuerySetColumn extends BIF {

	private static final long serialVersionUID = -268309857190767441L;

	public static String call(PageContext pc , Query query, String columnName,String newColumnName) throws PageException {
		columnName=columnName.trim();
		newColumnName=newColumnName.trim();
		Collection.Key src=KeyImpl.getInstance(columnName);
		Collection.Key trg=KeyImpl.getInstance(newColumnName);
		
		Query qp = Caster.toQuery(query,null);
		if(qp!=null) qp.rename(src, trg);
		else {
			QueryColumn qc = query.removeColumn(src);
			Array content=new ArrayImpl();
			int len=qc.size();
			for(int i=1;i<=len;i++){
				content.setE(i, qc.get(i,null));
			}
			query.addColumn(trg, content, qc.getType());
		}
		return null;
	}
	
	@Override
	public Object invoke(PageContext pc, Object[] args) throws PageException {
		return call(pc,Caster.toQuery(args[0]),Caster.toString(args[1]),Caster.toString(args[2]));
	}
}