openecomp-be/lib/openecomp-sdc-vendor-software-product-lib/openecomp-sdc-vendor-software-product-core/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/dao/impl/zusammen/OrchestrationTemplateCandidateDaoZusammenImpl.java

   1 package org.openecomp.sdc.vendorsoftwareproduct.dao.impl.zusammen;
   2
   3 import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
   4 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
   5 import com.amdocs.zusammen.datatypes.Id;
   6 import com.amdocs.zusammen.datatypes.SessionContext;
   7 import com.amdocs.zusammen.datatypes.item.Action;
   8 import com.amdocs.zusammen.datatypes.item.ElementContext;
   9 import com.amdocs.zusammen.utils.fileutils.FileUtils;
  10 import org.openecomp.core.utilities.json.JsonUtil;
  11 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
  12 import org.openecomp.core.zusammen.api.ZusammenUtil;
  13 import org.openecomp.sdc.logging.api.Logger;
  14 import org.openecomp.sdc.logging.api.LoggerFactory;
  15 import org.openecomp.sdc.vendorsoftwareproduct.dao.OrchestrationTemplateCandidateDao;
  16 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.OrchestrationTemplateCandidateData;
  17 import org.openecomp.sdc.vendorsoftwareproduct.types.candidateheat.FilesDataStructure;
  18 import org.openecomp.sdc.versioning.dao.types.Version;
  19 import org.openecomp.sdc.versioning.dao.types.VersionStatus;
  20
  21 import java.io.ByteArrayInputStream;
  22 import java.nio.ByteBuffer;
  23 import java.util.Optional;
  24
  25 public class OrchestrationTemplateCandidateDaoZusammenImpl
  26     implements OrchestrationTemplateCandidateDao {
  27   private static final Logger logger =
  28       LoggerFactory.getLogger(OrchestrationTemplateCandidateDaoZusammenImpl.class);
  29
  30   private ZusammenAdaptor zusammenAdaptor;
  31
  32   public OrchestrationTemplateCandidateDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
  33     this.zusammenAdaptor = zusammenAdaptor;
  34   }
  35
  36   @Override
  37   public void registerVersioning(String versionableEntityType) {
  38
  39   }
  40
  41   @Override
  42   public OrchestrationTemplateCandidateData get(String vspId, Version version) {
  43     logger.info("Getting orchestration template for vsp id -> " + vspId);
  44
  45     SessionContext context = ZusammenUtil.createSessionContext();
  46     Id itemId = new Id(vspId);
  47     Id versionId = VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor);
  48     ElementContext elementContext = new ElementContext(itemId, versionId,
  49         VspZusammenUtil.getVersionTag(version));
  50
  51     Optional<Element> candidateElement =
  52         zusammenAdaptor.getElementByName(context, elementContext, null,
  53             StructureElement.OrchestrationTemplateCandidate.name());
  54     if (candidateElement.isPresent()) {
  55       OrchestrationTemplateCandidateData candidateData = new OrchestrationTemplateCandidateData();
  56       candidateData.setFilesDataStructure(
  57           new String(FileUtils.toByteArray(candidateElement.get().getData())));
  58
  59       zusammenAdaptor
  60           .getElementByName(context, elementContext, candidateElement.get().getElementId(),
  61               StructureElement.OrchestrationTemplateCandidateContent.name())
  62           .ifPresent(candidateContentElement -> candidateData.setContentData(
  63               ByteBuffer.wrap(FileUtils.toByteArray(candidateContentElement.getData()))));
  64       logger.info("Finished getting orchestration template for vsp id -> " + vspId);
  65       return candidateData;
  66     }
  67     logger.info(String.format("Orchestration template for vsp id %s does not exist", vspId));
  68     return null;
  69   }
  70
  71   @Override
  72   public void update(String vspId, OrchestrationTemplateCandidateData candidateData) {
  73     logger.info("Uploading candidate data entity for vsp id -> " + vspId);
  74
  75     SessionContext context = ZusammenUtil.createSessionContext();
  76     Id itemId = new Id(vspId);
  77     Id versionId = VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor);
  78     ElementContext elementContext = new ElementContext(itemId, versionId);
  79
  80     ZusammenElement candidateElement = VspZusammenUtil
  81         .buildStructuralElement(StructureElement.OrchestrationTemplateCandidate, Action.UPDATE);
  82     candidateElement
  83         .setData(new ByteArrayInputStream(candidateData.getFilesDataStructure().getBytes()));
  84     ZusammenElement candidateContentElement = VspZusammenUtil
  85         .buildStructuralElement(StructureElement.OrchestrationTemplateCandidateContent, Action.UPDATE);
  86     candidateContentElement
  87         .setData(new ByteArrayInputStream(candidateData.getContentData().array()));
  88     candidateElement.addSubElement(candidateContentElement);
  89
  90     zusammenAdaptor.saveElement(context, elementContext, candidateElement,
  91         "Update Orchestration Template Candidate");
  92     logger.info("Finished uploading candidate data entity for vsp id -> " + vspId);
  93   }
  94
  95   @Override
  96   public void updateStructure(String vspId, Version version, FilesDataStructure fileDataStructure) {
  97     logger.info("Updating orchestration template for VSP id -> " + vspId);
  98
  99     SessionContext context = ZusammenUtil.createSessionContext();
 100     Id itemId = new Id(vspId);
 101     Id versionId = VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor);
 102     ElementContext elementContext = new ElementContext(itemId, versionId);
 103
 104     ZusammenElement candidateElement = VspZusammenUtil
 105         .buildStructuralElement(StructureElement.OrchestrationTemplateCandidate, Action.UPDATE);
 106     candidateElement
 107         .setData(new ByteArrayInputStream(JsonUtil.object2Json(fileDataStructure).getBytes()));
 108     zusammenAdaptor.saveElement(context, elementContext, candidateElement,
 109         "Update Orchestration Template Candidate structure");
 110     logger.info("Finished uploading candidate data entity for vsp id -> " + vspId);
 111   }
 112
 113
 114   @Override
 115   public Optional<String> getStructure(String vspId, Version version) {
 116     logger.info("Getting orchestration template structure for vsp id -> " + vspId);
 117
 118     SessionContext context = ZusammenUtil.createSessionContext();
 119     Id itemId = new Id(vspId);
 120     Id versionId = VspZusammenUtil.getFirstVersionId(context, itemId, zusammenAdaptor);
 121     ElementContext elementContext = new ElementContext(itemId, versionId,
 122         VspZusammenUtil.getVersionTag(version));
 123
 124     logger.info("Finished getting orchestration template structure for vsp id -> " + vspId);
 125     Optional<Element> element = zusammenAdaptor.getElementByName(context, elementContext, null,
 126         StructureElement.OrchestrationTemplateCandidate.name());
 127     if (element.isPresent()) {
 128       return Optional.of(new String(FileUtils.toByteArray(element.get().getData())));
 129     } else {
 130       return Optional.empty();
 131     }
 132
 133   }
 134 }