AttributeEqualsQueryCreator.java

package org.molgenis.data.annotation.core.query;

import org.molgenis.data.Entity;
import org.molgenis.data.Query;
import org.molgenis.data.annotation.core.entity.QueryCreator;
import org.molgenis.data.meta.model.Attribute;

import java.util.Arrays;
import java.util.Collection;

import static org.molgenis.data.support.QueryImpl.EQ;

/**
 * Create a query that finds rows that match gene name
 */
public class AttributeEqualsQueryCreator implements QueryCreator
{
	private final Attribute attribute;

	public AttributeEqualsQueryCreator(Attribute attribute)
	{
		this.attribute = attribute;
	}

	@Override
	public Collection<Attribute> getRequiredAttributes()
	{
		return Arrays.asList(attribute);
	}

	@Override
	public Query<Entity> createQuery(Entity entity)
	{
		Object value = entity.get(attribute.getName());
		return EQ(attribute.getName(), value);
	}

}