InMemoryQueryVisitor.java example

Explorer
strolch-master
/*
 * Copyright 2013 Robert von Burg <eitch@eitchnet.ch>
 * 
 * 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 li.strolch.runtime.query.inmemory;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

import li.strolch.model.GroupedParameterizedElement;
import li.strolch.model.query.AndSelection;
import li.strolch.model.query.BooleanSelection;
import li.strolch.model.query.IdSelection;
import li.strolch.model.query.NameSelection;
import li.strolch.model.query.NotSelection;
import li.strolch.model.query.OrSelection;
import li.strolch.model.query.ParameterBagSelection;
import li.strolch.model.query.ParameterBagSelection.NullParameterBagSelection;
import li.strolch.model.query.ParameterSelection.AnyTypeParameterSelection;
import li.strolch.model.query.ParameterSelection.BooleanParameterSelection;
import li.strolch.model.query.ParameterSelection.DateParameterSelection;
import li.strolch.model.query.ParameterSelection.DateRangeParameterSelection;
import li.strolch.model.query.ParameterSelection.DurationParameterSelection;
import li.strolch.model.query.ParameterSelection.FloatListParameterSelection;
import li.strolch.model.query.ParameterSelection.FloatParameterSelection;
import li.strolch.model.query.ParameterSelection.IntegerListParameterSelection;
import li.strolch.model.query.ParameterSelection.IntegerParameterSelection;
import li.strolch.model.query.ParameterSelection.LongListParameterSelection;
import li.strolch.model.query.ParameterSelection.LongParameterSelection;
import li.strolch.model.query.ParameterSelection.NullParameterSelection;
import li.strolch.model.query.ParameterSelection.StringListParameterSelection;
import li.strolch.model.query.ParameterSelection.StringParameterSelection;
import li.strolch.model.query.ParameterSelectionVisitor;
import li.strolch.model.query.Selection;
import li.strolch.model.query.StrolchRootElementSelectionVisitor;
import li.strolch.model.query.ordering.OrderById;
import li.strolch.model.query.ordering.OrderByName;
import li.strolch.model.query.ordering.OrderByParameter;
import li.strolch.model.query.ordering.StrolchQueryOrderingVisitor;
import li.strolch.persistence.api.StrolchDao;
import li.strolch.runtime.query.inmemory.ParameterBagSelector.NullParameterBagSelector;
import li.strolch.runtime.query.inmemory.ParameterSelector.StringParameterSelector;
import li.strolch.utils.dbc.DBC;

/**
 * @author Robert von Burg <eitch@eitchnet.ch>
 */
