gerrit.onap Code Review - sdc/sdc-workflow-designer.git/blob

   1 package org.onap.sdc.workflow.persistence.impl;
   2
   3 import static org.openecomp.core.zusammen.api.ZusammenUtil.buildStructuralElement;
   4 import static org.openecomp.core.zusammen.api.ZusammenUtil.createSessionContext;
   5
   6 import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
   7 import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
   8 import com.amdocs.zusammen.datatypes.SessionContext;
   9 import com.amdocs.zusammen.datatypes.item.Action;
  10 import com.amdocs.zusammen.datatypes.item.ElementContext;
  11 import java.io.ByteArrayInputStream;
  12 import java.io.IOException;
  13 import java.io.InputStream;
  14 import java.util.Arrays;
  15 import java.util.Map;
  16 import java.util.Optional;
  17 import org.apache.commons.io.IOUtils;
  18 import org.onap.sdc.workflow.persistence.ArtifactRepository;
  19 import org.onap.sdc.workflow.persistence.types.ArtifactEntity;
  20 import org.openecomp.core.zusammen.api.ZusammenAdaptor;
  21 import org.openecomp.core.zusammen.api.ZusammenAdaptorFactory;
  22 import org.openecomp.sdc.datatypes.model.ElementType;
  23 import org.springframework.stereotype.Repository;
  24
  25 @Repository
  26 public class ArtifactRepositoryImpl implements ArtifactRepository {
  27
  28     private static final String FILE_NAME_PROPERTY = "fileName";
  29     private static final String EMPTY_DATA = "{}";
  30     private ZusammenAdaptor zusammenAdaptor = ZusammenAdaptorFactory.getInstance().createInterface();
  31
  32
  33     @Override
  34     public void update(String id, String versionId, ArtifactEntity artifactEntity) {
  35
  36         ZusammenElement artifactElement = buildStructuralElement(ElementType.Artifact, Action.UPDATE);
  37         artifactElement.setData(artifactEntity.getArtifactData());
  38         artifactElement.getInfo().addProperty(FILE_NAME_PROPERTY, artifactEntity.getFileName());
  39
  40         SessionContext context = createSessionContext();
  41         ElementContext elementContext = new ElementContext(id, versionId);
  42
  43         zusammenAdaptor
  44                 .saveElement(context, elementContext, artifactElement, "Update WorkflowVersion Artifact Element");
  45     }
  46
  47     @Override
  48     public Optional<ArtifactEntity> get(String id, String versionId) {
  49         SessionContext context = createSessionContext();
  50         ElementContext elementContext = new ElementContext(id, versionId);
  51
  52         Optional<Element> elementOptional =
  53                 zusammenAdaptor.getElementByName(context, elementContext, null, ElementType.Artifact.name());
  54
  55         if (!elementOptional.isPresent() || hasEmptyData(elementOptional.get().getData())) {
  56             return Optional.empty();
  57         }
  58
  59         Element artifactElement = elementOptional.get();
  60
  61         ArtifactEntity artifact = new ArtifactEntity(artifactElement.getInfo().getProperty(FILE_NAME_PROPERTY),
  62                 artifactElement.getData());
  63
  64         return Optional.of(artifact);
  65     }
  66
  67     @Override
  68     public void createStructure(String id, String versionId) {
  69         SessionContext context = createSessionContext();
  70         ElementContext elementContext = new ElementContext(id, versionId);
  71
  72         ZusammenElement artifactElement = buildStructuralElement(ElementType.Artifact, Action.CREATE);
  73         artifactElement.setData(new ByteArrayInputStream(EMPTY_DATA.getBytes()));
  74
  75         zusammenAdaptor
  76                 .saveElement(context, elementContext, artifactElement, "Create WorkflowVersion Artifact Element");
  77
  78     }
  79
  80     @Override
  81     public void delete(String id, String versionId) {
  82         SessionContext context = createSessionContext();
  83         ElementContext elementContext = new ElementContext(id, versionId);
  84
  85         ZusammenElement artifactElement = buildStructuralElement(ElementType.Artifact, Action.UPDATE);
  86         artifactElement.setData(new ByteArrayInputStream(EMPTY_DATA.getBytes()));
  87         artifactElement.getInfo().getProperties().remove(FILE_NAME_PROPERTY);
  88
  89         zusammenAdaptor
  90                 .saveElement(context, elementContext, artifactElement, "Update WorkflowVersion Artifact Element");
  91
  92     }
  93
  94     private boolean hasEmptyData(InputStream elementData) {
  95
  96         byte[] byteElementData;
  97         try {
  98             byteElementData = IOUtils.toByteArray(elementData);
  99         } catch (IOException ex) {
 100             return false;
 101         }
 102         return Arrays.equals(EMPTY_DATA.getBytes(), byteElementData);
 103     }
 104 }