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 **/
019package lucee.runtime.functions.query;
020
021import lucee.runtime.PageContext;
022import lucee.runtime.config.NullSupportHelper;
023import lucee.runtime.exp.FunctionException;
024import lucee.runtime.exp.PageException;
025import lucee.runtime.functions.BIF;
026import lucee.runtime.op.Caster;
027import lucee.runtime.type.Collection;
028import lucee.runtime.type.Query;
029import lucee.runtime.type.Struct;
030import lucee.runtime.type.StructImpl;
031
032/**
033 * implements BIF QueryRowData
034 */
035public class QueryRowData extends BIF {
036
037        public static Struct call(PageContext pc, Query query, double rowNumber) throws PageException {
038
039                int row = Caster.toInteger(rowNumber);
040
041                if ( row < 1 || row > query.getRecordcount() )
042                        throw new FunctionException( pc, QueryRowData.class.getSimpleName(), 2, "rowNumber", "The argument rowNumber [" + row + "] must be between 1 and the query's record count [" + query.getRecordcount() + "]" );
043
044                Collection.Key[] colNames = query.getColumnNames();
045
046                Struct result = new StructImpl();
047
048                for (int col=0; col<colNames.length; col++)
049                        result.setEL(colNames[ col ], query.getAt(colNames[ col ], row, NullSupportHelper.empty()));
050
051                return result;
052        }
053
054        @Override
055        public Object invoke(PageContext pc, Object[] args) throws PageException {
056
057                return call(pc, Caster.toQuery(args[0]), Caster.toInteger(args[1]));
058        }
059}