001    /*
002     * Cumulus4j - Securing your data in the cloud - http://cumulus4j.org
003     * Copyright (C) 2011 NightLabs Consulting GmbH
004     *
005     * This program is free software: you can redistribute it and/or modify
006     * it under the terms of the GNU Affero General Public License as
007     * published by the Free Software Foundation, either version 3 of the
008     * License, or (at your option) any later version.
009     *
010     * This program 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
013     * GNU Affero General Public License for more details.
014     *
015     * You should have received a copy of the GNU Affero General Public License
016     * along with this program.  If not, see <http://www.gnu.org/licenses/>.
017     */
018    package org.cumulus4j.store.query.method;
019    
020    import java.util.Set;
021    
022    import org.cumulus4j.store.model.FieldMetaRole;
023    import org.cumulus4j.store.query.QueryEvaluator;
024    import org.cumulus4j.store.query.eval.ExpressionHelper;
025    import org.cumulus4j.store.query.eval.InvokeExpressionEvaluator;
026    import org.cumulus4j.store.query.eval.ResultDescriptor;
027    import org.datanucleus.query.QueryUtils;
028    import org.datanucleus.query.expression.Expression;
029    import org.datanucleus.query.expression.Literal;
030    import org.datanucleus.query.expression.ParameterExpression;
031    import org.datanucleus.query.expression.PrimaryExpression;
032    import org.datanucleus.query.expression.VariableExpression;
033    
034    /**
035     * Evaluator for "Map.containsKey(key)".
036     */
037    public class MapContainsKeyEvaluator extends AbstractMethodEvaluator
038    {
039            /* (non-Javadoc)
040             * @see org.cumulus4j.store.query.method.MethodEvaluator#evaluate(org.cumulus4j.store.query.QueryEvaluator, org.datanucleus.query.expression.InvokeExpression, org.datanucleus.query.expression.Expression, org.cumulus4j.store.query.eval.ResultDescriptor)
041             */
042            @Override
043            public Set<Long> evaluate(QueryEvaluator queryEval, InvokeExpressionEvaluator invokeExprEval, 
044                            Expression invokedExpr, ResultDescriptor resultDesc) {
045                    if (invokeExprEval.getExpression().getArguments().size() != 1)
046                            throw new IllegalStateException("containsKey(...) expects exactly one argument, but there are " + 
047                                            invokeExprEval.getExpression().getArguments().size());
048    
049                    if (invokedExpr instanceof PrimaryExpression) {
050                            // Evaluate the invoke argument
051                            Expression invokeArgExpr = invokeExprEval.getExpression().getArguments().get(0);
052                            Object invokeArgument;
053                            if (invokeArgExpr instanceof Literal)
054                                    invokeArgument = ((Literal)invokeArgExpr).getLiteral();
055                            else if (invokeArgExpr instanceof ParameterExpression)
056                                    invokeArgument = QueryUtils.getValueForParameterExpression(queryEval.getParameterValues(), (ParameterExpression)invokeArgExpr);
057                            else if (invokeArgExpr instanceof VariableExpression)
058                                    return new ExpressionHelper.ContainsVariableResolver(
059                                                    queryEval, (PrimaryExpression) invokedExpr, FieldMetaRole.mapKey, (VariableExpression) invokeArgExpr,
060                                                    resultDesc.isNegated()
061                                    ).query();
062                            else
063                                    throw new UnsupportedOperationException("NYI");
064    
065                            return new ExpressionHelper.ContainsConstantResolver(
066                                            queryEval, (PrimaryExpression) invokedExpr, FieldMetaRole.mapKey, invokeArgument,
067                                            resultDesc.isNegated()
068                            ).query();
069                    }
070                    else {
071                            throw new UnsupportedOperationException("NYI invocation of Map.containsKey on a variable");
072                    }
073            }
074    }