130405be1d99def651cde8d3c04eb33125cb7ea8
[sdc.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.healing.healers;
22
23 import org.openecomp.core.utilities.json.JsonSchemaDataGenerator;
24 import org.openecomp.sdc.common.utils.SdcCommon;
25 import org.openecomp.sdc.healing.interfaces.Healer;
26 import org.openecomp.sdc.logging.context.impl.MdcDataDebugMessage;
27 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDao;
28 import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDaoFactory;
29 import org.openecomp.sdc.vendorsoftwareproduct.dao.NetworkDao;
30 import org.openecomp.sdc.vendorsoftwareproduct.dao.NetworkDaoFactory;
31 import org.openecomp.sdc.vendorsoftwareproduct.dao.NicDao;
32 import org.openecomp.sdc.vendorsoftwareproduct.dao.NicDaoFactory;
33 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDao;
34 import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDaoFactory;
35 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
36 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.CompositionEntity;
37 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NetworkEntity;
38 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.NicEntity;
39 import org.openecomp.sdc.vendorsoftwareproduct.services.schemagenerator.SchemaGenerator;
40 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.CompositionEntityType;
41 import org.openecomp.sdc.vendorsoftwareproduct.types.schemagenerator.SchemaTemplateContext;
42 import org.openecomp.sdc.versioning.dao.types.Version;
43
44 import java.util.Collection;
45 import java.util.Map;
46 import java.util.Objects;
47
48 public class SubEntitiesQuestionnaireHealer implements Healer {
49   private static Version version00 = new Version(0, 0);
50   private MdcDataDebugMessage mdcDataDebugMessage = new MdcDataDebugMessage();
51
52   private static final VendorSoftwareProductDao vendorSoftwareProductDao =
53       VendorSoftwareProductDaoFactory.getInstance().createInterface();
54   private static ComponentDao componentDao = ComponentDaoFactory.getInstance().createInterface();
55   private static NicDao nicDao = NicDaoFactory.getInstance().createInterface();
56   private static NetworkDao networkDao = NetworkDaoFactory.getInstance().createInterface();
57
58   @Override
59   public Object heal(Map<String, Object> healingParams) throws Exception {
60
61
62     mdcDataDebugMessage.debugEntryMessage(null);
63
64     String vspId = (String) healingParams.get(SdcCommon.VSP_ID);
65     Version version = version00.equals(healingParams.get(SdcCommon.VERSION)) ? new Version
66         (0, 1)
67         : (Version) healingParams.get(SdcCommon.VERSION);
68
69     Collection<ComponentEntity> componentEntities =
70         componentDao.listCompositionAndQuestionnaire(vspId, version);
71
72     networkDao.list(new NetworkEntity(vspId, version, null));
73
74     Collection<NicEntity> nicEntities = vendorSoftwareProductDao.listNicsByVsp(vspId, version);
75
76     healCompositionEntityQuestionnaire(componentEntities, version, CompositionEntityType.component);
77     healCompositionEntityQuestionnaire(nicEntities, version, CompositionEntityType.nic);
78
79     mdcDataDebugMessage.debugExitMessage(null);
80     return new Object();
81   }
82
83
84   private void healCompositionEntityQuestionnaire(Collection
85                                                       compositionEntities,
86                                                   Version newVersion, CompositionEntityType type) {
87
88
89     mdcDataDebugMessage.debugEntryMessage(null);
90
91     for (Object entity : compositionEntities) {
92       CompositionEntity compositionEntity = (CompositionEntity) entity;
93       if (Objects.isNull(compositionEntity.getQuestionnaireData()) ||
94           "".equals(compositionEntity.getQuestionnaireData())) {
95         compositionEntity.setVersion(newVersion);
96         updateNullQuestionnaire(compositionEntity, type);
97       }
98     }
99
100     mdcDataDebugMessage.debugExitMessage(null);
101   }
102
103   private void updateNullQuestionnaire(CompositionEntity entity,
104                                        CompositionEntityType type) {
105
106
107     mdcDataDebugMessage.debugEntryMessage(null);
108
109     entity.setQuestionnaireData(
110         new JsonSchemaDataGenerator(SchemaGenerator
111             .generate(SchemaTemplateContext.questionnaire, type,
112                 null)).generateData());
113
114     switch (type) {
115       case component:
116         ComponentEntity component = (ComponentEntity) entity;
117         componentDao.updateQuestionnaireData(component.getVspId(), component
118             .getVersion(), component.getId(), component.getQuestionnaireData());
119         break;
120
121       case nic:
122         NicEntity nic = (NicEntity) entity;
123         nicDao.updateQuestionnaireData(nic.getVspId(), nic.getVersion(), nic.getComponentId(),
124             nic.getId(), nic.getQuestionnaireData());
125         break;
126     }
127     mdcDataDebugMessage.debugExitMessage(null);
128   }
129
130 }