Examples with BatchQueue_PK org.apache.airavata.registry.core.app.catalog.model.BatchQueue_PK used on opensource projects

Example 1 with BatchQueue_PK

use of org.apache.airavata.registry.core.app.catalog.model.BatchQueue_PK in project airavata by apache.

the class BatchQueueResource method isExists.

@Override
public boolean isExists(Object identifier) throws AppCatalogException {
    HashMap<String, String> ids;
    if (identifier instanceof Map) {
        ids = (HashMap<String, String>) identifier;
    } else {
        logger.error("Identifier should be a map with the field name and it's value");
        throw new AppCatalogException("Identifier should be a map with the field name and it's value");
    }
    EntityManager em = null;
    try {
        em = AppCatalogJPAUtils.getEntityManager();
        BatchQueue batchQueue = em.find(BatchQueue.class, new BatchQueue_PK(ids.get(BatchQueueConstants.COMPUTE_RESOURCE_ID), ids.get(BatchQueueConstants.QUEUE_NAME)));
        if (em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
        return batchQueue != null;
    } catch (ApplicationSettingsException e) {
        logger.error(e.getMessage(), e);
        throw new AppCatalogException(e);
    } finally {
        if (em != null && em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
    }
}

Example 2 with BatchQueue_PK

use of org.apache.airavata.registry.core.app.catalog.model.BatchQueue_PK in project airavata by apache.

the class BatchQueueResource method save.

@Override
public void save() throws AppCatalogException {
    EntityManager em = null;
    try {
        em = AppCatalogJPAUtils.getEntityManager();
        BatchQueue existingBatchQueue = em.find(BatchQueue.class, new BatchQueue_PK(computeResourceId, queueName));
        if (em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
        BatchQueue batchQueue;
        em = AppCatalogJPAUtils.getEntityManager();
        em.getTransaction().begin();
        if (existingBatchQueue == null) {
            batchQueue = new BatchQueue();
        } else {
            batchQueue = existingBatchQueue;
        }
        batchQueue.setComputeResourceId(getComputeResourceId());
        ComputeResource computeResource = em.find(ComputeResource.class, getComputeResourceId());
        batchQueue.setComputeResource(computeResource);
        batchQueue.setMaxRuntime(getMaxRuntime());
        batchQueue.setMaxJobInQueue(getMaxJobInQueue());
        batchQueue.setQueueDescription(getQueueDescription());
        batchQueue.setQueueName(getQueueName());
        batchQueue.setMaxProcessors(getMaxProcessors());
        batchQueue.setMaxNodes(getMaxNodes());
        batchQueue.setMaxMemory(getMaxMemory());
        batchQueue.setCpuPerNode(getCpuPerNode());
        batchQueue.setDefaultCPUCount(getDefaultCPUCount());
        batchQueue.setDefaultNodeCount(getDefaultNodeCount());
        batchQueue.setDefaultWalltime(getDefaultWalltime());
        batchQueue.setQueueSpecificMacros(getQueueSpecificMacros());
        batchQueue.setIsDefaultQueue(isDefaultQueue());
        if (existingBatchQueue == null) {
            em.persist(batchQueue);
        } else {
            em.merge(batchQueue);
        }
        em.getTransaction().commit();
        if (em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
    } catch (Exception e) {
        logger.error(e.getMessage(), e);
        throw new AppCatalogException(e);
    } finally {
        if (em != null && em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
    }
}