AbstractQuery.java

/**
 * Copyright (C) 2010 Michael Mosmann <michael@mosmann.de>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package de.flapdoodle.mongoom.datastore.query;

import java.util.Arrays;

import com.mongodb.DBObject;

import de.flapdoodle.mongoom.datastore.factories.IDBObjectFactory;
import de.flapdoodle.mongoom.exceptions.MappingException;
import de.flapdoodle.mongoom.mapping.ITransformation;
import de.flapdoodle.mongoom.mapping.naming.PropertyName;
import de.flapdoodle.mongoom.mapping.properties.IPropertyMappedName;
import de.flapdoodle.mongoom.mapping.properties.Property;
import de.flapdoodle.mongoom.mapping.properties.PropertyReference;

public abstract class AbstractQuery<T, C extends ITransformation> {

	private final C _converter;
	private final IDBObjectFactory _queryBuilder;

	public AbstractQuery(C converter, IDBObjectFactory queryBuilder) {
		_converter = converter;
		_queryBuilder = queryBuilder;
	}

	protected C getConverter() {
		return _converter;
	}

	protected IDBObjectFactory getQueryBuilder() {
		return _queryBuilder;
	}

	public DBObject asDBObject() {
		return getQueryBuilder().get();
	}

	protected <T> MappedNameTransformation getConverter(PropertyReference<T> field) {
		C entityConverter = getConverter();

		ITransformation converter = entityConverter;
		
		return Queries.getConverter(field, converter,null);
	}

	protected MappedNameTransformation getConverter(String... field) {
		C entityConverter = getConverter();

		IPropertyMappedName name = null;
		ITransformation converter = entityConverter;
		for (String f : field) {
			ITransformation lastConverter = converter;
			PropertyName propertyName = converter.propertyName(f);
			if (name == null)
				name = propertyName;
			else {
				name = Property.append(name, propertyName);
			}
			converter = converter.propertyTransformation(propertyName);
			if (converter == null)
				throw new MappingException("No Converter for " + Arrays.asList(field) + " in " + lastConverter);
		}
		return new MappedNameTransformation(name, converter);
	}

}