com.holonplatform.vaadin.internal.data.DatastoreItemDataProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of holon-vaadin Show documentation
Holon Vaadin integration
There is a newer version: 5.4.0
/*
 * Copyright 2016-2017 Axioma srl.
 * 
 * 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 com.holonplatform.vaadin.internal.data;

import java.util.LinkedList;
import java.util.List;
import java.util.Optional;
import java.util.stream.Stream;

import com.holonplatform.core.ParameterSet;
import com.holonplatform.core.datastore.DataTarget;
import com.holonplatform.core.datastore.Datastore;
import com.holonplatform.core.exceptions.DataAccessException;
import com.holonplatform.core.internal.utils.ObjectUtils;
import com.holonplatform.core.property.Property;
import com.holonplatform.core.property.PropertyBox;
import com.holonplatform.core.property.PropertySet;
import com.holonplatform.core.query.Query;
import com.holonplatform.core.query.QueryConfigurationProvider;
import com.holonplatform.core.query.QueryFilter;
import com.holonplatform.core.query.QuerySort;
import com.holonplatform.vaadin.data.ItemDataProvider;
import com.holonplatform.vaadin.data.ItemIdentifierProvider;
import com.vaadin.shared.Registration;

/**
 * An {@link ItemDataProvider} using a {@link Datastore} to perform item set count and load operations, using
 * {@link Property} as item property type and {@link PropertyBox} as concrete item data container.
 * 
 * @since 5.0.0
 */
public class DatastoreItemDataProvider implements ItemDataProvider {

	private static final long serialVersionUID = -3647676181555142846L;

	/**
	 * Datastore
	 */
	private final Datastore datastore;

	/**
	 * Data target
	 */
	private final DataTarget target;

	/**
	 * Property set provider
	 */
	private final PropertySet propertySet;

	/**
	 * Item identifier
	 */
	private ItemIdentifierProvider itemIdentifier;

	/**
	 * Query configuration providers
	 */
	private List queryConfigurationProviders = new LinkedList<>();

	/**
	 * Construct a new DatastoreItemDataProvider.
	 * @param datastore Datastore to use (not null)
	 * @param target Data target (not null)
	 * @param propertySet Property set to load
	 */
	public DatastoreItemDataProvider(Datastore datastore, DataTarget target, PropertySet propertySet) {
		super();
		ObjectUtils.argumentNotNull(datastore, "Datastore must be not null");
		ObjectUtils.argumentNotNull(target, "DataTarget must be not null");
		ObjectUtils.argumentNotNull(propertySet, "PropertySet must be not null");
		this.datastore = datastore;
		this.target = target;
		this.propertySet = propertySet;
	}

	public DatastoreItemDataProvider(Datastore datastore, DataTarget target, PropertySet propertySet,
			ItemIdentifierProvider itemIdentifier) {
		super();
		ObjectUtils.argumentNotNull(datastore, "Datastore must be not null");
		ObjectUtils.argumentNotNull(target, "DataTarget must be not null");
		ObjectUtils.argumentNotNull(propertySet, "PropertySet must be not null");
		this.datastore = datastore;
		this.target = target;
		this.propertySet = propertySet;
		this.itemIdentifier = itemIdentifier;
	}

	/**
	 * Get the {@link Datastore} to use to perform count and load operations.
	 * @return the datastore
	 */
	protected Datastore getDatastore() {
		return datastore;
	}

	/**
	 * Get the data target to use.
	 * @return the data target
	 */
	protected DataTarget getTarget() {
		return target;
	}

	/**
	 * Get the item identifier provider.
	 * @return the optional item identifier provider
	 */
	public Optional> getItemIdentifier() {
		return Optional.ofNullable(itemIdentifier);
	}

	/**
	 * Set the item identifier provider.
	 * @param itemIdentifier the item identifier provider to set
	 */
	public void setItemIdentifier(ItemIdentifierProvider itemIdentifier) {
		this.itemIdentifier = itemIdentifier;
	}

	public Registration addQueryConfigurationProvider(QueryConfigurationProvider queryConfigurationProvider) {
		ObjectUtils.argumentNotNull(queryConfigurationProvider, "QueryConfigurationProvider must be not null");
		queryConfigurationProviders.add(queryConfigurationProvider);
		return () -> queryConfigurationProviders.remove(queryConfigurationProvider);
	}

	/*
	 * (non-Javadoc)
	 * @see com.holonplatform.vaadin.data.ItemSetCounter#size(com.holonplatform.core.query.QueryConfigurationProvider)
	 */
	@Override
	public long size(QueryConfigurationProvider configuration) throws DataAccessException {
		try {
			return buildQuery(configuration, false).count();
		} catch (Exception e) {
			throw new DataAccessException(e);
		}
	}

	/*
	 * (non-Javadoc)
	 * @see com.holonplatform.vaadin.data.ItemSetLoader#load(com.holonplatform.core.query.QueryConfigurationProvider,
	 * int, int)
	 */
	@Override
	public Stream load(QueryConfigurationProvider configuration, int offset, int limit)
			throws DataAccessException {
		Query q = buildQuery(configuration, true);
		// paging
		if (limit > 0) {
			q.limit(limit);
			q.offset(offset);
		}
		// execute
		Stream results = q.stream(propertySet);
		if (getItemIdentifier().isPresent()) {
			return results.map(pb -> new IdentifiablePropertyBox(pb, getItemIdentifier().get()));
		}
		return results;
	}

	/*
	 * (non-Javadoc)
	 * @see com.holonplatform.vaadin.data.ItemDataProvider#refresh(java.lang.Object)
	 */
	@Override
	public PropertyBox refresh(PropertyBox item) throws UnsupportedOperationException, DataAccessException {
		return getDatastore().refresh(getTarget(), item);
	}

	/**
	 * Build a {@link Query} using the Datastore and configuring query filters and sorts.
	 * @param configuration Query configuration
	 * @param withSorts Whether to apply sorts, if any, to query
	 * @return Query instance
	 */
	protected Query buildQuery(QueryConfigurationProvider configuration, boolean withSorts) {
		Query q = getDatastore().query();

		// target
		if (getTarget() != null) {
			q.target(getTarget());
		}

		// filter
		final List filters = new LinkedList<>();

		QueryFilter filter = configuration.getQueryFilter();
		if (filter != null) {
			filters.add(filter);
		}

		queryConfigurationProviders.forEach(p -> {
			QueryFilter qf = p.getQueryFilter();
			if (qf != null) {
				filters.add(qf);
			}
		});

		QueryFilter.allOf(filters).ifPresent(f -> q.filter(f));

		// sort
		if (withSorts) {

			final List sorts = new LinkedList<>();

			QuerySort sort = configuration.getQuerySort();
			if (sort != null) {
				sorts.add(sort);
			}

			queryConfigurationProviders.forEach(p -> {
				QuerySort qs = p.getQuerySort();
				if (qs != null) {
					sorts.add(qs);
				}
			});

			if (!sorts.isEmpty()) {
				if (sorts.size() == 1) {
					q.sort(sorts.get(0));
				} else {
					q.sort(QuerySort.of(sorts));
				}
			}
		}
		// parameters
		ParameterSet parameters = configuration.getQueryParameters();
		if (parameters != null) {
			parameters.forEachParameter((n, v) -> q.parameter(n, v));
		}

		queryConfigurationProviders.forEach(p -> {
			if (p.getQueryParameters() != null) {
				p.getQueryParameters().forEachParameter((n, v) -> q.parameter(n, v));
			}
		});

		return q;
	}

}