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