bundles/org.eclipse.equinox.p2.metadata/src/org/eclipse/equinox/internal/p2/metadata/expression/Select.java

/*******************************************************************************
 * Copyright (c) 2009, 2017 Cloudsmith Inc. and others.
 *
 * This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *     Cloudsmith Inc. - initial API and implementation
 *******************************************************************************/
package org.eclipse.equinox.internal.p2.metadata.expression;

import java.util.Iterator;
import org.eclipse.equinox.p2.metadata.expression.IEvaluationContext;

/**
 * An expression that yields a new collection consisting of all elements of the
 * <code>collection</code> for which the <code>filter</code> yields <code>true</code>.
 */
final class Select extends CollectionFilter {
	Select(Expression collection, LambdaExpression lambda) {
		super(collection, lambda);
	}

	@Override
	protected Object evaluate(IEvaluationContext context, Iterator<?> itor) {
		return evaluateAsIterator(context, itor);
	}

	@Override
	protected Iterator<?> evaluateAsIterator(final IEvaluationContext context, Iterator<?> itor) {
		return new MatchIteratorFilter<Object>(itor) {
			@Override
			protected boolean isMatch(Object val) {
				lambda.getItemVariable().setValue(context, val);
				return lambda.evaluate(context) == Boolean.TRUE;
			}
		};
	}

	@Override
	public int getExpressionType() {
		return TYPE_SELECT;
	}

	@Override
	public String getOperator() {
		return KEYWORD_SELECT;
	}

	boolean isCollection() {
		return true;
	}
}