NominalLabelResourceFacadeImp.java

/*
 * Implements operations for managing the nominal_label table of the aidr_predict DB
 * 
 * @author Koushik
 */
package qa.qcri.aidr.dbmanager.ejb.remote.facade.imp;

import java.util.ArrayList;
import java.util.List;

import javax.ejb.Stateless;

import org.apache.log4j.Logger;
import org.hibernate.Hibernate;
import org.hibernate.Query;
import org.hibernate.Session;
import org.hibernate.criterion.Restrictions;

import qa.qcri.aidr.common.exception.PropertyNotSetException;
import qa.qcri.aidr.dbmanager.dto.NominalLabelDTO;
import qa.qcri.aidr.dbmanager.ejb.local.facade.impl.CoreDBServiceFacadeImp;
import qa.qcri.aidr.dbmanager.ejb.remote.facade.NominalLabelResourceFacade;
import qa.qcri.aidr.dbmanager.entities.model.NominalLabel;


@Stateless(name="NominalLabelResourceFacadeImp")
public class NominalLabelResourceFacadeImp extends CoreDBServiceFacadeImp<NominalLabel, Long> implements NominalLabelResourceFacade {

	private Logger logger = Logger.getLogger("db-manager-log");

	protected NominalLabelResourceFacadeImp(){
		super(NominalLabel.class);
	}

	@Override
	public void saveNominalLabel(NominalLabelDTO nominalLabel) throws PropertyNotSetException {
		save(nominalLabel.toEntity());
	}

	@Override
	public NominalLabelDTO addNominalLabel(NominalLabelDTO nominalLabel) {
		try {
			NominalLabel nb = nominalLabel.toEntity();
			em.persist(nb);
			em.flush();
			em.refresh(nb);
			return new NominalLabelDTO(nb);
		} catch (Exception e) {
			logger.error("Error in addNominalLabel.", e);
			return null;
		}
	}

	@Override
	public NominalLabelDTO editNominalLabel(NominalLabelDTO nominalLabel) throws PropertyNotSetException {
		try {
			NominalLabel label = nominalLabel.toEntity();
			NominalLabel oldLabel = getById(label.getNominalLabelId());
			
			if (oldLabel != null) {
				label.setSequence(oldLabel.getSequence());
				oldLabel = em.merge(label);
				em.flush();
				em.refresh(oldLabel);
				return (oldLabel != null) ? new NominalLabelDTO(oldLabel) : null;
			} else {
				throw new RuntimeException("Not found");
			}
		} catch (Exception e) {
			logger.error("Exception in merging/updating nominalLabel: " + nominalLabel.getNominalLabelId(), e);
		}
		return null;
	}

	@Override
	public Integer deleteNominalLabel(NominalLabelDTO nominalLabel) throws PropertyNotSetException {
		if (nominalLabel != null) {
			NominalLabel managed = em.merge(nominalLabel.toEntity());
			em.remove(managed);
			return 1;
		}
		return 0;
	}

	@Override
	public Integer deleteNominalLabelByID(Long nominalLabelID) {
		NominalLabel nb = this.getById(nominalLabelID);
		if (nb != null) {
			em.remove(nb);
			return 1;
		} 
		return 0;
	}

	@Override
	public NominalLabelDTO getNominalLabelByID(Long nominalLabelID) throws PropertyNotSetException {
		NominalLabel nb = this.getById(nominalLabelID);
		return nb != null ? new NominalLabelDTO(nb) : null;
	}

	@Override
	public NominalLabelDTO getNominalLabelWithAllFieldsByID(Long nominalLabelID) throws PropertyNotSetException {
		NominalLabel nb = this.getById(nominalLabelID);
		if (nb != null) {
			Hibernate.initialize(nb.getModelNominalLabels());
			Hibernate.initialize(nb.getNominalAttribute());
			Hibernate.initialize(nb.getDocumentNominalLabels());
			return new NominalLabelDTO(nb);
		} 
		return null;
	}

	@Override
	public NominalLabelDTO getNominalLabelByCode(String code) throws PropertyNotSetException {
		NominalLabel nb = this.getByCriteria(Restrictions.eq("nominalLabelCode", code));
		return nb != null ? new NominalLabelDTO(nb) : null;
	}

	@Override
	public NominalLabelDTO getNominalLabelWithAllFieldsByCode(String code) throws PropertyNotSetException {
		NominalLabel nb = this.getByCriteria(Restrictions.eq("nominalLabelCode", code));
		if (nb != null) {
			Hibernate.initialize(nb.getModelNominalLabels());
			Hibernate.initialize(nb.getNominalAttribute());
			Hibernate.initialize(nb.getDocumentNominalLabels());
			return new NominalLabelDTO(nb);
		} 
		return null;
	}

	@Override
	public List<NominalLabelDTO> getAllNominalLabels() throws PropertyNotSetException {
		List<NominalLabelDTO> dtoList = new ArrayList<NominalLabelDTO>();
		List<NominalLabel> list = this.getAll();
		if (list != null && !list.isEmpty()) {
			for (NominalLabel nb: list) {
				dtoList.add(new NominalLabelDTO(nb));
			}
		}
		return dtoList;
	}

	@Override
	public Boolean isNominalLabelExists(Long nominalLabelID) {
		NominalLabel nb = this.getById(nominalLabelID);
		return nb != null ? true : false;
	}

	@Override
	public Boolean isNominalLabelExists(String code) {
		NominalLabel nb = this.getByCriteria(Restrictions.eq("nominalLabelCode", code));
		return nb != null ? true : false;
	}
	
	@Override
	public List<NominalLabelDTO> getNominalLabelByAttributeID(Long attributeID) throws PropertyNotSetException {
		List<NominalLabel> nominalLabels = this.getAllByCriteria(Restrictions.eq("nominalAttribute.nominalAttributeId", attributeID));
		List<NominalLabelDTO> dtoList = new ArrayList<NominalLabelDTO>();
		if (nominalLabels != null && !nominalLabels.isEmpty()) {
			for (NominalLabel nb: nominalLabels) {
				dtoList.add(new NominalLabelDTO(nb));
			}
		}
		
		return dtoList;
	}
	
	@Override
	public List<Long> getNominalLabelIdsByAttributeID(Long attributeID) throws Exception {
		List<NominalLabel> nominalLabels = this.getAllByCriteria(Restrictions.eq("nominalAttribute.nominalAttributeId", attributeID));
		List<Long> idList = new ArrayList<Long>();
		if (nominalLabels != null && !nominalLabels.isEmpty()) {
			for (NominalLabel nb: nominalLabels) {
				idList.add(nb.getNominalLabelId());
			}
		}
		
		return idList;
	}

	@Override
	public Long getNominalLabelCountForCollectionID(Long collectionId) throws Exception {
		Session session = getCurrentSession();
		Long count = 0L;
		String queryText = "select count(1) from NominalLabel label, ModelFamily fam where fam.collection.id = :collectionId and fam.nominalAttribute.nominalAttributeId = label.nominalAttribute.nominalAttributeId";
		try {
			Query query = session.createQuery(queryText);
			query.setParameter("collectionId", collectionId);
			count = (Long) query.uniqueResult();
		} catch (Exception e) {
			logger.error("Error in fetching count for collection : " + collectionId, e);
			throw e;
		}
		return count;

	}
}