adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/openstack/utils/MsoYamlEditorWithEnvt.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * Copyright (C) 2017 Huawei Technologies Co., Ltd. All rights 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 package org.openecomp.mso.openstack.utils;
  23
  24
  25
  26 import java.io.ByteArrayInputStream;
  27 import java.io.InputStream;
  28 import java.util.HashSet;
  29 import java.util.Iterator;
  30 import java.util.Map;
  31 import java.util.Set;
  32 import java.util.Map.Entry;
  33 import org.openecomp.mso.db.catalog.beans.HeatTemplateParam;
  34 import org.codehaus.jackson.map.ObjectMapper;
  35
  36 import org.yaml.snakeyaml.Yaml;
  37
  38 import org.openecomp.mso.logger.MsoLogger;
  39
  40 public class MsoYamlEditorWithEnvt {
  41
  42     private static final MsoLogger LOGGER = MsoLogger.getMsoLogger (MsoLogger.Catalog.RA);
  43
  44     private Map <String, Object> yml;
  45     private Yaml yaml = new Yaml ();
  46     private static final ObjectMapper JSON_MAPPER = new ObjectMapper();
  47
  48         public MsoYamlEditorWithEnvt() {
  49                 super();
  50         }
  51         public MsoYamlEditorWithEnvt(byte[] b) {
  52                 init(b);
  53         }
  54
  55     @SuppressWarnings("unchecked")
  56     private synchronized void init (byte[] body) {
  57         InputStream input = new ByteArrayInputStream (body);
  58         yml = (Map <String, Object>) yaml.load (input);
  59     }
  60
  61     @SuppressWarnings("unchecked")
  62         public synchronized Set <MsoHeatEnvironmentParameter> getParameterListFromEnvt() {
  63         // In an environment entry, the parameters section can only contain the name:value -
  64         // not other attributes.
  65         Set <MsoHeatEnvironmentParameter> paramSet = new HashSet<>();
  66         Map<String, Object> resourceMap = null;
  67         try {
  68                 resourceMap = (Map<String,Object>) yml.get("parameters");
  69         } catch (Exception e) {
  70             LOGGER.debug("Exception:", e);
  71                 return paramSet;
  72         }
  73         if (resourceMap == null) {
  74                 return paramSet;
  75         }
  76         Iterator <Entry <String, Object>> it = resourceMap.entrySet().iterator();
  77
  78         while (it.hasNext()) {
  79                 MsoHeatEnvironmentParameter hep = new MsoHeatEnvironmentParameter();
  80                 Map.Entry <String, Object> pair = it.next();
  81                 String value;
  82                 Object obj = pair.getValue();
  83                 if (obj instanceof java.lang.String) {
  84                         value = yaml.dump(obj);
  85                         // but this adds an extra '\n' at the end - which won't hurt - but we don't need it
  86                         value = value.substring(0, value.length() - 1);
  87                 } else if (obj instanceof java.util.LinkedHashMap) {
  88                         //Handle that it's json
  89                         try {
  90                                 value = JSON_MAPPER.writeValueAsString(obj);
  91                         } catch (Exception e) {
  92                             LOGGER.debug("Exception:", e);
  93                                 value = "_BAD_JSON_MAPPING";
  94                         }
  95                 } else {
  96                         //this handles integers/longs/floats/etc.
  97                         value = String.valueOf(obj);
  98                 }
  99                 hep.setName((String) pair.getKey());
 100                 hep.setValue(value);
 101                 paramSet.add(hep);
 102         }
 103         return paramSet;
 104     }
 105     public synchronized Set <MsoHeatEnvironmentResource> getResourceListFromEnvt() {
 106         try {
 107                 Set<MsoHeatEnvironmentResource> resourceList = new HashSet<>();
 108                 @SuppressWarnings("unchecked")
 109                 Map<String, Object> resourceMap = (Map<String,Object>) yml.get("resource_registry");
 110                 Iterator<Entry <String,Object>> it = resourceMap.entrySet().iterator();
 111
 112                 while (it.hasNext()) {
 113                         MsoHeatEnvironmentResource her = new MsoHeatEnvironmentResource();
 114                         Map.Entry<String, Object> pair = it.next();
 115                         her.setName((String) pair.getKey());
 116                         her.setValue((String) pair.getValue());
 117                         resourceList.add(her);
 118                 }
 119                 return resourceList;
 120         } catch (Exception e) {
 121             LOGGER.debug("Exception:", e);
 122         }
 123         return null;
 124     }
 125     public synchronized Set <HeatTemplateParam> getParameterList () {
 126         Set <HeatTemplateParam> paramSet = new HashSet <> ();
 127         @SuppressWarnings("unchecked")
 128         Map <String, Object> resourceMap = (Map <String, Object>) yml.get ("parameters");
 129         Iterator <Entry <String, Object>> it = resourceMap.entrySet ().iterator ();
 130
 131         while (it.hasNext ()) {
 132             HeatTemplateParam param = new HeatTemplateParam ();
 133             Map.Entry <String, Object> pair = it.next ();
 134             @SuppressWarnings("unchecked")
 135             Map <String, String> resourceEntry = (Map <String, String>) pair.getValue ();
 136             String value = null;
 137             try {
 138                 value = resourceEntry.get ("default");
 139             } catch (java.lang.ClassCastException cce) {
 140                 LOGGER.debug("Exception:", cce);
 141                 // This exception only - the value is an integer. For what we're doing
 142                 // here - we don't care - so set value to something - and it will
 143                 // get marked as not being required - which is correct.
 144                 //System.out.println("cce exception!");
 145                 value = "300";
 146                 // okay
 147             }
 148             param.setParamName ((String) pair.getKey ());
 149             if (value != null) {
 150                 param.setRequired (false);
 151             } else {
 152                 param.setRequired (true);
 153             }
 154             value = resourceEntry.get ("type");
 155             param.setParamType (value);
 156
 157             paramSet.add (param);
 158
 159         }
 160         return paramSet;
 161
 162     }
 163
 164
 165 }