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 queryaddrow
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.op.Decision;
029import lucee.runtime.type.Query;
030
031public final class QueryAddRow extends BIF {
032
033        private static final long serialVersionUID = 1252130736067181453L;
034
035        public static double call(PageContext pc , Query query) {
036                query.addRow(1);
037                return query.getRecordcount();
038        }
039        public static double call(PageContext pc , Query query, Object numberOrData) throws PageException {
040                if(numberOrData==null) return call(pc, query);
041                else if(Decision.isNumeric(numberOrData)) {
042                        query.addRow(Caster.toIntValue(numberOrData));
043                }
044                else {
045                        QueryNew.populate(pc, query, numberOrData);
046                }
047                return query.getRecordcount();
048        }
049        
050        @Override
051        public Object invoke(PageContext pc, Object[] args) throws PageException {
052                if(args.length==1)return call(pc,Caster.toQuery(args[0]));
053                return call(pc,Caster.toQuery(args[0]),args[1]);
054        }
055}