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();
}
}