argos/dmp-backend/src/main/java/eu/eudat/dao/entities/DatasetDaoImpl.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.responses.IDLabelPair;
import eu.eudat.models.criteria.DatasetCriteria;
import eu.eudat.models.dataset.DatasetTableRequest;
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);
		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()));
		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);
	}
}