1 files changed, 52 insertions, 0 deletions
diff --git a/vid-app-common/src/main/java/org/onap/vid/model/aaiTree/RelatedVnf.java b/vid-app-common/src/main/java/org/onap/vid/model/aaiTree/RelatedVnf.java
new file mode 100644
index 000000000..043d2ccd9
--- /dev/null
+++ b/vid-app-common/src/main/java/org/onap/vid/model/aaiTree/RelatedVnf.java
@@ -0,0 +1,52 @@
+package org.onap.vid.model.aaiTree;
+
+import org.onap.vid.aai.util.AAITreeConverter;
+
+import static org.onap.vid.aai.util.AAITreeConverter.VNF_TYPE;
+
+public class RelatedVnf extends Node {
+
+    private String serviceInstanceId;
+    private String serviceInstanceName;
+    private String tenantName;
+
+    public String getServiceInstanceId() {
+        return serviceInstanceId;
+    }
+
+    public void setServiceInstanceId(String serviceInstanceId) {
+        this.serviceInstanceId = serviceInstanceId;
+    }
+
+    public String getServiceInstanceName() {
+        return serviceInstanceName;
+    }
+
+    public void setServiceInstanceName(String serviceInstanceName) {
+        this.serviceInstanceName = serviceInstanceName;
+    }
+
+    public String getTenantName() {
+        return tenantName;
+    }
+
+    public void setTenantName(String tenantName) {
+        this.tenantName = tenantName;
+    }
+
+    public RelatedVnf(AAITreeNode node) {
+        super(node, AAITreeConverter.ModelType.vnf);
+    }
+
+    public static RelatedVnf from(AAITreeNode node) {
+        RelatedVnf vnf = new RelatedVnf(node);
+        vnf.setServiceInstanceId(node.getParent().getId());
+        vnf.setServiceInstanceName(node.getParent().getName());
+
+        if (node.getAdditionalProperties().get(VNF_TYPE) != null) {
+            vnf.setInstanceType(node.getAdditionalProperties().get(VNF_TYPE).toString());
+        }
+
+        return vnf;
+    }
+}