[SDC] Onboarding 1710 rebase.
[sdc.git] / openecomp-be / lib / openecomp-healing-lib / openecomp-sdc-healing-impl / src / main / java / org / openecomp / sdc / healing / healers / HeatToToscaTranslationHealer.java
1 package org.openecomp.sdc.healing.healers;
2
3
4 import org.openecomp.core.model.dao.ServiceModelDao;
5 import org.openecomp.core.model.dao.ServiceModelDaoFactory;
6 import org.openecomp.core.model.dao.ServiceTemplateDaoFactory;
7 import org.openecomp.core.model.dao.ServiceTemplateDaoInter;
8 import org.openecomp.core.model.types.ServiceElement;
9 import org.openecomp.core.translator.datatypes.TranslatorOutput;
10 import org.openecomp.core.utilities.file.FileContentHandler;
11 import org.openecomp.sdc.common.utils.CommonUtil;
12 import org.openecomp.sdc.common.utils.SdcCommon;
13 import org.openecomp.sdc.healing.interfaces.Healer;
14 import org.openecomp.sdc.logging.context.impl.MdcDataDebugMessage;
15 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
16 import org.openecomp.sdc.translator.services.heattotosca.HeatToToscaUtil;
17 import org.openecomp.sdc.vendorsoftwareproduct.dao.OrchestrationTemplateDao;
18 import org.openecomp.sdc.vendorsoftwareproduct.dao.OrchestrationTemplateDaoFactory;
19 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.UploadDataEntity;
20 import org.openecomp.sdc.versioning.dao.types.Version;
21
22 import java.util.Map;
23 import java.util.Objects;
24 import java.util.Optional;
25
26 public class HeatToToscaTranslationHealer implements Healer  {
27
28   private static final OrchestrationTemplateDao orchestrationTemplateDao =
29       OrchestrationTemplateDaoFactory.getInstance().createInterface();
30   private static final ServiceModelDao<ToscaServiceModel, ServiceElement> serviceModelDao =
31       ServiceModelDaoFactory.getInstance().createInterface();
32   public static final ServiceTemplateDaoInter
33       templateDao = ServiceTemplateDaoFactory.getInstance().createInterface();
34   private static MdcDataDebugMessage mdcDataDebugMessage = new MdcDataDebugMessage();
35
36   public HeatToToscaTranslationHealer(){
37
38   }
39
40   @Override
41   public Object heal(Map<String, Object> healingParams) throws Exception {
42     mdcDataDebugMessage.debugEntryMessage(null, null);
43     String vspId = (String) healingParams.get(SdcCommon.VSP_ID);
44     Version version = (Version) healingParams.get(SdcCommon.VERSION);
45     String user = (String) healingParams.get(SdcCommon.USER);
46     UploadDataEntity uploadData =
47         orchestrationTemplateDao.getOrchestrationTemplate(vspId, version);
48
49     if (Objects.isNull(uploadData) || Objects.isNull(uploadData.getContentData())) {
50       return Optional.empty();
51     }
52
53     FileContentHandler fileContentHandler;
54     TranslatorOutput translatorOutput;
55     try {
56       fileContentHandler = CommonUtil.validateAndUploadFileContent(uploadData
57           .getContentData().array());
58       translatorOutput =
59           HeatToToscaUtil.loadAndTranslateTemplateData(fileContentHandler);
60     }catch (Exception e){
61       return Optional.empty();
62     }
63
64     if(Objects.isNull(translatorOutput)){
65       return Optional.empty();
66     }
67
68     if (translatorOutput != null && translatorOutput.getToscaServiceModel() == null) {
69       return Optional.empty();
70     }
71       //templateDao.deleteAll(vspId, version);
72     serviceModelDao.deleteAll(vspId,version);
73     serviceModelDao.storeServiceModel(vspId, version,
74         translatorOutput.getToscaServiceModel());
75     mdcDataDebugMessage.debugExitMessage("VSP id", vspId);
76
77     return translatorOutput;
78
79   }
80 }