EnumTupleOrdinalIndexedGetter.java

package org.simpleflatmapper.querydsl.getter;

import com.mysema.query.Tuple;
import org.simpleflatmapper.querydsl.TupleElementKey;
import org.simpleflatmapper.util.EnumHelper;
import org.simpleflatmapper.reflect.Getter;

public final class EnumTupleOrdinalIndexedGetter<E extends Enum<E>> implements  Getter<Tuple, E> {

	private final int index;
	private final Class<?> type;
	private final E[] values;
	

	public EnumTupleOrdinalIndexedGetter(TupleElementKey key, final Class<E> enumType) {
		this.index = key.getIndex();
		this.type = key.getExpression().getType();
		this.values = EnumHelper.getValues(enumType);
	}

	@Override
	public E get(final Tuple target) throws Exception {
		Number number = (Number) target.get(index, type);
		if (number != null) {
			return values[number.intValue()];
		} else {
			return null;
		}
	}
}