org.sfm.jdbc.querydsl.QueryDslMapperBuilderImpl Maven / Gradle / Ivy

Go to download
package org.sfm.jdbc.querydsl;

import java.lang.reflect.Type;

import org.sfm.map.AbstractMapperBuilderImpl;
import org.sfm.map.Mapper;
import org.sfm.map.MapperBuildingException;
import org.sfm.map.MapperImpl;
import org.sfm.reflect.ReflectionService;
import org.sfm.reflect.meta.ClassMeta;

import com.mysema.query.Tuple;
import com.mysema.query.types.Expression;
import com.mysema.query.types.Path;
import com.mysema.query.types.PathMetadata;
import com.mysema.query.types.PathType;

public final class QueryDslMapperBuilderImpl 
	extends AbstractMapperBuilderImpl, Mapper, QueryDslMapperBuilder> 
	implements QueryDslMapperBuilder {


	public QueryDslMapperBuilderImpl(final Type target) throws MapperBuildingException {
		this(target, new ReflectionService());
	}
	
	@SuppressWarnings("unchecked")
	public QueryDslMapperBuilderImpl(final Type target, ReflectionService reflectService) throws MapperBuildingException {
		this(target, (ClassMeta) reflectService.getClassMeta(target));
	}
	
	public QueryDslMapperBuilderImpl(final Type target, final ClassMeta classMeta) throws MapperBuildingException {
		super(target, Tuple.class, classMeta, new TupleGetterFactory(), new TupleFieldMapperFactory(new TupleGetterFactory()));
	}

	@Override
	public Mapper mapper() {
		return new MapperImpl(fields(), getInstantiator());
	}
	
	@Override
	protected QueryDslMapperBuilder newMapperBuilder(Type type, ClassMeta classMeta) {
		return new  QueryDslMapperBuilderImpl(type, classMeta);
	}

	@Override
	public  QueryDslMapperBuilder addMapping(Expression expr, int i) {
		String propertyName = null;
		if  (expr instanceof Path) {
			@SuppressWarnings("rawtypes")
			PathMetadata metadata = ((Path) expr).getMetadata();
			if (metadata.getPathType() == PathType.PROPERTY) {
				propertyName = metadata.getExpression().toString();
			}
		} 
		
		if (propertyName == null) {
			throw new MapperBuildingException("Unsupported expression " + expr);
		}
		addMapping(propertyName, new TupleElementKey(expr, i));
		return this;
	}
}