107 lines
5.5 KiB
Java
107 lines
5.5 KiB
Java
package eu.old.eudat.migration;
|
|
|
|
import eu.eudat.commons.JsonHandlingService;
|
|
import eu.eudat.commons.XmlHandlingService;
|
|
import eu.eudat.commons.enums.IsActive;
|
|
import eu.eudat.data.DescriptionTagEntity;
|
|
import eu.eudat.data.TagEntity;
|
|
import eu.old.eudat.data.dao.entities.DatasetDao;
|
|
import eu.old.eudat.data.entities.Dataset;
|
|
import eu.old.eudat.elastic.criteria.DatasetCriteria;
|
|
import eu.old.eudat.elastic.entities.Tag;
|
|
import eu.old.eudat.elastic.repository.DatasetRepository;
|
|
import eu.old.eudat.logic.services.operations.DatabaseRepository;
|
|
import gr.cite.tools.data.query.QueryFactory;
|
|
import gr.cite.tools.logging.LoggerService;
|
|
import jakarta.persistence.EntityManager;
|
|
import jakarta.xml.bind.JAXBException;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.stereotype.Service;
|
|
import org.xml.sax.SAXException;
|
|
|
|
import javax.xml.parsers.ParserConfigurationException;
|
|
import java.io.IOException;
|
|
import java.time.Instant;
|
|
import java.util.*;
|
|
|
|
@Service
|
|
public class TagMigrationService {
|
|
|
|
private static final LoggerService logger = new LoggerService(LoggerFactory.getLogger(TagMigrationService.class));
|
|
private final DatabaseRepository databaseRepository;
|
|
private final QueryFactory queryFactory;
|
|
private final DatasetRepository datasetRepository;
|
|
private static final int PageSize = 500;
|
|
private static final boolean TestMode = false;
|
|
private final EntityManager entityManager;
|
|
|
|
public TagMigrationService(DatabaseRepository databaseRepository, QueryFactory queryFactory, DatasetRepository datasetRepository, EntityManager entityManager) {
|
|
this.databaseRepository = databaseRepository;
|
|
this.queryFactory = queryFactory;
|
|
this.datasetRepository = datasetRepository;
|
|
this.entityManager = entityManager;
|
|
}
|
|
|
|
public void migrate() throws IOException, JAXBException, ParserConfigurationException, InstantiationException, IllegalAccessException, SAXException {
|
|
DatasetDao datasetDao = databaseRepository.getDatasetDao();
|
|
long total = datasetDao.asQueryable().count();
|
|
logger.debug("Migrate Tags for Dataset Total : " + total);
|
|
int page = 0;
|
|
|
|
Set<String> savedTagNames = new HashSet<>();
|
|
Map<String, UUID> savedTagIdsByName = new HashMap<>();
|
|
|
|
DatasetCriteria criteria = new DatasetCriteria();
|
|
List<Dataset> items;
|
|
do {
|
|
items = datasetDao.asQueryable().orderBy((builder, root) -> builder.asc(root.get("created"))).orderBy((builder, root) -> builder.asc(root.get("ID"))).skip(page * PageSize).take(PageSize).toList();
|
|
if (items != null && !items.isEmpty()) {
|
|
logger.debug("Migrate Dataset tags " + page * PageSize + " of " + total);
|
|
|
|
criteria.setIds(items.stream().map(Dataset::getId).toList());
|
|
// List<eu.old.eudat.elastic.entities.Dataset> elasticDatasets = this.datasetRepository.queryIds(criteria);
|
|
List<eu.old.eudat.elastic.entities.Dataset> elasticDatasets = this.datasetRepository.findByIds(items.stream().map(x -> x.getId().toString()).toList());
|
|
|
|
for (Dataset item : items) {
|
|
List<eu.old.eudat.elastic.entities.Dataset> found = elasticDatasets.stream().filter(x -> item.getId().toString().equals(x.getId())).toList();
|
|
if (found.isEmpty()) {
|
|
logger.error("No dataset with id {} found on elastic search. Skipping tag migration for this dataset", item.getId());
|
|
continue;
|
|
}
|
|
eu.old.eudat.elastic.entities.Dataset elasticDataset = found.getFirst();
|
|
boolean tagAlreadyExists;
|
|
if (elasticDataset.getTags() != null && !elasticDataset.getTags().isEmpty()) {
|
|
for(Tag tag : elasticDataset.getTags()) {
|
|
tagAlreadyExists = savedTagNames.contains(tag.getName());
|
|
savedTagNames.add(tag.getName());
|
|
if (!tagAlreadyExists) {
|
|
TagEntity tagEntity = new TagEntity();
|
|
tagEntity.setId(UUID.fromString(tag.getId()));
|
|
tagEntity.setLabel(tag.getName());
|
|
tagEntity.setCreatedAt(Instant.now());
|
|
tagEntity.setUpdatedAt(Instant.now());
|
|
tagEntity.setIsActive(IsActive.Active);
|
|
savedTagIdsByName.put(tag.getName(), tagEntity.getId());
|
|
this.entityManager.persist(tagEntity);
|
|
logger.debug("New tag saved '{}'", tag.getName());
|
|
}
|
|
DescriptionTagEntity descriptionTagEntity = new DescriptionTagEntity();
|
|
descriptionTagEntity.setId(UUID.randomUUID());
|
|
descriptionTagEntity.setTagId(savedTagIdsByName.get(tag.getName()));
|
|
descriptionTagEntity.setDescriptionId(item.getId());
|
|
descriptionTagEntity.setCreatedAt(Instant.now());
|
|
descriptionTagEntity.setUpdatedAt(Instant.now());
|
|
descriptionTagEntity.setIsActive(IsActive.Active);
|
|
this.entityManager.persist(descriptionTagEntity);
|
|
}
|
|
}
|
|
this.entityManager.flush();
|
|
}
|
|
|
|
page++;
|
|
}
|
|
} while (items != null && !items.isEmpty() && !TestMode);
|
|
}
|
|
|
|
}
|