argos/dmp-backend/src/main/java/dao/entities/DatasetDaoImpl.java

package dao.entities;

import java.util.List;
import java.util.UUID;
import java.util.stream.Collectors;

import javax.persistence.TypedQuery;
import javax.transaction.Transactional;

import org.hibernate.query.Query;

import dao.JpaDao;
import entities.Dataset;
import entities.responses.IDLabelPair;

public class DatasetDaoImpl extends JpaDao<Dataset, UUID> implements DatasetDao {

	public Dataset loadDetails(Dataset t) {
		// TODO Auto-generated method stub
		return null;
	}
	
	@Override
	public List<UUID> listAllIDs() {
		String queryString = "SELECT dataset.id FROM Dataset dataset";
		TypedQuery<UUID> typedQuery = entityManager.createQuery(queryString, UUID.class);
		return typedQuery.getResultList();
	}
	
	
	@Override
	public List<IDLabelPair> listAllIDsLabels() {
		String queryString = "SELECT dataset.id, dataset.label FROM Dataset dataset";
		Query query = (Query) entityManager.createQuery(queryString);
		List<Object[]> rows = query.list();
		return rows.stream().map(row -> {
			return new IDLabelPair(row[0].toString(), row[1].toString());
		})
		.collect(Collectors.toList());
	}
	
}