64 lines
2.7 KiB
Java
64 lines
2.7 KiB
Java
package eu.eudat.dao.entities;
|
|
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
import java.util.UUID;
|
|
import java.util.stream.Collectors;
|
|
|
|
import javax.persistence.Query;
|
|
import javax.persistence.TypedQuery;
|
|
import javax.persistence.criteria.CriteriaBuilder;
|
|
import javax.persistence.criteria.CriteriaQuery;
|
|
import javax.persistence.criteria.Root;
|
|
|
|
import eu.eudat.dao.JpaDao;
|
|
import eu.eudat.dao.databaselayer.service.DatabaseService;
|
|
import eu.eudat.entities.Dataset;
|
|
import eu.eudat.entities.Project;
|
|
import eu.eudat.entities.UserInfo;
|
|
import eu.eudat.entities.responses.IDLabelPair;
|
|
import eu.eudat.models.criteria.DatasetCriteria;
|
|
import eu.eudat.models.dataset.DatasetTableRequest;
|
|
import eu.eudat.models.security.Principal;
|
|
import eu.eudat.queryable.QueryableList;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
@Component("datasetDao")
|
|
public class DatasetDaoImpl implements DatasetDao {
|
|
|
|
@Autowired
|
|
DatabaseService<Dataset> databaseService;
|
|
|
|
@Override
|
|
public QueryableList<Dataset> getWithCriteria(DatasetCriteria criteria) {
|
|
QueryableList<Dataset> query = databaseService.getQueryable(Dataset.class, Dataset.getHints());
|
|
if (criteria.getLike() != null && !criteria.getLike().isEmpty())query.where((builder, root) -> builder.like(root.get("label"), "%" + criteria.getLike() + "%"));
|
|
if (criteria.getStatus() != null)query.where((builder, root) -> builder.equal(root.get("status"), criteria.getStatus()));
|
|
if (criteria.getPeriodEnd() != null)query.where((builder, root) -> builder.lessThan(root.get("created"), criteria.getPeriodEnd()));
|
|
if (criteria.getPeriodStart() != null) query.where((builder, root) -> builder.greaterThan(root.get("created"), criteria.getPeriodStart()));
|
|
if (criteria.getDmpIds() != null && !criteria.getDmpIds().isEmpty()) query.where((builder, root) -> root.get("dmp").get("id").in( criteria.getDmpIds()));
|
|
return query;
|
|
}
|
|
|
|
@Override
|
|
public Dataset createOrUpdate(Dataset item) {
|
|
return databaseService.createOrUpdate(item, Dataset.class);
|
|
}
|
|
|
|
@Override
|
|
public Dataset find(UUID id) {
|
|
return databaseService.getQueryable(Dataset.class).where((builder, root) -> builder.equal((root.get("id")), id)).toList().get(0);
|
|
}
|
|
|
|
public Long count() {
|
|
return this.databaseService.count(Dataset.class);
|
|
}
|
|
|
|
@Override
|
|
public QueryableList<Dataset> getAuthenticated(QueryableList<Dataset> query, UserInfo principal) {
|
|
query.where((builder, root) -> builder.or(builder.equal(root.get("dmp").get("creator"), principal), builder.isMember(principal, root.get("dmp").get("users"))));
|
|
return query;
|
|
}
|
|
}
|