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

Example 1 with PreJobCommand

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

the class PreJobCommandResource method isExists.

public boolean isExists(Object identifier) throws AppCatalogException {
    HashMap<String, String> ids;
    if (identifier instanceof Map) {
        ids = (HashMap) 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();
        PreJobCommand preJobCommand = em.find(PreJobCommand.class, new PreJobCommandPK(ids.get(PreJobCommandConstants.DEPLOYMENT_ID), ids.get(PreJobCommandConstants.COMMAND)));
        if (em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
        return preJobCommand != 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 PreJobCommand

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

the class PreJobCommandResource method getIds.

public List<String> getIds(String fieldName, Object value) throws AppCatalogException {
    List<String> gsiSSHPreJobResourceIDs = new ArrayList<String>();
    EntityManager em = null;
    try {
        em = AppCatalogJPAUtils.getEntityManager();
        em.getTransaction().begin();
        Query q;
        AppCatalogQueryGenerator generator = new AppCatalogQueryGenerator(PRE_JOBCOMMAND);
        List results;
        if (fieldName.equals(PreJobCommandConstants.DEPLOYMENT_ID)) {
            generator.setParameter(PreJobCommandConstants.DEPLOYMENT_ID, value);
            q = generator.selectQuery(em);
            results = q.getResultList();
            if (results.size() != 0) {
                for (Object result : results) {
                    PreJobCommand preJobCommand = (PreJobCommand) result;
                    gsiSSHPreJobResourceIDs.add(preJobCommand.getDeploymentId());
                }
            }
        } else if (fieldName.equals(PreJobCommandConstants.COMMAND)) {
            generator.setParameter(PreJobCommandConstants.COMMAND, value);
            q = generator.selectQuery(em);
            results = q.getResultList();
            if (results.size() != 0) {
                for (Object result : results) {
                    PreJobCommand preJobCommand = (PreJobCommand) result;
                    gsiSSHPreJobResourceIDs.add(preJobCommand.getDeploymentId());
                }
            }
        } else {
            em.getTransaction().commit();
            if (em.isOpen()) {
                if (em.getTransaction().isActive()) {
                    em.getTransaction().rollback();
                }
                em.close();
            }
            logger.error("Unsupported field name for GSISSH Pre Job resource.", new IllegalArgumentException());
            throw new IllegalArgumentException("Unsupported field name for GSISSH Pre JOb Resource.");
        }
        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();
        }
    }
    return gsiSSHPreJobResourceIDs;
}

Example 3 with PreJobCommand

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

the class PreJobCommandResource method save.

public void save() throws AppCatalogException {
    EntityManager em = null;
    try {
        em = AppCatalogJPAUtils.getEntityManager();
        PreJobCommand existingPreJobCommand = em.find(PreJobCommand.class, new PreJobCommandPK(appDeploymentId, command));
        if (em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
        em = AppCatalogJPAUtils.getEntityManager();
        em.getTransaction().begin();
        ApplicationDeployment deployment = em.find(ApplicationDeployment.class, appDeploymentId);
        if (existingPreJobCommand != null) {
            existingPreJobCommand.setDeploymentId(appDeploymentId);
            existingPreJobCommand.setCommand(command);
            existingPreJobCommand.setApplicationDeployment(deployment);
            existingPreJobCommand.setOrder(order);
            em.merge(existingPreJobCommand);
        } else {
            PreJobCommand preJobCommand = new PreJobCommand();
            preJobCommand.setDeploymentId(appDeploymentId);
            preJobCommand.setCommand(command);
            preJobCommand.setOrder(order);
            preJobCommand.setApplicationDeployment(deployment);
            em.persist(preJobCommand);
        }
        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();
        }
    }
}

Example 4 with PreJobCommand

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

the class PreJobCommandResource method get.

public AppCatalogResource get(Object identifier) throws AppCatalogException {
    HashMap<String, String> ids;
    if (identifier instanceof Map) {
        ids = (HashMap) 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();
        em.getTransaction().begin();
        AppCatalogQueryGenerator generator = new AppCatalogQueryGenerator(PRE_JOBCOMMAND);
        generator.setParameter(PreJobCommandConstants.DEPLOYMENT_ID, ids.get(PreJobCommandConstants.DEPLOYMENT_ID));
        generator.setParameter(PreJobCommandConstants.COMMAND, ids.get(PreJobCommandConstants.COMMAND));
        Query q = generator.selectQuery(em);
        PreJobCommand preJobCommand = (PreJobCommand) q.getSingleResult();
        PreJobCommandResource preJobCommandResource = (PreJobCommandResource) AppCatalogJPAUtils.getResource(AppCatalogResourceType.PRE_JOBCOMMAND, preJobCommand);
        em.getTransaction().commit();
        if (em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
        return preJobCommandResource;
    } 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 5 with PreJobCommand

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

the class PreJobCommandResource method get.

public List<AppCatalogResource> get(String fieldName, Object value) throws AppCatalogException {
    List<AppCatalogResource> preJobCommandResources = new ArrayList<AppCatalogResource>();
    EntityManager em = null;
    try {
        em = AppCatalogJPAUtils.getEntityManager();
        em.getTransaction().begin();
        Query q;
        AppCatalogQueryGenerator generator = new AppCatalogQueryGenerator(PRE_JOBCOMMAND);
        List results;
        if (fieldName.equals(PreJobCommandConstants.DEPLOYMENT_ID) || fieldName.equals(PreJobCommandConstants.COMMAND)) {
            generator.setParameter(fieldName, value);
            q = generator.selectQuery(em);
            results = q.getResultList();
            if (results.size() != 0) {
                for (Object result : results) {
                    PreJobCommand preJobCommand = (PreJobCommand) result;
                    PreJobCommandResource preJobCommandResource = (PreJobCommandResource) AppCatalogJPAUtils.getResource(AppCatalogResourceType.PRE_JOBCOMMAND, preJobCommand);
                    preJobCommandResources.add(preJobCommandResource);
                }
                Collections.sort(preJobCommandResources, (o1, o2) -> ((PreJobCommandResource) o1).getOrder() - ((PreJobCommandResource) o2).getOrder());
            }
        } else {
            em.getTransaction().commit();
            if (em.isOpen()) {
                if (em.getTransaction().isActive()) {
                    em.getTransaction().rollback();
                }
                em.close();
            }
            logger.error("Unsupported field name for Pre Job Command Resource.", new IllegalArgumentException());
            throw new IllegalArgumentException("Unsupported field name for Pre Job Command Resource.");
        }
        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();
        }
    }
    return preJobCommandResources;
}