ModelFamilyResource.java

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

import java.util.Collections;
import java.util.List;

import javax.ejb.EJB;
import javax.ejb.Stateless;
import javax.ws.rs.Consumes;
import javax.ws.rs.DELETE;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriInfo;

import org.apache.log4j.Logger;

import qa.qcri.aidr.dbmanager.dto.ModelFamilyDTO;
import qa.qcri.aidr.dbmanager.dto.taggerapi.ModelWrapper;
import qa.qcri.aidr.dbmanager.dto.taggerapi.TaggersForCodes;
import qa.qcri.aidr.dbmanager.dto.taggerapi.TaggersForCodesRequest;
import qa.qcri.aidr.predictui.facade.ModelFamilyFacade;
import qa.qcri.aidr.predictui.util.ResponseWrapper;
import qa.qcri.aidr.predictui.util.TaggerAPIConfigurationProperty;
import qa.qcri.aidr.predictui.util.TaggerAPIConfigurator;

/**
 * REST Web Service
 *
 * @author Imran
 */
@Path("/modelfamily")
@Stateless
public class ModelFamilyResource {

    @Context
    private UriInfo context;
    @EJB
    private ModelFamilyFacade modelFamilyLocalEJB;

    //private static Logger logger = Logger.getLogger(ModelFamilyResource.class);
    private static Logger logger = Logger.getLogger("aidr-tagger-api");

    public ModelFamilyResource() {
    }

    @GET
    @Produces("application/json")
    @Path("/all")
    public Response getAllModelFamilies() {
        List<ModelFamilyDTO> modelFamilyList = modelFamilyLocalEJB.getAllModelFamilies();
        ResponseWrapper response = new ResponseWrapper();
        response.setMessage("SUCCESS");
        response.setModelFamilies(modelFamilyList);
        return Response.ok(response).build();
    }

    @GET
    @Produces(MediaType.APPLICATION_JSON)
    @Path("{id}")
    public Response getModelByID(@PathParam("id") Long id) {
        ModelFamilyDTO modelFamily = modelFamilyLocalEJB.getModelFamilyByID(id);
        if (modelFamily != null) {
        	return Response.ok(modelFamily).build();
        } else {
        	return Response.ok(new ModelFamilyDTO()).build();
        }
    }

    @POST
    @Consumes(MediaType.APPLICATION_JSON)
    @Produces(MediaType.APPLICATION_JSON)
    public Response addCrisisAttribute(ModelFamilyDTO modelFamilyDTO) {
        //      because ModelFamily has @XmlTransient annotation for crises and crisis was always null
        ResponseWrapper response = new ResponseWrapper();
        logger.info("Received request to add attribute = " + modelFamilyDTO.getNominalAttributeDTO().getNominalAttributeId() + " for crisis = " + modelFamilyDTO.getCrisisDTO().getCrisisID());
        try {
            boolean retval = modelFamilyLocalEJB.addCrisisAttribute(modelFamilyDTO);
            if (retval) {
                // TODO: move the following code to db-manager and return modelFamilyID directly from db-manager
            	List<ModelFamilyDTO> dtoList = modelFamilyLocalEJB.getAllModelFamiliesByCrisis(modelFamilyDTO.getCrisisDTO().getCrisisID());
                for (ModelFamilyDTO mf : dtoList) {
                    if (mf.getCrisisDTO().getCrisisID().equals(modelFamilyDTO.getCrisisDTO().getCrisisID())
                            && mf.getNominalAttributeDTO().getNominalAttributeId().equals(modelFamilyDTO.getNominalAttributeDTO().getNominalAttributeId())) {
                        response.setStatusCode(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_SUCCESS));
                        response.setMessage("Adding Attribute to crisis " + modelFamilyDTO.getCrisisDTO().getCrisisID() + " succeeded");
                        response.setEntityID(mf.getModelFamilyId());
                        return Response.ok(response).build();
                    }
                }
            }
            response.setStatusCode(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_FAILED));
            response.setMessage("Adding Attribute to crisis " + modelFamilyDTO.getCrisisDTO().getCrisisID() + " failed");
            response.setEntityID(new Long(-1));
            return Response.ok(response).build();
        } catch (RuntimeException e) {
            logger.error("Error while adding Crisis attribute. Possible causes could be duplication of primary key, incomplete data, incompatible data format: " + modelFamilyDTO.getCrisisDTO().getCode() + "," + modelFamilyDTO.getNominalAttributeDTO().getCode(), e);
            response.setStatusCode(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_FAILED));
            response.setMessage("Adding Attribute to crisis " + modelFamilyDTO.getCrisisDTO().getCrisisID() + " failed due to exception" + e);
            response.setEntityID(new Long(-1));
            return Response.ok(response).build();
        }
    }

    @POST
    @Consumes(MediaType.APPLICATION_JSON)
    @Produces(MediaType.APPLICATION_JSON)
    @Path("/taggers-by-codes")
    public Response getTaggersByCodes(TaggersForCodesRequest codesRequest) {
        List<TaggersForCodes> taggersByCodes = null;
        if(!codesRequest.getCodes().isEmpty()){
        	taggersByCodes = modelFamilyLocalEJB.getTaggersByCodes(codesRequest.getCodes());
        }
        if (taggersByCodes.isEmpty() || taggersByCodes == null) {
            taggersByCodes = Collections.emptyList();
        }
        ResponseWrapper response = new ResponseWrapper();
        response.setTaggersForCodes(taggersByCodes);
        return Response.ok(response).build();
    }

    @DELETE
    @Path("{id}")
    @Produces(MediaType.APPLICATION_JSON)
    public Response deleteAttribute(@PathParam("id") Long modelFamilyID) {
    	
    	boolean success = Boolean.FALSE;
		ResponseWrapper responseWrapper;
        try {
        	success = modelFamilyLocalEJB.deleteModelFamilyData(modelFamilyID);
        } catch (RuntimeException e) {
            logger.error("Error while deleting Classifier for modelFamily: " + modelFamilyID);
        }
        
        if(success) {
			responseWrapper = new ResponseWrapper(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_SUCCESS));	
		} else {
			responseWrapper = new ResponseWrapper(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_FAILED), "Error while deleting Classifier.");
		}
        
		return Response.ok(responseWrapper).build();
    }
    
    @GET
    @Produces("application/json")
    @Path("/crisis/{id}")
    public Response getModelFamilyAggregateData(@PathParam("id") Long crisisID) {
        ResponseWrapper response = new ResponseWrapper();
        List<ModelWrapper> modelFamilyList = modelFamilyLocalEJB.getModelFamilyAggregateDataForCollection(crisisID);
        if (modelFamilyList == null){
            response.setMessage("No models found for the given crisis-id");
            return Response.ok(response).build();
        }
        response.setModelWrapper(modelFamilyList);
        return Response.ok(response).build();
    }
}