openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/zusammen/ServiceModelDaoZusammenImpl.java

   1 package org.openecomp.sdc.model.impl.zusammen;
   2
   3 import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
   4 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ElementInfo;
   5 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
   6 import com.amdocs.zusammen.datatypes.Id;
   7 import com.amdocs.zusammen.datatypes.SessionContext;
   8 import com.amdocs.zusammen.datatypes.item.Action;
   9 import com.amdocs.zusammen.datatypes.item.ElementContext;
  10 import com.amdocs.zusammen.datatypes.item.Info;
  11 import com.amdocs.zusammen.datatypes.item.ItemVersion;
  12 import org.apache.commons.io.IOUtils;
  13 import org.openecomp.core.model.dao.ServiceModelDao;
  14 import org.openecomp.core.model.types.ServiceElement;
  15 import org.openecomp.core.utilities.file.FileContentHandler;
  16 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
  17 import org.openecomp.core.zusammen.api.ZusammenUtil;
  18 import org.openecomp.sdc.logging.api.Logger;
  19 import org.openecomp.sdc.logging.api.LoggerFactory;
  20 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
  21 import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
  22 import org.openecomp.sdc.tosca.services.ToscaExtensionYamlUtil;
  23 import org.openecomp.sdc.versioning.dao.types.Version;
  24 import org.openecomp.sdc.versioning.dao.types.VersionStatus;
  25
  26 import java.io.ByteArrayInputStream;
  27 import java.io.IOException;
  28 import java.util.Collection;
  29 import java.util.Map;
  30 import java.util.Objects;
  31 import java.util.Optional;
  32 import java.util.stream.Collectors;
  33
  34 public class ServiceModelDaoZusammenImpl
  35     implements ServiceModelDao<ToscaServiceModel, ServiceElement> {
  36   private static final Logger logger = LoggerFactory.getLogger(ServiceModelDaoZusammenImpl.class);
  37
  38   protected ZusammenAdaptor zusammenAdaptor;
  39   protected String name;
  40
  41   public ServiceModelDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
  42     this.zusammenAdaptor = zusammenAdaptor;
  43     this.name = StructureElement.ServiceModel.name();
  44   }
  45
  46   @Override
  47   public void registerVersioning(String versionableEntityType) {
  48
  49   }
  50
  51   @Override
  52   public ToscaServiceModel getServiceModel(String vspId, Version version) {
  53     SessionContext context = ZusammenUtil.createSessionContext();
  54     Id itemId = new Id(vspId);
  55     ElementContext elementContext = new ElementContext(itemId, getFirstVersionId(context, itemId),
  56         version.getStatus() == VersionStatus.Locked ? null : version.toString());
  57
  58     Optional<ElementInfo> serviceModelElement = zusammenAdaptor
  59         .getElementInfoByName(context, elementContext, null, name);
  60
  61     if (serviceModelElement.isPresent()) {
  62       String entryDefinitionServiceTemplate =
  63           serviceModelElement.get().getInfo().getProperty("base");
  64       Id serviceModelElementId = serviceModelElement.get().getId();
  65       Map<String, ServiceTemplate> serviceTemplates =
  66           getTemplates(context, elementContext, serviceModelElementId);
  67       if (serviceTemplates == null) {
  68         return null;
  69       }
  70       FileContentHandler artifacts = getArtifacts(context, elementContext, serviceModelElementId);
  71
  72
  73       return new ToscaServiceModel(
  74           artifacts, serviceTemplates, entryDefinitionServiceTemplate);
  75     } else {
  76       return null;
  77     }
  78   }
  79
  80   protected Map<String, ServiceTemplate> getTemplates(SessionContext context,
  81                                                       ElementContext elementContext,
  82                                                       Id serviceModelElementId) {
  83     Optional<ElementInfo> templatesElementInfo = zusammenAdaptor.getElementInfoByName(
  84         context, elementContext, serviceModelElementId, StructureElement.Templates.name());
  85
  86     if (templatesElementInfo.isPresent()) {
  87
  88       //Map<String, ServiceTemplate> serviceTemplateMap = new HashMap<>();
  89       Collection<Element> elements = zusammenAdaptor.listElementData(context, elementContext,
  90           templatesElementInfo.get().getId());
  91
  92       return elements.stream().collect(Collectors.toMap(
  93           element -> element.getInfo().getName(),
  94           this::elementToServiceTemplate));
  95     /*elements.stream().forEach(element ->serviceTemplateMap.put(element.getInfo().getName(),
  96         elementToServiceTemplate(element)));
  97     return serviceTemplateMap;*/
  98     }
  99     return null;
 100   }
 101
 102   protected FileContentHandler getArtifacts(SessionContext context, ElementContext elementContext,
 103                                             Id serviceModelElementId) {
 104     Optional<ElementInfo> artifactsElement = zusammenAdaptor.getElementInfoByName(
 105         context, elementContext, serviceModelElementId, StructureElement.Artifacts.name());
 106
 107     if (artifactsElement.isPresent()) {
 108
 109       Collection<Element> elements = zusammenAdaptor.listElementData(context, elementContext,
 110           artifactsElement.get().getId());
 111       FileContentHandler fileContentHandler = new FileContentHandler();
 112       elements.forEach(element -> fileContentHandler.addFile(element.getInfo().getName(),
 113           element.getData()));
 114       return fileContentHandler;
 115     }
 116
 117     return null;
 118   }
 119
 120   @Override
 121   public void storeServiceModel(String vspId, Version version, ToscaServiceModel serviceModel) {
 122     logger.info("Storing service model for vsp id -> " + vspId);
 123
 124     ZusammenElement templatesElement =
 125         buildStructuralElement(StructureElement.Templates.name(), null);
 126     serviceModel.getServiceTemplates().entrySet().forEach(entry -> templatesElement.addSubElement(
 127         buildServiceTemplateElement(entry.getKey(), entry.getValue(),
 128             serviceModel.getEntryDefinitionServiceTemplate(), Action.CREATE)));
 129
 130     ZusammenElement artifactsElement =
 131         buildStructuralElement(StructureElement.Artifacts.name(), Action.UPDATE);
 132     if (Objects.nonNull(serviceModel.getArtifactFiles())) {
 133       serviceModel.getArtifactFiles().getFiles().entrySet().forEach(entry -> artifactsElement
 134           .addSubElement(buildArtifactElement(entry.getKey(), entry.getValue(), Action.CREATE)));
 135     }
 136
 137     ZusammenElement serviceModelElement = buildStructuralElement(name, Action.UPDATE);
 138     serviceModelElement.getInfo()
 139         .addProperty("base", serviceModel.getEntryDefinitionServiceTemplate());
 140
 141     serviceModelElement.addSubElement(templatesElement);
 142     serviceModelElement.addSubElement(artifactsElement);
 143
 144     SessionContext context = ZusammenUtil.createSessionContext();
 145     Id itemId = new Id(vspId);
 146     ElementContext elementContext = new ElementContext(itemId, getFirstVersionId(context, itemId));
 147     zusammenAdaptor
 148         .saveElement(context, elementContext, serviceModelElement, "Store service model");
 149
 150     logger.info("Finished storing service model for vsp id -> " + vspId);
 151   }
 152
 153   @Override
 154   public ServiceElement getServiceModelInfo(String vspId, Version version, String name) {
 155     return null;
 156   }
 157
 158   @Override
 159   public void deleteAll(String vspId, Version version) {
 160     logger.info("started deleting service model for vsp id -> " + vspId);
 161     SessionContext context = ZusammenUtil.createSessionContext();
 162     Id itemId = new Id(vspId);
 163     ElementContext elementContext = new ElementContext(itemId, getFirstVersionId(context, itemId));
 164
 165     ZusammenElement zusammenElement = ZusammenUtil.buildStructuralElement(name, Action.DELETE);
 166     zusammenAdaptor.saveElement(context, elementContext, zusammenElement, "delete:" + name + ".");
 167     logger.info("Finished deleting service model for vsp id -> " + vspId);
 168   }
 169
 170   protected ZusammenElement buildArtifactElement(String name, byte[] artifact, Action action) {
 171     ZusammenElement artifactElement = new ZusammenElement();
 172     artifactElement.setAction(action);
 173     Info info = new Info();
 174     info.setName(name);
 175     info.addProperty("type", ElementType.Artifact.name());
 176     artifactElement.setInfo(info);
 177     artifactElement.setData(new ByteArrayInputStream(artifact));
 178
 179     return artifactElement;
 180   }
 181
 182   private ServiceTemplate elementToServiceTemplate(Element element){
 183
 184     try {
 185       String yamlContent = IOUtils.toString(element.getData());
 186       return new ToscaExtensionYamlUtil().
 187           yamlToObject(yamlContent, ServiceTemplate.class);
 188     }catch (Exception e){
 189       return null;
 190     }
 191   }
 192
 193   private Element buildServiceTemplateElement(String name, ServiceTemplate serviceTemplate,
 194                                               String entryDefinitionServiceTemplate,
 195                                               Action action) {
 196     ZusammenElement zusammenElement = new ZusammenElement();
 197     zusammenElement.setAction(action);
 198     Info info = new Info();
 199     info.setName(name);
 200     info.setDescription(serviceTemplate.getDescription());
 201     info.addProperty("type", ElementType.Servicetemplate.name());
 202     info.addProperty("base", entryDefinitionServiceTemplate);
 203     String yaml = new ToscaExtensionYamlUtil().objectToYaml(serviceTemplate);
 204     zusammenElement.setData(new ByteArrayInputStream(yaml.getBytes()));
 205     zusammenElement.setInfo(info);
 206     return zusammenElement;
 207   }
 208
 209   protected Id getFirstVersionId(SessionContext context, Id vspId) {
 210     Optional<ItemVersion> itemVersionOptional = zusammenAdaptor.getFirstVersion(context, vspId);
 211     ItemVersion itemVersion = itemVersionOptional.orElseThrow(() ->
 212         new RuntimeException(String.format("Vsp %s does not contain any version.", vspId))); //todo
 213     return itemVersion.getId();
 214   }
 215
 216   protected ZusammenElement buildStructuralElement(String structureElement, Action action) {
 217     return ZusammenUtil.buildStructuralElement(structureElement, action);
 218   }
 219 }