/*
***************************************************************************************
* Copyright (C) 2006 EsperTech, Inc. All rights reserved. *
* http://www.espertech.com/esper *
* http://www.espertech.com *
* ---------------------------------------------------------------------------------- *
* The software in this package is published under the terms of the GPL license *
* a copy of which has been included with this distribution in the license.txt file. *
***************************************************************************************
*/
package com.espertech.esper.epl.expression.subquery;
import com.espertech.esper.client.EventBean;
import com.espertech.esper.epl.agg.service.AggregationService;
import com.espertech.esper.epl.expression.core.ExprEvaluator;
import com.espertech.esper.epl.expression.core.ExprEvaluatorContext;
import com.espertech.esper.util.SimpleNumberCoercer;
import java.util.Collection;
/**
* Strategy for subselects with "=/!=/>< ALL".
*/
public class SubselectEvalStrategyNREqualsAllWGroupBy extends SubselectEvalStrategyNREqualsBase {
private final ExprEvaluator havingEval;
public SubselectEvalStrategyNREqualsAllWGroupBy(ExprEvaluator valueEval, ExprEvaluator selectEval, boolean resultWhenNoMatchingEvents, boolean notIn, SimpleNumberCoercer coercer, ExprEvaluator havingEval) {
super(valueEval, selectEval, resultWhenNoMatchingEvents, notIn, coercer);
this.havingEval = havingEval;
}
protected Object evaluateInternal(Object leftResult, EventBean[] events, boolean isNewData, Collection<EventBean> matchingEvents, ExprEvaluatorContext exprEvaluatorContext, AggregationService aggregationServiceAnyPartition) {
AggregationService aggregationService = aggregationServiceAnyPartition.getContextPartitionAggregationService(exprEvaluatorContext.getAgentInstanceId());
Collection<Object> groupKeys = aggregationService.getGroupKeys(exprEvaluatorContext);
boolean hasNullRow = false;
for (Object groupKey : groupKeys) {
if (leftResult == null) {
return null;
}
aggregationService.setCurrentAccess(groupKey, exprEvaluatorContext.getAgentInstanceId(), null);
if (havingEval != null) {
Boolean pass = (Boolean) havingEval.evaluate(events, true, exprEvaluatorContext);
if ((pass == null) || (!pass)) {
continue;
}
}
Object rightResult;
if (selectEval != null) {
rightResult = selectEval.evaluate(events, true, exprEvaluatorContext);
} else {
rightResult = events[0].getUnderlying();
}
if (rightResult != null) {
if (coercer == null) {
boolean eq = leftResult.equals(rightResult);
if ((isNot && eq) || (!isNot && !eq)) {
return false;
}
} else {
Number left = coercer.coerceBoxed((Number) leftResult);
Number right = coercer.coerceBoxed((Number) rightResult);
boolean eq = left.equals(right);
if ((isNot && eq) || (!isNot && !eq)) {
return false;
}
}
} else {
hasNullRow = true;
}
}
if (hasNullRow) {
return null;
}
return true;
}
}