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