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.cache.tag.query;
020
021import lucee.commons.io.res.util.UDFFilterSupport;
022import lucee.runtime.cache.tag.CacheHandlerFilter;
023import lucee.runtime.engine.ThreadLocalPageContext;
024import lucee.runtime.exp.ExpressionException;
025import lucee.runtime.exp.PageException;
026import lucee.runtime.exp.PageRuntimeException;
027import lucee.runtime.op.Caster;
028import lucee.runtime.type.Query;
029import lucee.runtime.type.UDF;
030
031public class QueryCacheHandlerFilterUDF extends UDFFilterSupport implements CacheHandlerFilter {
032        
033        private UDF udf;
034
035        public QueryCacheHandlerFilterUDF(UDF udf) throws ExpressionException{
036                super(udf);
037                this.udf=udf;
038        }
039
040        @Override
041    public boolean accept(Object obj) {
042                if(!(obj instanceof Query)) return false;
043                
044        args[0]=((Query)obj).getSql();
045        try {
046                        return Caster.toBooleanValue(udf.call(ThreadLocalPageContext.get(), args, true));
047                } 
048        catch (PageException e) {
049                        throw new PageRuntimeException(e);
050                }
051    }
052
053}