summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--BRMSGateway/src/main/java/org/onap/policy/brms/api/BrmsPush.java210
-rw-r--r--BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsHandlerTest.java5
-rw-r--r--BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsPushTest.java6
3 files changed, 128 insertions, 93 deletions
diff --git a/BRMSGateway/src/main/java/org/onap/policy/brms/api/BrmsPush.java b/BRMSGateway/src/main/java/org/onap/policy/brms/api/BrmsPush.java
index 5e8046e22..8f55bd85c 100644
--- a/BRMSGateway/src/main/java/org/onap/policy/brms/api/BrmsPush.java
+++ b/BRMSGateway/src/main/java/org/onap/policy/brms/api/BrmsPush.java
@@ -57,7 +57,7 @@ import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.EntityTransaction;
import javax.persistence.Persistence;
-import javax.persistence.Query;
+import javax.persistence.TypedQuery;
import javax.ws.rs.ProcessingException;
import org.apache.commons.io.FileUtils;
@@ -73,6 +73,7 @@ import org.apache.maven.shared.invoker.InvocationRequest;
import org.apache.maven.shared.invoker.InvocationResult;
import org.apache.maven.shared.invoker.Invoker;
import org.codehaus.plexus.util.IOUtil;
+
import org.codehaus.plexus.util.WriterFactory;
import org.eclipse.persistence.config.PersistenceUnitProperties;
import org.onap.policy.api.PEDependency;
@@ -481,31 +482,39 @@ public class BrmsPush {
private void syncGroupInfo() {
// Sync DB to JMemory.
final EntityTransaction et = em.getTransaction();
- et.begin();
- Query query = em.createQuery("select b from BrmsGroupInfo AS b");
- List<?> result = query.getResultList();
- if (result.size() != groupMap.size()) {
- for (final Object value : result) {
- final BrmsGroupInfo brmsGroupInfo = (BrmsGroupInfo) value;
- final PEDependency dependency = new PEDependency();
- dependency.setArtifactId(brmsGroupInfo.getArtifactId());
- dependency.setGroupId(brmsGroupInfo.getGroupId());
- dependency.setVersion(brmsGroupInfo.getVersion());
- final ArrayList<Object> values = new ArrayList<>();
- values.add(dependency);
- groupMap.put(brmsGroupInfo.getControllerName(), values);
+ try {
+ et.begin();
+ final TypedQuery<BrmsGroupInfo> groupInfoQuery =
+ em.createQuery("select b from BrmsGroupInfo AS b", BrmsGroupInfo.class);
+ final List<BrmsGroupInfo> groupInfoResult = groupInfoQuery.getResultList();
+ if (groupInfoResult.size() != groupMap.size()) {
+ for (final BrmsGroupInfo brmsGroupInfo : groupInfoResult) {
+ final PEDependency dependency = new PEDependency();
+ dependency.setArtifactId(brmsGroupInfo.getArtifactId());
+ dependency.setGroupId(brmsGroupInfo.getGroupId());
+ dependency.setVersion(brmsGroupInfo.getVersion());
+ final ArrayList<Object> values = new ArrayList<>();
+ values.add(dependency);
+ groupMap.put(brmsGroupInfo.getControllerName(), values);
+ }
}
- }
- query = em.createQuery("select g from BrmsPolicyInfo AS g");
- result = query.getResultList();
- if (result.size() != policyMap.size()) {
- for (final Object value : result) {
- final BrmsPolicyInfo brmsPolicyInfo = (BrmsPolicyInfo) value;
- policyMap.put(brmsPolicyInfo.getPolicyName(), brmsPolicyInfo.getControllerName().getControllerName());
+
+ final TypedQuery<BrmsPolicyInfo> policyInfoQuery =
+ em.createQuery("select g from BrmsPolicyInfo AS g", BrmsPolicyInfo.class);
+ final List<BrmsPolicyInfo> policyInfoResult = policyInfoQuery.getResultList();
+ if (policyInfoResult.size() != policyMap.size()) {
+ for (final BrmsPolicyInfo brmsPolicyInfo : policyInfoResult) {
+ policyMap.put(brmsPolicyInfo.getPolicyName(),
+ brmsPolicyInfo.getControllerName().getControllerName());
+ }
}
+ et.commit();
+ LOGGER.info("Updated Local Memory values with values from database.");
+ } catch (final Exception exception) {
+ LOGGER.error("Unable to sync group info", exception);
+ et.rollback();
+ throw exception;
}
- et.commit();
- LOGGER.info("Updated Local Memory values with values from database.");
}
private void manageProject(final String selectedName, final String ksessionName, final String name,
@@ -524,37 +533,47 @@ public class BrmsPush {
* Add Policy to JMemory and DataBase.
*/
private void addToPolicy(final String policyName, final String controllerName) {
- policyMap.put(policyName, controllerName);
+
final EntityTransaction et = em.getTransaction();
- et.begin();
- Query query = em.createQuery("select b from BrmsPolicyInfo as b where b.policyName = :pn");
- query.setParameter("pn", policyName);
- final List<?> pList = query.getResultList();
- boolean create = false;
- BrmsPolicyInfo brmsPolicyInfo = new BrmsPolicyInfo();
- if (!pList.isEmpty()) {
- // Already exists.
- brmsPolicyInfo = (BrmsPolicyInfo) pList.get(0);
- if (!brmsPolicyInfo.getControllerName().getControllerName().equals(controllerName)) {
+ try {
+ et.begin();
+ boolean create = false;
+ final TypedQuery<BrmsPolicyInfo> policyInfoQuery =
+ em.createQuery("select b from BrmsPolicyInfo as b where b.policyName = :pn", BrmsPolicyInfo.class);
+ policyInfoQuery.setParameter("pn", policyName);
+ final List<BrmsPolicyInfo> policyInfoResultList = policyInfoQuery.getResultList();
+ BrmsPolicyInfo brmsPolicyInfo = new BrmsPolicyInfo();
+ if (!policyInfoResultList.isEmpty()) {
+ // Already exists.
+ brmsPolicyInfo = policyInfoResultList.get(0);
+ if (!brmsPolicyInfo.getControllerName().getControllerName().equals(controllerName)) {
+ create = true;
+ }
+ } else {
create = true;
}
- } else {
- create = true;
- }
- if (create) {
- query = em.createQuery("select b from BrmsGroupInfo as b where b.controllerName = :cn");
- query.setParameter("cn", controllerName);
- final List<?> bList = query.getResultList();
- BrmsGroupInfo brmsGroupInfo = new BrmsGroupInfo();
- if (!bList.isEmpty()) {
- brmsGroupInfo = (BrmsGroupInfo) bList.get(0);
+ if (create) {
+ final TypedQuery<BrmsGroupInfo> groupInfoQuery = em.createQuery(
+ "select b from BrmsGroupInfo as b where b.controllerName = :cn", BrmsGroupInfo.class);
+ groupInfoQuery.setParameter("cn", controllerName);
+ final List<BrmsGroupInfo> groupInfoResultList = groupInfoQuery.getResultList();
+ BrmsGroupInfo brmsGroupInfo = new BrmsGroupInfo();
+ if (!groupInfoResultList.isEmpty()) {
+ brmsGroupInfo = groupInfoResultList.get(0);
+ }
+ brmsPolicyInfo.setPolicyName(policyName);
+ brmsPolicyInfo.setControllerName(brmsGroupInfo);
+ em.persist(brmsPolicyInfo);
+ em.flush();
}
- brmsPolicyInfo.setPolicyName(policyName);
- brmsPolicyInfo.setControllerName(brmsGroupInfo);
- em.persist(brmsPolicyInfo);
- em.flush();
+ et.commit();
+
+ policyMap.put(policyName, controllerName);
+ } catch (final Exception exception) {
+ LOGGER.error("Unable add policy to database", exception);
+ et.rollback();
+ throw exception;
}
- et.commit();
}
private void syncProject(final String selectedName) {
@@ -692,10 +711,10 @@ public class BrmsPush {
final String repUrl = repUrlList.get(0);
restWrapper =
new NexusRestWrapper(repUrl.substring(0, repUrl.indexOf(repUrl.split(":[0-9]+\\/nexus")[1])),
- repUserName, repPassword);
+ repUserName, repPassword);
final NexusRestSearchParameters searchParameters = new NexusRestSearchParameters();
- searchParameters.useFilterSearch(
- getGroupId(selectedName), getArtifactId(selectedName), version, null, null);
+ searchParameters.useFilterSearch(getGroupId(selectedName), getArtifactId(selectedName), version, null,
+ null);
final List<NexusArtifact> resultList = restWrapper.findArtifact(searchParameters).getArtifactList();
if (resultList != null) {
@@ -705,7 +724,7 @@ public class BrmsPush {
} catch (NexusRestWrapperException | ProcessingException e) {
LOGGER.error(XACMLErrorConstants.ERROR_DATA_ISSUE + "Connection to remote Nexus has failed. "
+ e.getMessage(), e);
- } finally {
+ } finally {
if (null != restWrapper) {
restWrapper.close();
}
@@ -932,7 +951,7 @@ public class BrmsPush {
pomWriter.write(writer, model);
} catch (final Exception e) {
LOGGER.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + "Error while creating POM for " + getArtifactId(name)
- + e.getMessage(), e);
+ + e.getMessage(), e);
} finally {
IOUtil.close(writer);
}
@@ -1083,29 +1102,37 @@ public class BrmsPush {
}
private void addToGroup(final String name, final PEDependency dependency) {
- final ArrayList<Object> values = new ArrayList<>();
- values.add(dependency);
- groupMap.put(name, values);
final EntityTransaction et = em.getTransaction();
- et.begin();
- final Query query = em.createQuery("select b from BrmsGroupInfo as b where b.controllerName = :cn");
- query.setParameter("cn", name);
- final List<?> groupList = query.getResultList();
- BrmsGroupInfo brmsGroupInfo = null;
- if (!groupList.isEmpty()) {
- LOGGER.info("Controller name already Existing in DB. Will be updating the DB Values" + name);
- brmsGroupInfo = (BrmsGroupInfo) groupList.get(0);
- }
- if (brmsGroupInfo == null) {
- brmsGroupInfo = new BrmsGroupInfo();
+ try {
+ et.begin();
+ final TypedQuery<BrmsGroupInfo> query = em
+ .createQuery("select b from BrmsGroupInfo as b where b.controllerName = :cn", BrmsGroupInfo.class);
+ query.setParameter("cn", name);
+ final List<BrmsGroupInfo> groupList = query.getResultList();
+ BrmsGroupInfo brmsGroupInfo = null;
+ if (!groupList.isEmpty()) {
+ LOGGER.info("Controller name already Existing in DB. Will be updating the DB Values" + name);
+ brmsGroupInfo = groupList.get(0);
+ }
+ if (brmsGroupInfo == null) {
+ brmsGroupInfo = new BrmsGroupInfo();
+ }
+ brmsGroupInfo.setControllerName(name);
+ brmsGroupInfo.setGroupId(dependency.getGroupId());
+ brmsGroupInfo.setArtifactId(dependency.getArtifactId());
+ brmsGroupInfo.setVersion(dependency.getVersion());
+ em.persist(brmsGroupInfo);
+ em.flush();
+ et.commit();
+
+ final ArrayList<Object> values = new ArrayList<>();
+ values.add(dependency);
+ groupMap.put(name, values);
+ } catch (final Exception exception) {
+ LOGGER.error("Unable add/update policy group to database for controller name: " + name, exception);
+ et.rollback();
+ throw exception;
}
- brmsGroupInfo.setControllerName(name);
- brmsGroupInfo.setGroupId(dependency.getGroupId());
- brmsGroupInfo.setArtifactId(dependency.getArtifactId());
- brmsGroupInfo.setVersion(dependency.getVersion());
- em.persist(brmsGroupInfo);
- em.flush();
- et.commit();
}
private String getArtifactId(final String name) {
@@ -1139,22 +1166,29 @@ public class BrmsPush {
// Removes Policy from Memory and Database.
private void removePolicyFromGroup(final String policyName, final String controllerName) {
- policyMap.remove(policyName);
final EntityTransaction et = em.getTransaction();
- et.begin();
- final Query query = em.createQuery("select b from BrmsPolicyInfo as b where b.policyName = :pn");
- query.setParameter("pn", policyName);
- final List<?> pList = query.getResultList();
- BrmsPolicyInfo brmsPolicyInfo;
- if (!pList.isEmpty()) {
- // Already exists.
- brmsPolicyInfo = (BrmsPolicyInfo) pList.get(0);
- if (brmsPolicyInfo.getControllerName().getControllerName().equals(controllerName)) {
- em.remove(brmsPolicyInfo);
- em.flush();
+ try {
+ et.begin();
+ final TypedQuery<BrmsPolicyInfo> query =
+ em.createQuery("select b from BrmsPolicyInfo as b where b.policyName = :pn", BrmsPolicyInfo.class);
+ query.setParameter("pn", policyName);
+ final List<BrmsPolicyInfo> pList = query.getResultList();
+ BrmsPolicyInfo brmsPolicyInfo;
+ if (!pList.isEmpty()) {
+ // Already exists.
+ brmsPolicyInfo = pList.get(0);
+ if (brmsPolicyInfo.getControllerName().getControllerName().equals(controllerName)) {
+ em.remove(brmsPolicyInfo);
+ em.flush();
+ }
}
+ et.commit();
+ policyMap.remove(policyName);
+ } catch (final Exception exception) {
+ LOGGER.error("Unable remove policy from group to database for policy name: " + policyName, exception);
+ et.rollback();
+ throw exception;
}
- et.commit();
}
private void setVersion(final String selectedName) {
diff --git a/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsHandlerTest.java b/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsHandlerTest.java
index 58bacef13..1a6dc4719 100644
--- a/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsHandlerTest.java
+++ b/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsHandlerTest.java
@@ -30,6 +30,7 @@ import javax.persistence.EntityManagerFactory;
import javax.persistence.EntityTransaction;
import javax.persistence.Persistence;
import javax.persistence.Query;
+import javax.persistence.TypedQuery;
import org.junit.Test;
import org.junit.runner.RunWith;
@@ -54,8 +55,8 @@ public class BrmsHandlerTest {
PowerMockito.when(Persistence.createEntityManagerFactory(any(), any())).thenReturn(emf);
final EntityTransaction et = Mockito.mock(EntityTransaction.class);
Mockito.when(em.getTransaction()).thenReturn(et);
- final Query query = Mockito.mock(Query.class);
- Mockito.when(em.createQuery(Mockito.anyString())).thenReturn(query);
+ final TypedQuery<?> query = Mockito.mock(TypedQuery.class);
+ Mockito.when(em.createQuery(Mockito.anyString(), Mockito.any())).thenReturn((TypedQuery<Object>) query);
// Mock backup monitor
PowerMockito.mockStatic(BackUpMonitor.class);
diff --git a/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsPushTest.java b/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsPushTest.java
index 2b13640d2..2c76a2eff 100644
--- a/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsPushTest.java
+++ b/BRMSGateway/src/test/java/org/onap/policy/brms/api/BrmsPushTest.java
@@ -33,7 +33,7 @@ import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.EntityTransaction;
import javax.persistence.Persistence;
-import javax.persistence.Query;
+import javax.persistence.TypedQuery;
import org.apache.maven.model.Dependency;
import org.junit.Rule;
@@ -65,8 +65,8 @@ public class BrmsPushTest {
PowerMockito.when(Persistence.createEntityManagerFactory(Mockito.any(), Mockito.any())).thenReturn(emf);
final EntityTransaction et = Mockito.mock(EntityTransaction.class);
Mockito.when(em.getTransaction()).thenReturn(et);
- final Query query = Mockito.mock(Query.class);
- Mockito.when(em.createQuery(Mockito.anyString())).thenReturn(query);
+ final TypedQuery<?> query = Mockito.mock(TypedQuery.class);
+ Mockito.when(em.createQuery(Mockito.anyString(), Mockito.any())).thenReturn((TypedQuery<Object>) query);
// Mock backup monitor
PowerMockito.mockStatic(BackUpMonitor.class);