Search in sources :

Example 6 with UserResourceProfile

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

the class UserStoragePreferenceResource method save.

@Override
public void save() throws AppCatalogException {
    EntityManager em = null;
    try {
        em = AppCatalogJPAUtils.getEntityManager();
        UserStoragePreference existingPreference = em.find(UserStoragePreference.class, new UserStoragePreferencePK(userId, gatewayID, storageResourceId));
        if (em.isOpen()) {
            if (em.getTransaction().isActive()) {
                em.getTransaction().rollback();
            }
            em.close();
        }
        em = AppCatalogJPAUtils.getEntityManager();
        em.getTransaction().begin();
        UserResourceProfile userResourceProfile = em.find(UserResourceProfile.class, new UserResourceProfilePK(userId, gatewayID));
        if (existingPreference != null) {
            existingPreference.setStorageResourceId(storageResourceId);
            existingPreference.setUserId(userId);
            existingPreference.setGatewayID(gatewayID);
            existingPreference.setUserResourceProfile(userResourceProfile);
            existingPreference.setLoginUserName(loginUserName);
            existingPreference.setComputeResourceCSToken(resourceCSToken);
            existingPreference.setFsRootLocation(fsRootLocation);
            em.merge(existingPreference);
        } else {
            UserStoragePreference resourcePreference = new UserStoragePreference();
            resourcePreference.setStorageResourceId(storageResourceId);
            resourcePreference.setGatewayID(gatewayID);
            resourcePreference.setUserId(userId);
            resourcePreference.setUserResourceProfile(userResourceProfile);
            resourcePreference.setLoginUserName(loginUserName);
            resourcePreference.setComputeResourceCSToken(resourceCSToken);
            resourcePreference.setFsRootLocation(fsRootLocation);
            em.persist(resourcePreference);
        }
        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();
        }
    }
}
Also used : UserStoragePreferencePK(org.apache.airavata.registry.core.app.catalog.model.UserStoragePreferencePK) UserResourceProfilePK(org.apache.airavata.registry.core.app.catalog.model.UserResourceProfilePK) EntityManager(javax.persistence.EntityManager) AppCatalogException(org.apache.airavata.registry.cpi.AppCatalogException) UserStoragePreference(org.apache.airavata.registry.core.app.catalog.model.UserStoragePreference) UserResourceProfile(org.apache.airavata.registry.core.app.catalog.model.UserResourceProfile) AppCatalogException(org.apache.airavata.registry.cpi.AppCatalogException) ApplicationSettingsException(org.apache.airavata.common.exception.ApplicationSettingsException)

Aggregations

EntityManager (javax.persistence.EntityManager)6 ApplicationSettingsException (org.apache.airavata.common.exception.ApplicationSettingsException)6 UserResourceProfile (org.apache.airavata.registry.core.app.catalog.model.UserResourceProfile)6 AppCatalogException (org.apache.airavata.registry.cpi.AppCatalogException)6 Query (javax.persistence.Query)4 AppCatalogQueryGenerator (org.apache.airavata.registry.core.app.catalog.util.AppCatalogQueryGenerator)4 ArrayList (java.util.ArrayList)3 List (java.util.List)3 UserResourceProfilePK (org.apache.airavata.registry.core.app.catalog.model.UserResourceProfilePK)2 CompositeIdentifier (org.apache.airavata.registry.cpi.CompositeIdentifier)2 HashMap (java.util.HashMap)1 Map (java.util.Map)1 UserStoragePreference (org.apache.airavata.registry.core.app.catalog.model.UserStoragePreference)1 UserStoragePreferencePK (org.apache.airavata.registry.core.app.catalog.model.UserStoragePreferencePK)1