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