openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/main/java/org/openecomp/sdc/enrichment/impl/external/artifact/ProcessArtifactEnricher.java

   1 package org.openecomp.sdc.enrichment.impl.external.artifact;
   2
   3 import org.openecomp.core.enrichment.types.ArtifactCategory;
   4 import org.openecomp.core.enrichment.types.ComponentProcessInfo;
   5 import org.openecomp.core.model.dao.EnrichedServiceModelDao;
   6 import org.openecomp.core.model.dao.EnrichedServiceModelDaoFactory;
   7 import org.openecomp.core.model.types.ServiceArtifact;
   8 import org.openecomp.core.utilities.file.FileUtils;
   9 import org.openecomp.sdc.datatypes.error.ErrorMessage;
  10 import org.openecomp.sdc.enrichment.EnrichmentInfo;
  11 import org.openecomp.sdc.enrichment.inter.ExternalArtifactEnricherInterface;
  12 import org.openecomp.sdc.logging.context.impl.MdcDataDebugMessage;
  13 import org.openecomp.sdc.vendorsoftwareproduct.dao.ProcessDao;
  14 import org.openecomp.sdc.vendorsoftwareproduct.dao.ProcessDaoFactory;
  15 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDao;
  16 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDaoFactory;
  17 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
  18 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessEntity;
  19 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessType;
  20 import org.openecomp.sdc.versioning.dao.types.Version;
  21
  22 import java.io.File;
  23 import java.io.IOException;
  24 import java.util.Collection;
  25 import java.util.HashMap;
  26 import java.util.List;
  27 import java.util.Map;
  28
  29 public class ProcessArtifactEnricher implements ExternalArtifactEnricherInterface {
  30
  31   private VendorSoftwareProductDao vendorSoftwareProductDao;
  32   //private ProcessArtifactDao processArtifactDao;
  33   private ProcessDao processDao;
  34   private static MdcDataDebugMessage mdcDataDebugMessage = new MdcDataDebugMessage();
  35   private EnrichedServiceModelDao enrichedServiceModelDao;
  36
  37   @Override
  38   public Map<String, List<ErrorMessage>> enrich(EnrichmentInfo enrichmentInfo) throws IOException {
  39     Map<String, List<ErrorMessage>> errors = new HashMap<>();
  40     String vspId = enrichmentInfo.getKey();
  41     Version version = enrichmentInfo.getVersion();
  42
  43     Collection<ComponentEntity> components =
  44         getVendorSoftwareProductDao().listComponents(vspId, version);
  45     components.stream()
  46         .forEach(componentEntry -> errors.putAll(enrichComponent(componentEntry,
  47             vspId, version)));
  48
  49     return errors;
  50   }
  51
  52   Map<String, List<ErrorMessage>> enrichComponent(ComponentEntity componentEntry, String vspId,
  53                                                   Version version) {
  54     mdcDataDebugMessage.debugEntryMessage("LifeCycleOperationArtifactEnricher vspId ",
  55         vspId);
  56
  57     Map<String, List<ErrorMessage>> errors = new HashMap<>();
  58     enrichComponentProcessArtifact(componentEntry, vspId, version, errors);
  59
  60     mdcDataDebugMessage.debugExitMessage("LifeCycleOperationArtifactEnricher vspId ",
  61         vspId);
  62     return errors;
  63   }
  64
  65   void enrichComponentProcessArtifact(ComponentEntity componentEntity,
  66                                       String vspId, Version version,
  67                                       Map<String, List<ErrorMessage>> errors) {
  68
  69
  70     mdcDataDebugMessage.debugEntryMessage(null, null);
  71
  72     String componentId = componentEntity.getId();
  73     ProcessEntity processEntity = new ProcessEntity(vspId, version, componentId, null);
  74     final Collection<ProcessEntity> processes = getProcessDao().list(processEntity);
  75
  76     /*processes.stream()
  77         .filter(entity -> entity.getType().equals(ProcessType.Lifecycle_Operations))
  78         .forEach(entity -> {
  79           ProcessArtifactEntity artifactEntity = new ProcessArtifactEntity(vspId, version,
  80                   componentId, entity.getId());*/
  81
  82     processes.stream()
  83         .forEach(entity -> {
  84           ProcessEntity artifactEntity = new ProcessEntity(vspId, version,
  85               componentId, entity.getId());
  86
  87           ProcessEntity artifactProcessEntity = getProcessDao().get(artifactEntity);
  88           //ProcessArtifactEntity artifact = getProcessArtifactDao().get(artifactEntity);
  89           if (artifactProcessEntity != null && ProcessType.Lifecycle_Operations.equals(
  90               artifactProcessEntity.getType())
  91               && artifactProcessEntity.getArtifactName() != null ) {
  92             String componentName = componentEntity.getComponentCompositionData().getName();
  93             String path = componentName + File.separator
  94                 + ArtifactCategory.DEPLOYMENT.getDisplayName() + File.separator
  95                 + "Lifecycle Operations" + File.separator + artifactProcessEntity.getArtifactName();
  96
  97             ComponentProcessInfo componentProcessInfo = new ComponentProcessInfo();
  98             componentProcessInfo.setName(path);
  99             componentProcessInfo.setContent(artifactProcessEntity.getArtifact().array());
 100
 101             ServiceArtifact processServiceArtifact = new ServiceArtifact();
 102             processServiceArtifact.setVspId(vspId);
 103             processServiceArtifact.setVersion(version);
 104             enrichServiceArtifact(componentProcessInfo, processServiceArtifact, errors);
 105           }
 106         });
 107
 108     mdcDataDebugMessage.debugExitMessage(null, null);
 109   }
 110
 111   void enrichServiceArtifact(ComponentProcessInfo componentProcessInfo,
 112                              ServiceArtifact processServiceArtifact,
 113                              Map<String, List<ErrorMessage>> errors) {
 114
 115
 116     mdcDataDebugMessage.debugEntryMessage(null, null);
 117
 118     processServiceArtifact.setName(componentProcessInfo.getName());
 119     processServiceArtifact.setContentData(FileUtils.toByteArray(componentProcessInfo.getContent()));
 120     getEnrichedServiceModelDao().storeExternalArtifact(processServiceArtifact);
 121     mdcDataDebugMessage.debugExitMessage(null, null);
 122   }
 123
 124   private VendorSoftwareProductDao getVendorSoftwareProductDao() {
 125     if (vendorSoftwareProductDao == null) {
 126       vendorSoftwareProductDao = VendorSoftwareProductDaoFactory.getInstance().createInterface();
 127     }
 128     return vendorSoftwareProductDao;
 129   }
 130
 131   private ProcessDao getProcessDao() {
 132     if (processDao == null) {
 133       processDao = ProcessDaoFactory.getInstance().createInterface();
 134     }
 135     return processDao;
 136   }
 137
 138   private EnrichedServiceModelDao getEnrichedServiceModelDao() {
 139
 140     if (enrichedServiceModelDao == null) {
 141       enrichedServiceModelDao = EnrichedServiceModelDaoFactory.getInstance().createInterface();
 142     }
 143     return enrichedServiceModelDao;
 144   }
 145 }