vid-app-common/src/main/java/org/onap/vid/model/aaiTree/Vnf.java

   1 package org.onap.vid.model.aaiTree;
   2
   3 import org.onap.vid.aai.util.AAITreeConverter;
   4 import org.onap.vid.services.AAITreeNodeBuilder;
   5
   6 import java.util.HashMap;
   7 import java.util.Map;
   8
   9 import static org.onap.vid.aai.util.AAITreeConverter.VNF_TYPE;
  10
  11 public class Vnf extends Node {
  12
  13     private Map<String, Map<String, VfModule>> vfModules = new HashMap<>();
  14     private Map<String, Network> networks = new HashMap<>();
  15
  16     public Vnf(AAITreeNode node) {
  17         super(node, AAITreeConverter.ModelType.vnf);
  18     }
  19
  20     public Map<String, Map<String, VfModule>> getVfModules() {
  21         return vfModules;
  22     }
  23
  24     public void setVfModules(Map<String, Map<String, VfModule>> vfModules) {
  25         this.vfModules = vfModules;
  26     }
  27
  28     public Map<String, Network> getNetworks() {
  29         return networks;
  30     }
  31
  32     public void setNetworks(Map<String, Network> networks) {
  33         this.networks = networks;
  34     }
  35
  36     public static Vnf from(AAITreeNode node) {
  37         Vnf vnf = new Vnf(node);
  38         if (node.getAdditionalProperties().get(VNF_TYPE) != null) {
  39             vnf.setInstanceType(node.getAdditionalProperties().get(VNF_TYPE).toString());
  40         }
  41
  42         node.getChildren().forEach(child -> {
  43             if (child.getType().equals(AAITreeNodeBuilder.VF_MODULE)) {
  44                 vnf.getVfModules().putIfAbsent(child.getNodeKey(), new HashMap<>());
  45                 vnf.getVfModules().get(child.getNodeKey())
  46                                     .put(child.getUniqueNodeKey(), VfModule.from(child));
  47             } else if (child.getType().equals(AAITreeNodeBuilder.NETWORK)) {
  48                 vnf.getNetworks().put(child.getUniqueNodeKey(), Network.from(child));
  49             }
  50         });
  51
  52         return vnf;
  53     }
  54 }