/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package qa.qcri.aidr.predictui.facade.imp;

import java.util.List;

import javax.ejb.EJB;
import javax.ejb.Stateless;

import org.apache.log4j.Logger;

import qa.qcri.aidr.common.exception.PropertyNotSetException;
import qa.qcri.aidr.dbmanager.dto.NominalLabelDTO;
import qa.qcri.aidr.dbmanager.ejb.remote.facade.DocumentNominalLabelResourceFacade;
import qa.qcri.aidr.predictui.facade.NominalLabelResourceFacade;

/**
 *
 * @author Imran
 * updated by Koushik
 */
@Stateless
public class NominalLabelResourceImp implements NominalLabelResourceFacade {

	private Logger logger = Logger.getLogger(NominalLabelResourceImp.class);
	
    @EJB
	private qa.qcri.aidr.dbmanager.ejb.remote.facade.NominalLabelResourceFacade remoteNominalLabelEJB;
    
    @EJB
    private DocumentNominalLabelResourceFacade remoteDocumentNominalLabelEJB;

    @Override
    public NominalLabelDTO addNominalLabel(NominalLabelDTO label) {
        try {
			return remoteNominalLabelEJB.addNominalLabel(label);
		} catch (Exception e) {
			logger.error("Error in addNominalLabel.", e);
			return null;
		}
    }

    @Override
    public NominalLabelDTO getNominalLabelByID(Long id) {
        try {
			return remoteNominalLabelEJB.getNominalLabelByID(id);
		} catch (PropertyNotSetException e) {
			logger.error("Error in getNominalLabelByID for id : " + id, e);
			return null;
		}
    }

    @Override
    public NominalLabelDTO editNominalLabel(NominalLabelDTO label) {
        try {
			return remoteNominalLabelEJB.editNominalLabel(label);
		} catch (PropertyNotSetException e) {
			logger.error("Error in editNominalLabel for label code : " + label.getNominalLabelCode(), e  );
			return null;
		}
    }

    @Override
    public List<NominalLabelDTO> getAllNominalLabel() {
        try {
			return remoteNominalLabelEJB.getAllNominalLabels();
		} catch (PropertyNotSetException e) {
			logger.error("Error in getAllNominalLabel.", e);
			return null;
		}
    }

    @Override
    public void deleteNominalLabel(Long labelID) {
        try {
			remoteNominalLabelEJB.deleteNominalLabelByID(labelID);
		} catch (PropertyNotSetException e) {
			logger.error("Error in deleteNominalLabel by ID : " + labelID, e);
		}
    }

	@Override
	public void deleteNominalLabelDataByAttribute(Long attributeID) {
		// TODO Auto-generated method stub
		
		 try {
			List<NominalLabelDTO> labelDTOs = remoteNominalLabelEJB.getNominalLabelByAttributeID(attributeID);
			for (NominalLabelDTO nominalLabelDTO : labelDTOs) {
				remoteDocumentNominalLabelEJB.deleteDocumentNominalLabelByNominalLabel(nominalLabelDTO.getNominalLabelId());
				deleteNominalLabel(nominalLabelDTO.getNominalLabelId());
			}
		 }catch (Exception e) {
			 logger.error("Error in deleteNominalLabelDataByAttribute for attributeID : " + attributeID, e);
		 }
	}

	@Override
	public Long getLabelCountByCollectionId(Long collectionId) {
		Long labelCount = 0L;
		try {
			labelCount = remoteNominalLabelEJB.getNominalLabelCountForCollectionID(collectionId);
		} catch(Exception e) {
			logger.error("Error in fetch label count.", e);
		}
		return labelCount;
	}
}