gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.vendorsoftwareproduct.impl.orchestration.process;
   2
   3 import org.apache.commons.collections4.MapUtils;
   4 import org.openecomp.core.converter.ToscaConverter;
   5 import org.openecomp.core.impl.ToscaConverterImpl;
   6 import org.openecomp.core.utilities.file.FileContentHandler;
   7 import org.openecomp.core.utilities.orchestration.OnboardingTypesEnum;
   8 import org.openecomp.core.validation.util.MessageContainerUtil;
   9 import org.openecomp.sdc.common.errors.CoreException;
  10 import org.openecomp.sdc.datatypes.error.ErrorLevel;
  11 import org.openecomp.sdc.datatypes.error.ErrorMessage;
  12 import org.openecomp.sdc.heat.datatypes.structure.HeatStructureTree;
  13 import org.openecomp.sdc.heat.services.tree.ToscaTreeManager;
  14 import org.openecomp.sdc.logging.api.Logger;
  15 import org.openecomp.sdc.logging.api.LoggerFactory;
  16 import org.openecomp.sdc.logging.messages.AuditMessages;
  17 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
  18 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentMonitoringUploadEntity;
  19 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.OrchestrationTemplateCandidateData;
  20 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessEntity;
  21 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.VspDetails;
  22 import org.openecomp.sdc.vendorsoftwareproduct.factory.CandidateServiceFactory;
  23 import org.openecomp.sdc.vendorsoftwareproduct.impl.orchestration.OrchestrationUtil;
  24 import org.openecomp.sdc.vendorsoftwareproduct.services.filedatastructuremodule.CandidateService;
  25 import org.openecomp.sdc.vendorsoftwareproduct.types.OrchestrationTemplateActionResponse;
  26 import org.openecomp.sdc.vendorsoftwareproduct.types.UploadFileResponse;
  27 import org.openecomp.sdc.versioning.dao.types.Version;
  28
  29 import java.io.ByteArrayInputStream;
  30 import java.io.IOException;
  31 import java.util.ArrayList;
  32 import java.util.Collection;
  33 import java.util.HashMap;
  34 import java.util.List;
  35 import java.util.Map;
  36 import java.util.Optional;
  37
  38 public class OrchestrationTemplateProcessCsarHandler implements OrchestrationTemplateProcessHandler {
  39
  40   private static Logger logger =
  41       LoggerFactory.getLogger(OrchestrationTemplateProcessCsarHandler.class);
  42   private ToscaConverter toscaConverter = new ToscaConverterImpl();
  43   private CandidateService candidateService =
  44       CandidateServiceFactory.getInstance().createInterface();
  45   ToscaTreeManager toscaTreeManager = new ToscaTreeManager();
  46
  47   @Override
  48   public OrchestrationTemplateActionResponse process(VspDetails vspDetails,
  49                                                      OrchestrationTemplateCandidateData candidateData,
  50                                                      String user) {
  51     String vspId = vspDetails.getId();
  52     Version version = vspDetails.getVersion();
  53     logger.audit(AuditMessages.AUDIT_MSG + AuditMessages.CSAR_VALIDATION_STARTED + vspId);
  54     OrchestrationTemplateActionResponse response = new OrchestrationTemplateActionResponse();
  55     UploadFileResponse uploadFileResponse = new UploadFileResponse();
  56     Optional<FileContentHandler> fileContent =
  57         OrchestrationUtil
  58             .getFileContentMap(
  59                 OnboardingTypesEnum.CSAR, uploadFileResponse, candidateData.getContentData().array());
  60
  61     if(fileContent.isPresent()){
  62       try {
  63         FileContentHandler fileContentHandler = fileContent.get();
  64         processCsar(vspId, version, fileContentHandler, candidateData, response);
  65       } catch (CoreException e){
  66         logger.error(e.getMessage());
  67         throw e;
  68       } catch (Exception e){
  69         logger.error(e.getMessage());
  70       }
  71     }
  72
  73     return response;
  74   }
  75
  76   private void processCsar(String vspId, Version version,
  77                            FileContentHandler fileContentHandler,
  78                            OrchestrationTemplateCandidateData candidateData,
  79                            OrchestrationTemplateActionResponse response) throws IOException {
  80
  81
  82     response.setFileNames(new ArrayList<>(fileContentHandler.getFileList()));
  83     Map<String, List<ErrorMessage>> errors = validateCsar(fileContentHandler, response);
  84     if(!isValid(errors)){
  85       return;
  86     }
  87
  88     HeatStructureTree tree = toscaTreeManager.getTree();
  89
  90     Map<String, String> componentsQuestionnaire = new HashMap<>();
  91     Map<String, Map<String, String>> componentNicsQuestionnaire = new HashMap<>();
  92     Map<String, Collection<ComponentMonitoringUploadEntity>> componentMibList = new HashMap<>();
  93     Map<String, Collection<ProcessEntity>> processes = new HashMap<>();
  94     Map<String, ProcessEntity> processArtifact = new HashMap<>();
  95
  96     OrchestrationUtil.backupComponentsQuestionnaireBeforeDelete(vspId,
  97         version, componentsQuestionnaire,
  98         componentNicsQuestionnaire, componentMibList, processes, processArtifact);
  99
 100     Optional<ByteArrayInputStream> zipByteArrayInputStream = candidateService
 101         .fetchZipFileByteArrayInputStream(vspId, candidateData, null, OnboardingTypesEnum.CSAR, errors);
 102
 103     OrchestrationUtil.deleteUploadDataAndContent(vspId, version);
 104     OrchestrationUtil.saveUploadData(
 105         vspId, version, zipByteArrayInputStream.get(), fileContentHandler, tree);
 106
 107     ToscaServiceModel toscaServiceModel = toscaConverter.convert(fileContentHandler);
 108     OrchestrationUtil.saveServiceModel(vspId, version, toscaServiceModel, toscaServiceModel);
 109
 110   }
 111
 112   private void addFiles(FileContentHandler fileContentHandler){
 113     for(Map.Entry<String, byte[]> fileEntry : fileContentHandler.getFiles().entrySet()){
 114       toscaTreeManager.addFile(fileEntry.getKey(), fileEntry.getValue());
 115     }
 116   }
 117
 118   private Map<String, List<ErrorMessage>> validateCsar(FileContentHandler fileContentHandler,
 119                                          OrchestrationTemplateActionResponse response){
 120
 121
 122     Map<String, List<ErrorMessage>> errors = new HashMap<>();
 123     addFiles(fileContentHandler);
 124     toscaTreeManager.createTree();
 125     toscaTreeManager.addErrors(errors);
 126     //todo - add tosca validation here to the existing validation framework
 127     return errors;
 128   }
 129
 130   private boolean isValid(Map<String, List<ErrorMessage>> errors){
 131     return MapUtils.isEmpty(MessageContainerUtil.getMessageByLevel(ErrorLevel.ERROR, errors));
 132   }
 133 }