001/** 002 * 003 * Copyright (c) 2014, the Railo Company Ltd. All rights reserved. 004 * 005 * This library is free software; you can redistribute it and/or 006 * modify it under the terms of the GNU Lesser General Public 007 * License as published by the Free Software Foundation; either 008 * version 2.1 of the License, or (at your option) any later version. 009 * 010 * This library is distributed in the hope that it will be useful, 011 * but WITHOUT ANY WARRANTY; without even the implied warranty of 012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 013 * Lesser General Public License for more details. 014 * 015 * You should have received a copy of the GNU Lesser General Public 016 * License along with this library. If not, see <http://www.gnu.org/licenses/>. 017 * 018 **/ 019/** 020 * Implements the CFML Function queryaddcolumn 021 */ 022package lucee.runtime.functions.query; 023 024import lucee.runtime.PageContext; 025import lucee.runtime.exp.PageException; 026import lucee.runtime.functions.BIF; 027import lucee.runtime.op.Caster; 028import lucee.runtime.type.Array; 029import lucee.runtime.type.ArrayImpl; 030import lucee.runtime.type.Collection; 031import lucee.runtime.type.KeyImpl; 032import lucee.runtime.type.Query; 033import lucee.runtime.type.QueryColumn; 034 035public final class QuerySetColumn extends BIF { 036 037 private static final long serialVersionUID = -268309857190767441L; 038 039 public static String call(PageContext pc , Query query, String columnName,String newColumnName) throws PageException { 040 columnName=columnName.trim(); 041 newColumnName=newColumnName.trim(); 042 Collection.Key src=KeyImpl.getInstance(columnName); 043 Collection.Key trg=KeyImpl.getInstance(newColumnName); 044 045 Query qp = Caster.toQuery(query,null); 046 if(qp!=null) qp.rename(src, trg); 047 else { 048 QueryColumn qc = query.removeColumn(src); 049 Array content=new ArrayImpl(); 050 int len=qc.size(); 051 for(int i=1;i<=len;i++){ 052 content.setE(i, qc.get(i,null)); 053 } 054 query.addColumn(trg, content, qc.getType()); 055 } 056 return null; 057 } 058 059 @Override 060 public Object invoke(PageContext pc, Object[] args) throws PageException { 061 return call(pc,Caster.toQuery(args[0]),Caster.toString(args[1]),Caster.toString(args[2])); 062 } 063}