X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=holmes-actions%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Fholmes%2Fcommon%2Faai%2Fentity%2FRelationshipList.java;h=533760c25ae7311627b441377813418898ac9258;hb=3e52d0c33f6a7db4e81ebecc50623cd449c624ba;hp=8711e2cbf4fe09408bc6154b118d373a25f99d36;hpb=80aa5b49e5e3fb696231ce66acf198ff50d74731;p=holmes%2Fcommon.git

diff --git a/holmes-actions/src/main/java/org/onap/holmes/common/aai/entity/RelationshipList.java b/holmes-actions/src/main/java/org/onap/holmes/common/aai/entity/RelationshipList.java
index 8711e2c..533760c 100644
--- a/holmes-actions/src/main/java/org/onap/holmes/common/aai/entity/RelationshipList.java
+++ b/holmes-actions/src/main/java/org/onap/holmes/common/aai/entity/RelationshipList.java
@@ -26,6 +26,19 @@ public class RelationshipList {
 
     private List<Relationship> relationships;
 
+    public Relationship getRelationship(String relatedTo) {
+        Relationship relationship = null;
+        if(null == relationships || relationships.size() == 0)
+            return  relationship;
+        for(int i = 0; i < relationships.size(); i++) {
+            if (relatedTo.equals(relationships.get(i).getRelatedTo())) {
+                relationship = relationships.get(i);
+                break;
+            }
+        }
+        return relationship;
+    }
+
     @Setter
     @Getter
     public static class Relationship {
@@ -33,6 +46,34 @@ public class RelationshipList {
         private String relatedTo;
         private List<RelatedToProperty> relatedToPropertyList;
         private List<RelationshipData> relationshipDataList;
+
+        public String getRelatedToPropertyValue(String key) {
+            String value = "";
+            if (null == relatedToPropertyList || relatedToPropertyList.size() == 0) {
+                return "";
+            }
+            for(int i = 0; i < relatedToPropertyList.size(); i++) {
+                if (key.equals(relatedToPropertyList.get(i).getPropertyKey())) {
+                    value = relatedToPropertyList.get(i).getPropertyValue();
+                    break;
+                }
+            }
+            return value;
+        }
+
+        public String getRelationshipDataValue(String key) {
+            String value = "";
+            if (null == relationshipDataList || relationshipDataList.size() == 0) {
+                return "";
+            }
+            for(int i = 0; i < relationshipDataList.size(); i++) {
+                if (key.equals(relationshipDataList.get(i).getRelationshipKey())) {
+                    value = relationshipDataList.get(i).getRelationshipValue();
+                    break;
+                }
+            }
+            return value;
+        }
     }
 
     @Setter