aai-core/src/main/java/org/openecomp/aai/restcore/util/GenerateEdgeRules.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.openecomp.aai
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.aai.restcore.util;
  22
  23 import com.att.eelf.configuration.EELFLogger;
  24 import com.att.eelf.configuration.EELFManager;
  25 import freemarker.template.Configuration;
  26 import freemarker.template.Template;
  27 import freemarker.template.TemplateException;
  28
  29 import java.io.*;
  30 import java.util.*;
  31
  32 public class GenerateEdgeRules {
  33
  34     private static final EELFLogger LOG = EELFManager.getInstance().getLogger(GenerateEdgeRules.class);
  35
  36     public static void main(String[] args) throws IOException, TemplateException {
  37
  38         String filename = "/AAI8032.csv";
  39         InputStream inputStream = GenerateEdgeRules.class.getResourceAsStream(filename);
  40         Map<String, Integer> headers = new HashMap<>();
  41         Map<String, Object> edgeRulesMap = new TreeMap<String, Object>();
  42         List<Map<String, String>> edgeRules = new ArrayList<>();
  43
  44         try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) {
  45
  46             String line = null;
  47
  48             int rowNum = 0;
  49
  50             // Retrieve the header line to map the indexes to their column names
  51
  52             while ((line = reader.readLine()) != null) {
  53
  54                 if (rowNum == 0) {
  55                     headers = retrieveHeaderMap(line);
  56                 } else {
  57                     String[] columns = line.split(",");
  58
  59                     String originalNode = columns[headers.get("Orig NodeA|NodeB")];
  60                     String finalNode = columns[headers.get("Final NodeA|NodeB")];
  61                     String originalEdge = columns[headers.get("Orig EdgeLabel")];
  62                     String finalEdge = columns[headers.get("Final EdgeLabel")];
  63
  64                     String lineage = columns[headers.get("Final Lineage")];
  65                     String originalParent = columns[headers.get("Orig ParentOf")];
  66                     String usesResource = columns[headers.get("Revised UsesResource")];
  67                     String hasDelTarget = columns[headers.get("Revised hasDelTarget")];
  68                     String svcInfra = columns[headers.get("Final SVC-INFRA")];
  69                     String svcInfraRev = "";
  70
  71                     if(usesResource.equals("T"))
  72                         usesResource = "true";
  73                     else if(usesResource.equals("F"))
  74                         usesResource = "false";
  75
  76                     if (hasDelTarget.equals("T") || hasDelTarget.equals("AB")) {
  77                         hasDelTarget = "true";
  78                     } else if (hasDelTarget.equals("F")) {
  79                         hasDelTarget = "false";
  80                     }
  81
  82                     if (svcInfra.equals("T")) {
  83                         svcInfra = "true";
  84                     } else if (svcInfra.equals("F")) {
  85                         svcInfra = "false";
  86                     } else if (svcInfra.equals("R")) {
  87                         svcInfra = "reverse";
  88                     }
  89
  90                     if (originalParent.equals("T")) {
  91                         if (lineage.trim().equalsIgnoreCase("CHILD")) {
  92                             lineage = "true";
  93                         } else if (lineage.trim().equalsIgnoreCase("PARENT")) {
  94                             lineage = "reverse";
  95                         }
  96                     } else {
  97                         lineage = "false";
  98                     }
  99
 100                     Map<String, String> edgeMap = new HashMap<String, String>();
 101
 102                     edgeMap.put("lineage", lineage);
 103                     edgeMap.put("usesResource", usesResource);
 104                     edgeMap.put("hasDelTarget", hasDelTarget);
 105                     edgeMap.put("SVC-INFRA", svcInfra);
 106                     edgeMap.put("SVC-INFRA-REV", svcInfraRev);
 107                     edgeMap.put("nodes", finalNode);
 108                     edgeMap.put("edge", finalEdge);
 109                     edgeMap.put("direction", columns[headers.get("Orig Direction")]);
 110                     edgeMap.put("multiplicity", columns[headers.get("Revised Multiplicity")]);
 111
 112                     edgeRules.add(edgeMap);
 113
 114                 }
 115                 ++rowNum;
 116             }
 117         } catch(Exception ex){
 118             ex.printStackTrace();
 119         }
 120
 121         edgeRulesMap.put("edgeRules", edgeRules);
 122
 123         Collections.sort(edgeRules, new Comparator<Map<String, String>>() {
 124             @Override
 125             public int compare(Map<String, String> o1, Map<String, String> o2) {
 126                 return o1.get("nodes").compareTo(o2.get("nodes"));
 127             }
 128         });
 129
 130         Configuration configuration = new Configuration();
 131         Template template = configuration.getTemplate("ajsc-aai/src/main/resources/EdgeRules.ftl");
 132         Writer file = new FileWriter(new File("ajsc-aai/src/main/resources" + "/" + "EdgeRules.txt"));
 133         template.process(edgeRulesMap, file);
 134     }
 135
 136     private static Map<String, Integer> retrieveHeaderMap(String line){
 137
 138         if(line == null)
 139             throw new NullPointerException();
 140
 141         String[] columnNames = line.split(",");
 142
 143         Map<String, Integer> map = new HashMap<String, Integer>();
 144
 145         int index = 0;
 146
 147         for(String columnName : columnNames){
 148             map.put(columnName, index++);
 149         }
 150
 151         return map;
 152     }
 153
 154 }