CropTraitObservationServiceImpl.java

/*
 * Copyright 2021 Global Crop Diversity Trust
 *
 * 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 org.gringlobal.service.impl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.gringlobal.api.exception.InvalidApiUsageException;
import org.gringlobal.api.v1.impl.CropTraitObservationController;
import org.gringlobal.custom.elasticsearch.SearchException;
import org.gringlobal.model.CropTrait;
import org.gringlobal.model.CropTraitObservation;
import org.gringlobal.model.Inventory;
import org.gringlobal.model.Method;
import org.gringlobal.model.QCropTrait;
import org.gringlobal.model.QCropTraitObservation;
import org.gringlobal.model.QInventory;
import org.gringlobal.persistence.CropTraitObservationRepository;
import org.gringlobal.persistence.CropTraitRepository;
import org.gringlobal.persistence.InventoryRepository;
import org.gringlobal.persistence.MethodRepository;
import org.gringlobal.service.CropTraitObservationService;
import org.gringlobal.service.CropTraitService;
import org.gringlobal.service.CropTraitTranslationService.TranslatedCropTrait;
import org.gringlobal.service.filter.CropTraitObservationFilter;
import org.hibernate.Hibernate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.util.Pair;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;

import com.querydsl.core.BooleanBuilder;
import com.querydsl.jpa.impl.JPAQuery;


@Service
@Validated
@Transactional(readOnly = true)
@Slf4j
public class CropTraitObservationServiceImpl extends FilteredCRUDServiceImpl<CropTraitObservation, CropTraitObservationFilter, CropTraitObservationRepository> implements CropTraitObservationService {

	@Autowired
	private CropTraitService cropTraitService;

	@Autowired
	private MethodRepository methodRepository;

	@Autowired
	private InventoryRepository inventoryRepository;

	@Autowired
	private CropTraitRepository cropTraitRepository;

	@Override
	protected JPAQuery<CropTraitObservation> entityListQuery() {
		return jpaQueryFactory.selectFrom(QCropTraitObservation.cropTraitObservation)
				// method
				.join(QCropTraitObservation.cropTraitObservation.method()).fetchJoin()
				// inventory
				.join(QCropTraitObservation.cropTraitObservation.inventory()).fetchJoin()
				// trait
				.join(QCropTraitObservation.cropTraitObservation.cropTrait()).fetchJoin()
				// trait
				.leftJoin(QCropTraitObservation.cropTraitObservation.cropTraitCode()).fetchJoin()
			;
	}

	@Override
	@Transactional
	public CropTraitObservation create(CropTraitObservation source) {
		log.debug("Create CropTraitObservation. Input data {}", source);
		CropTraitObservation observation = new CropTraitObservation();
		observation.apply(source);

		CropTraitObservation saved = repository.save(observation);
		return _lazyLoad(saved);
	}

	@Override
	@Transactional
	public CropTraitObservation update(CropTraitObservation input, CropTraitObservation target) {
		log.debug("Update CropTraitObservation. Input data {}", input);
		target.apply(input);

		CropTraitObservation saved = repository.save(target);
		return _lazyLoad(saved);
	}

	@Override
	public Page<CropTraitObservation> list(CropTraitObservationFilter filter, Pageable page) throws SearchException {
		var r = super.list(filter, page);
		r.getContent().forEach(cto -> cto.lazyLoad());
		return r;
	}

	@Override
	public FilteredObservations search(CropTraitObservationFilter filter, Pageable page) {
		if (filter == null || CollectionUtils.isEmpty(filter.cropTraitId)) {
			throw new InvalidApiUsageException("List of traits is required");
		}

		BooleanBuilder predicate = new BooleanBuilder();
		predicate.and(filter.buildPredicate());

		Page<CropTraitObservation> resultPage = repository.findAll(predicate, page);

		Set<Method> methods = new HashSet<>();
		Set<TranslatedCropTrait> cropTraits = new HashSet<>();
		filter.cropTraitId.forEach((cropTraitId) -> cropTraits.add(cropTraitService.loadTranslated(cropTraitId)));

		Map<Inventory, Set<CropTraitObservation>> observations = new HashMap<>();

		resultPage.forEach(o -> {
			// initialize lazy data
			Hibernate.initialize(o.getMethod());
			Hibernate.initialize(o.getInventory());

			methods.add(o.getMethod());

			Set<CropTraitObservation> cto = observations.getOrDefault(o.getInventory(), new HashSet<>());
			cto.add(o);
			observations.put(o.getInventory(), cto);
		});

		FilteredObservations result = new FilteredObservations();
		result.filter = filter;
		result.methods = methods;
		result.cropTraits = cropTraits;
		result.observations = new HashSet<>(observations.size());

		observations.keySet().forEach(i -> result.observations.add(new InventoryWithObservations(i, observations.get(i))));

		return result;
	}

	@Override
	public Page<Inventory> getObservationInventoriesByMethod(Long methodId, Pageable pageable) {
		var method = methodRepository.getReferenceById(methodId);
		var observationPath = QCropTraitObservation.cropTraitObservation;
		var inventoryIds = jpaQueryFactory.select(observationPath.inventory().id).distinct().from(observationPath)
			.where(observationPath.method().id.eq(method.getId()).and(observationPath.inventory().isNotNull()))
			.fetch();
		return inventoryRepository.findAll(QInventory.inventory.id.in(inventoryIds), pageable);
	}

	@Override
	public Page<CropTrait> getObservationTraitsByMethod(Long methodId, Pageable pageable) {
		var method = methodRepository.getReferenceById(methodId);
		var observationPath = QCropTraitObservation.cropTraitObservation;
		var cropTraitIds = jpaQueryFactory.select(observationPath.cropTrait().id).distinct().from(observationPath)
			.where(observationPath.method().id.eq(method.getId()).and(observationPath.cropTrait().isNotNull()))
			.fetch();
		return cropTraitRepository.findAll(QCropTrait.cropTrait.id.in(cropTraitIds), pageable);
	}

	@Override
	@Transactional
	public int ensureObservations(CropTraitObservationController.EnsureObservationsRequest request) {
		if (request.methodId == null || CollectionUtils.isEmpty(request.inventoryId) || CollectionUtils.isEmpty(request.cropTraitId)) {
			throw new InvalidApiUsageException("Method id, CropTrait ids and inventory ids must be provided");
		}
		var method = methodRepository.getReferenceById(request.methodId);
		var observationPath = QCropTraitObservation.cropTraitObservation;

		var observations = jpaQueryFactory.selectFrom(observationPath).distinct()
			.where(observationPath.method().id.in(method.getId())
			.and(observationPath.cropTrait().id.in(request.cropTraitId))
			.and(observationPath.inventory().id.in(request.inventoryId))).fetch();

		var idPairs = observations.stream()
			.map(obs -> Pair.of(obs.getCropTrait().getId(), obs.getInventory().getId())).collect(Collectors.toList());

		List<CropTraitObservation> observationsForSave = new ArrayList<>();
		for (var traitId: request.cropTraitId) {
			for (var inventoryId: request.inventoryId) {
				if (!idPairs.contains(Pair.of(traitId, inventoryId))) {
					CropTraitObservation observation = new CropTraitObservation();
					observation.setCropTrait(cropTraitRepository.getReferenceById(traitId));
					observation.setInventory(inventoryRepository.getReferenceById(inventoryId));
					observation.setMethod(method);
					observationsForSave.add(observation);
				}
			}
		}

		var saved = repository.saveAll(observationsForSave);
		return saved.size();
	}

}