io.crnk.jpa.internal.JpaRepositoryBase Maven / Gradle / Ivy

Go to download
package io.crnk.jpa.internal;

import java.util.List;

import io.crnk.core.exception.ResourceNotFoundException;
import io.crnk.core.queryspec.QuerySpec;
import io.crnk.core.resource.list.ResourceList;
import io.crnk.core.resource.meta.HasMoreResourcesMetaInformation;
import io.crnk.core.resource.meta.PagedMetaInformation;
import io.crnk.jpa.JpaModule;
import io.crnk.jpa.JpaRepositoryConfig;
import io.crnk.jpa.JpaRepositoryFilter;
import io.crnk.jpa.mapping.JpaMapper;
import io.crnk.jpa.query.JpaQuery;
import io.crnk.jpa.query.JpaQueryExecutor;
import io.crnk.jpa.query.Tuple;

public abstract class JpaRepositoryBase {

	protected JpaModule module;

	protected JpaRepositoryConfig repositoryConfig;

	protected  JpaRepositoryBase(JpaModule module, JpaRepositoryConfig repositoryConfig) {
		this.module = module;
		this.repositoryConfig = repositoryConfig;
	}

	protected boolean isNextFetched(QuerySpec querySpec) {
		return querySpec.getLimit() != null && !module.isTotalResourceCountUsed()
				&& repositoryConfig.getListMetaClass() != null
				&& HasMoreResourcesMetaInformation.class.isAssignableFrom(repositoryConfig
				.getListMetaClass());
	}

	protected boolean isTotalFetched(QuerySpec querySpec) {
		return querySpec.getLimit() != null && module.isTotalResourceCountUsed()
				&& repositoryConfig.getListMetaClass() != null
				&& PagedMetaInformation.class.isAssignableFrom(repositoryConfig.getListMetaClass());
	}

	protected static  D getUnique(List list, Object id) {
		if (list.isEmpty()) {
			throw new ResourceNotFoundException("resource not found: id=" + id);
		}
		else if (list.size() == 1) {
			return list.get(0);
		}
		else {
			throw new IllegalStateException("unique result expected");
		}
	}

	/**
	 * By default LookupIncludeBehavior.ALWAYS is in place and we let the relationship repositories load the relations. There
	 * is no need to do join fetches, which can lead to problems with paging (evaluated in memory instead of the db).
	 *
	 * @param fieldName of the relation to fetch
	 * @return relation will be eagerly fetched if true
	 */
	protected boolean fetchRelations(String fieldName) { // NOSONAR
		return false;
	}

	protected QuerySpec filterQuerySpec(QuerySpec querySpec) {
		QuerySpec filteredQuerySpec = querySpec;
		for (JpaRepositoryFilter filter : module.getFilters()) {
			if (filter.accept(repositoryConfig.getResourceClass())) {
				filteredQuerySpec = filter.filterQuerySpec(this, filteredQuerySpec);
			}
		}
		return filteredQuerySpec;
	}

	protected  JpaQuery filterQuery(QuerySpec querySpec, JpaQuery query) {
		JpaQuery filteredQuery = query;
		for (JpaRepositoryFilter filter : module.getFilters()) {
			if (filter.accept(repositoryConfig.getResourceClass())) {
				filteredQuery = filter.filterQuery(this, querySpec, filteredQuery);
			}
		}
		return filteredQuery;
	}

	protected  JpaQueryExecutor filterExecutor(QuerySpec querySpec, JpaQueryExecutor executor) {
		JpaQueryExecutor filteredExecutor = executor;
		for (JpaRepositoryFilter filter : module.getFilters()) {
			if (filter.accept(repositoryConfig.getResourceClass())) {
				filteredExecutor = filter.filterExecutor(this, querySpec, filteredExecutor);
			}
		}
		return filteredExecutor;
	}

	protected List filterTuples(QuerySpec querySpec, List tuples) {
		List filteredTuples = tuples;
		for (JpaRepositoryFilter filter : module.getFilters()) {
			if (filter.accept(repositoryConfig.getResourceClass())) {
				filteredTuples = filter.filterTuples(this, querySpec, filteredTuples);
			}
		}
		return filteredTuples;
	}

	protected ResourceList filterResults(QuerySpec querySpec, ResourceList resources) {
		ResourceList filteredResources = resources;
		for (JpaRepositoryFilter filter : module.getFilters()) {
			if (filter.accept(repositoryConfig.getResourceClass())) {
				filteredResources = filter.filterResults(this, querySpec, filteredResources);
			}
		}
		return filteredResources;
	}

	protected ResourceList fillResourceList(List tuples, ResourceList resources) {
		for (Tuple tuple : tuples) {
			JpaMapper mapper = repositoryConfig.getMapper();
			resources.add(mapper.map(tuple));
		}
		return resources;
	}
}