/*
* *************************************************************************************
* Copyright (C) 2008 EsperTech, Inc. All rights reserved. *
* http://esper.codehaus.org *
* 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.enummethod.eval;
import com.espertech.esper.client.EventBean;
import com.espertech.esper.epl.expression.ExprEvaluator;
import com.espertech.esper.epl.expression.ExprEvaluatorContext;
import com.espertech.esper.event.arr.ObjectArrayEventBean;
import com.espertech.esper.event.arr.ObjectArrayEventType;
import java.util.ArrayDeque;
import java.util.Collection;
import java.util.Collections;
public class EnumEvalTakeWhileIndexEvents implements EnumEval {
private ExprEvaluator innerExpression;
private int streamNumLambda;
private ObjectArrayEventType indexEventType;
public EnumEvalTakeWhileIndexEvents(ExprEvaluator innerExpression, int streamNumLambda, ObjectArrayEventType indexEventType) {
this.innerExpression = innerExpression;
this.streamNumLambda = streamNumLambda;
this.indexEventType = indexEventType;
}
public int getStreamNumSize() {
return streamNumLambda + 2;
}
public Object evaluateEnumMethod(EventBean[] eventsLambda, Collection target, boolean isNewData, ExprEvaluatorContext context) {
if (target.isEmpty()) {
return target;
}
ObjectArrayEventBean indexEvent = new ObjectArrayEventBean(new Object[1], indexEventType);
Collection<EventBean> beans = (Collection<EventBean>) target;
if (target.size() == 1) {
EventBean item = beans.iterator().next();
indexEvent.getProperties()[0] = 0;
eventsLambda[streamNumLambda] = item;
eventsLambda[streamNumLambda + 1] = indexEvent;
Object pass = innerExpression.evaluate(eventsLambda, isNewData, context);
if (pass == null || (!(Boolean) pass)) {
return Collections.emptyList();
}
return Collections.singletonList(item);
}
ArrayDeque<Object> result = new ArrayDeque<Object>();
int count = -1;
for (EventBean next : beans) {
count++;
indexEvent.getProperties()[0] = count;
eventsLambda[streamNumLambda] = next;
eventsLambda[streamNumLambda + 1] = indexEvent;
Object pass = innerExpression.evaluate(eventsLambda, isNewData, context);
if (pass == null || (!(Boolean) pass)) {
break;
}
result.add(next);
}
return result;
}
}