src/main/java/org/onap/clamp/clds/sdc/controller/installer/BlueprintParser.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP CLAMP
   4  * ================================================================================
   5  * Copyright (C) 2019 Nokia Intellectual Property. All rights
   6  *                             reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  * http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END============================================
  20  * ===================================================================
  21  *
  22  */
  23 package org.onap.clamp.clds.sdc.controller.installer;
  24
  25 import com.google.gson.Gson;
  26 import com.google.gson.JsonArray;
  27 import com.google.gson.JsonElement;
  28 import com.google.gson.JsonObject;
  29 import java.util.Collections;
  30 import java.util.HashSet;
  31 import java.util.List;
  32 import java.util.Map;
  33 import java.util.Map.Entry;
  34 import java.util.Set;
  35 import org.json.JSONObject;
  36 import org.springframework.stereotype.Component;
  37 import org.yaml.snakeyaml.Yaml;
  38
  39 @Component
  40 public class BlueprintParser {
  41
  42     static final String TCA = "TCA";
  43     static final String HOLMES = "Holmes";
  44     private static final String TCA_POLICY = "tca_policy";
  45     private static final String HOLMES_PREFIX = "holmes";
  46     private static final String NODE_TEMPLATES = "node_templates";
  47     private static final String DCAE_NODES = "dcae.nodes.";
  48     private static final String TYPE = "type";
  49     private static final String PROPERTIES = "properties";
  50     private static final String NAME = "name";
  51     private static final String RELATIONSHIPS = "relationships";
  52     private static final String CLAMP_NODE_RELATIONSHIPS_GETS_INPUT_FROM = "clamp_node.relationships.gets_input_from";
  53     private static final String TARGET = "target";
  54
  55     BlueprintParser() {}
  56
  57     Set<MicroService> getMicroServices(String blueprintString) {
  58         Set <MicroService> microServices = new HashSet<>();
  59         JsonObject jsonObject = BlueprintParser.convertToJson(blueprintString);
  60         JsonObject results = jsonObject.get(NODE_TEMPLATES).getAsJsonObject();
  61
  62         for (Entry<String, JsonElement> entry : results.entrySet()) {
  63             JsonObject nodeTemplate = entry.getValue().getAsJsonObject();
  64             if (nodeTemplate.get(TYPE).getAsString().contains(DCAE_NODES)) {
  65                 MicroService microService = getNodeRepresentation(entry);
  66                 microServices.add(microService);
  67             }
  68         }
  69         microServices.removeIf(ms -> TCA_POLICY.equals(ms.getName()));
  70         return microServices;
  71     }
  72
  73     List<MicroService> fallbackToOneMicroService(String blueprintString) {
  74         JsonObject jsonObject = BlueprintParser.convertToJson(blueprintString);
  75         JsonObject results = jsonObject.get(NODE_TEMPLATES).getAsJsonObject();
  76         String theBiggestMicroServiceContent = "";
  77         String theBiggestMicroServiceKey = "";
  78         for (Entry<String, JsonElement> entry : results.entrySet()) {
  79             String msAsString = entry.getValue().toString();
  80             int len =msAsString.length();
  81             if(len > theBiggestMicroServiceContent.length()) {
  82                 theBiggestMicroServiceContent = msAsString;
  83                 theBiggestMicroServiceKey = entry.getKey();
  84             }
  85         }
  86         String msName = theBiggestMicroServiceKey.toLowerCase().contains(HOLMES_PREFIX) ? HOLMES : TCA;
  87         return Collections.singletonList(new MicroService(msName, ""));
  88     }
  89
  90     String getName(Entry<String, JsonElement> entry) {
  91         String microServiceYamlName = entry.getKey();
  92         JsonObject ob = entry.getValue().getAsJsonObject();
  93         if (ob.has(PROPERTIES)) {
  94             JsonObject properties = ob.get(PROPERTIES).getAsJsonObject();
  95             if (properties.has(NAME)) {
  96                 return properties.get(NAME).getAsString();
  97             }
  98         }
  99         return microServiceYamlName;
 100     }
 101
 102     String getInput(Entry<String, JsonElement> entry) {
 103         JsonObject ob = entry.getValue().getAsJsonObject();
 104         if (ob.has(RELATIONSHIPS)) {
 105             JsonArray relationships = ob.getAsJsonArray(RELATIONSHIPS);
 106             for (JsonElement element : relationships) {
 107                 String target = getTarget(element.getAsJsonObject());
 108                 if(!target.isEmpty()) {
 109                     return target;
 110                 }
 111             }
 112         }
 113         return "";
 114     }
 115
 116     MicroService getNodeRepresentation(Entry<String, JsonElement> entry) {
 117         String name = getName(entry);
 118         String getInputFrom = getInput(entry);
 119         return new MicroService(name, getInputFrom);
 120     }
 121
 122     private String getTarget(JsonObject elementObject) {
 123         if (elementObject.has(TYPE) &&
 124             elementObject.has(TARGET) &&
 125             elementObject.get(TYPE).getAsString().equals(CLAMP_NODE_RELATIONSHIPS_GETS_INPUT_FROM)) {
 126             return elementObject.get(TARGET).getAsString();
 127         }
 128         return "";
 129     }
 130
 131     private static JsonObject convertToJson(String yamlString) {
 132         Yaml yaml = new Yaml();
 133         Map<String, Object> map = yaml.load(yamlString);
 134
 135         JSONObject jsonObject = new JSONObject(map);
 136         return new Gson().fromJson(jsonObject.toString(), JsonObject.class);
 137     }
 138 }