public abstract class InMemoryQueryVisitor<T extends GroupedParameterizedElement, S extends StrolchDao<?>>
		implements StrolchRootElementSelectionVisitor, ParameterSelectionVisitor, StrolchQueryOrderingVisitor {

	private Comparator<T> comparator;
	private Navigator<T> navigator;
	private List<Selector<T>> selectors;
	private boolean any;

	public InMemoryQueryVisitor() {
		this.selectors = new ArrayList<>();
	}

	public List<Selector<T>> getSelectors() {
		return Collections.unmodifiableList(this.selectors);
	}

	public Navigator<T> getNavigator() {
		return this.navigator;
	}

	public Comparator<T> getComparator() {
		return this.comparator;
	}

	/**
	 * @param navigator
	 */
	protected void setNavigator(StrolchTypeNavigator<T> navigator) {
		this.navigator = navigator;
	}

	/**
	 * Returns a new instance of this concrete type for performing recursive {@link BooleanSelection BooleanSelections}
	 * 
	 * @return a new instance of this concrete type
	 */
	protected abstract InMemoryQueryVisitor<T, S> newInstance();

	private void assertNotAny() {
		DBC.INTERIM.assertFalse("Not allowed to use further Selections with Any!", this.any); //$NON-NLS-1$
	}

	protected void addSelector(Selector<T> selector) {
		assertNotAny();
		this.selectors.add(selector);
	}

	@Override
	public void visitAny() {
		DBC.PRE.assertEmpty("Only one selection allowed when using Any!", this.selectors); //$NON-NLS-1$
		addSelector(new AnySelector<T>());
		this.any = true;
	}

	@Override
	public void visitAnd(AndSelection andSelection) {
		assertNotAny();
		InMemoryQueryVisitor<T, S> query = newInstance();
		List<Selection> selections = andSelection.getSelections();
		for (Selection selection : selections) {
			selection.accept(query);
		}
		AndSelector<T> andSelector = new AndSelector<>(query.getSelectors());
		addSelector(andSelector);
	}

	@Override
	public void visitOr(OrSelection orSelection) {
		assertNotAny();
		InMemoryQueryVisitor<T, S> query = newInstance();
		List<Selection> selections = orSelection.getSelections();
		for (Selection selection : selections) {
			selection.accept(query);
		}
		OrSelector<T> orSelector = new OrSelector<>(query.getSelectors());
		addSelector(orSelector);
	}

	@Override
	public void visitNot(NotSelection notSelection) {
		assertNotAny();
		InMemoryQueryVisitor<T, S> query = newInstance();
		List<Selection> selections = notSelection.getSelections();
		for (Selection selection : selections) {
			selection.accept(query);
		}
		List<Selector<T>> notSelectors = query.getSelectors();
		if (!notSelectors.isEmpty()) {
			NotSelector<T> notSelector = new NotSelector<>(notSelectors.get(0));
			addSelector(notSelector);
		}
	}

	@Override
	public void visit(IdSelection selection) {
		addSelector(new IdSelector<T>(selection.getIds(), selection.getMatchMode()));
	}

	@Override
	public void visit(NameSelection selection) {
		addSelector(new NameSelector<T>(selection.getName(), selection.getMatchMode()));
	}

	@Override
	public void visit(StringParameterSelection selection) {
		StringParameterSelector<T> stringSelector = //
				ParameterSelector.<T> stringSelector( //
						selection.getBagKey(), //
						selection.getParamKey(), //
						selection.getValue(), //
						selection.getMatchMode());

		addSelector(stringSelector);
	}

	@Override
	public void visit(IntegerParameterSelection selection) {
		addSelector(ParameterSelector.<T> integerSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(BooleanParameterSelection selection) {
		addSelector(ParameterSelector.<T> booleanSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(LongParameterSelection selection) {
		addSelector(ParameterSelector.<T> longSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(FloatParameterSelection selection) {
		addSelector(ParameterSelector.<T> floatSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(DateParameterSelection selection) {
		addSelector(ParameterSelector.<T> dateSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(DurationParameterSelection selection) {
		addSelector(ParameterSelector.<T> durationSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(DateRangeParameterSelection selection) {
		addSelector(ParameterSelector.<T> dateRangeSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getDateRange()));
	}

	@Override
	public void visit(StringListParameterSelection selection) {
		addSelector(ParameterSelector.<T> stringListSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(IntegerListParameterSelection selection) {
		addSelector(ParameterSelector.<T> integerListSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(FloatListParameterSelection selection) {
		addSelector(ParameterSelector.<T> floatListSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(LongListParameterSelection selection) {
		addSelector(ParameterSelector.<T> longListSelector(selection.getBagKey(), selection.getParamKey(),
				selection.getValue()));
	}

	@Override
	public void visit(NullParameterSelection selection) {
		addSelector(ParameterSelector.<T> nullSelector(selection.getBagKey(), selection.getParamKey()));
	}

	@Override
	public void visit(NullParameterBagSelection selection) {
		addSelector(new NullParameterBagSelector<T>(selection.getBagKey()));
	}

	@Override
	public void visit(ParameterBagSelection selection) {
		addSelector(new ParameterBagSelector<T>(selection.getBagKey()));
	}

	@Override
	public void visit(AnyTypeParameterSelection selection) {
		addSelector(ParameterSelector.<T> anyTypeSelection(selection.getBagKey(), selection.getParamKey(),
				selection.getValue(), selection.getMatchMode()));
	}

	@Override
	public InMemoryQueryVisitor<T, S> visit(OrderById ordering) {
		this.comparator = new InMemoryStrolchQueryOrderingVisitor<T>().visit(ordering).getComparator();
		return this;
	}

	@Override
	public InMemoryQueryVisitor<T, S> visit(OrderByName ordering) {
		this.comparator = new InMemoryStrolchQueryOrderingVisitor<T>().visit(ordering).getComparator();
		return this;
	}

	@Override
	public InMemoryQueryVisitor<T, S> visit(OrderByParameter ordering) {
		this.comparator = new InMemoryStrolchQueryOrderingVisitor<T>().visit(ordering).getComparator();
		return this;
	}
}