com.googlecode.mjorm.query.QueryGroup Maven / Gradle / Ivy

Go to download
package com.googlecode.mjorm.query;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;

import com.googlecode.mjorm.ObjectMapper;
import com.googlecode.mjorm.mql.MqlCriterionFunction;
import com.googlecode.mjorm.mql.AbstractMqlCriterionFunction;
import com.googlecode.mjorm.query.criteria.AbstractCriterion;
import com.googlecode.mjorm.query.criteria.AbstractQueryCriterion;
import com.googlecode.mjorm.query.criteria.Criterion;
import com.googlecode.mjorm.query.criteria.EqualsCriterion;
import com.googlecode.mjorm.query.criteria.FieldCriterion;
import com.mongodb.BasicDBList;

public class QueryGroup
	extends AbstractCriterion {

	private List queryCriterions = new ArrayList();

	public QueryGroup() {
		
	}

	public QueryGroup(Collection queries) {
		addAll(queries);
	}

	public QueryGroup(Query... queries) {
		addAll(queries);
	}

	/**
	 * Adds a {@link AbstractQueryCriterion} to the list of conditions.
	 * @return the {@link AbstractQueryCriterion}
	 */
	public Query add() {
		Query ret = new Query();
		add(ret);
		return ret;
	}

	/**
	 * Adds a {@link AbstractQueryCriterion} to this group.
	 * @param queryCriterion the {@link AbstractQueryCriterion}
	 */
	public QueryGroup add(Query queryCriterion) {
		queryCriterions.add(queryCriterion);
		return this;
	}

	/**
	 * Adds a {@link AbstractQueryCriterion} to this group.
	 * @param queryCriterion the {@link AbstractQueryCriterion}
	 */
	public QueryGroup addAll(Collection queries) {
		queryCriterions.addAll(queries);
		return this;
	}

	/**
	 * Adds a {@link AbstractQueryCriterion} to this group.
	 * @param queryCriterion the {@link AbstractQueryCriterion}
	 */
	public QueryGroup addAll(Query... queries) {
		queryCriterions.addAll(Arrays.asList(queries));
		return this;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public Object toQueryObject(ObjectMapper mapper) {
		BasicDBList list = new BasicDBList();
		for (Query queryCriterion : queryCriterions) {
			list.add(queryCriterion.toQueryObject(mapper));
		}
		return list;
	}

	public static MqlCriterionFunction createMqlDocumentFunction(
		final String functionName, final String operatorName, final boolean allowQueryGroup, final boolean allowQuery) {
		return createMqlDocumentFunction(
			functionName, operatorName, allowQueryGroup, allowQuery, -1, -1, -1);
	}

	public static MqlCriterionFunction createMqlDocumentFunction(
		final String functionName, final String operatorName,
		final int exactArgs, final int minArgs, final int maxArgs, final Class... types) {
		return createMqlDocumentFunction(
			functionName, operatorName, false, false, exactArgs, minArgs, maxArgs, types);
	}

	public static MqlCriterionFunction createMqlDocumentFunction(
		final String functionName, final String operatorName,
		final int exactArgs, final Class... types) {
		return createMqlDocumentFunction(
			functionName, operatorName, false, false, exactArgs, -1, -1, types);
	}

	public static MqlCriterionFunction createMqlDocumentFunction(
		final String functionName, final String operatorName,
		final boolean allowQueryGroup, final boolean allowQuery,
		final int exactArgs, final int minArgs, final int maxArgs, final Class... types) {
		return new AbstractMqlCriterionFunction() {
			protected void init() {
				setFunctionName(functionName);
				setAllowQueryGroup(allowQueryGroup);
				setAllowQuery(allowQuery);
				setExactArgs(exactArgs);
				setMinArgs(minArgs);
				setMaxArgs(maxArgs);
				setTypes(types);
			}
			
			@Override
			protected Criterion doCreate(Query query) {
				return new FieldCriterion(operatorName, new QueryGroup(query));
			}
			@Override
			protected Criterion doCreate(QueryGroup queryGroup) {
				return new FieldCriterion(operatorName, queryGroup);
			}
			@Override
			protected Criterion doCreate(Object[] values) {
				return new FieldCriterion(operatorName, new EqualsCriterion(values));
			}
		};
	}

}