4a0bf13f41290171c7ee9e01cf81001240708cca
[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.impl;
22
23 import org.openecomp.core.utilities.CommonMethods;
24 import org.openecomp.core.utilities.file.FileUtils;
25 import org.openecomp.core.utilities.json.JsonUtil;
26 import org.openecomp.sdc.common.errors.CoreException;
27 import org.openecomp.sdc.common.errors.ErrorCode;
28 import org.openecomp.sdc.common.errors.Messages;
29 import org.openecomp.sdc.common.session.SessionContext;
30 import org.openecomp.sdc.common.session.SessionContextProviderFactory;
31 import org.openecomp.sdc.datatypes.model.ItemType;
32 import org.openecomp.sdc.healing.api.HealingManager;
33 import org.openecomp.sdc.healing.dao.HealingDao;
34 import org.openecomp.sdc.healing.interfaces.Healer;
35 import org.openecomp.sdc.healing.types.HealCode;
36 import org.openecomp.sdc.healing.types.HealerType;
37 import org.openecomp.sdc.versioning.VersioningManager;
38 import org.openecomp.sdc.versioning.dao.types.Version;
39 import org.openecomp.sdc.versioning.dao.types.VersionStatus;
40 import org.openecomp.sdc.versioning.types.VersionCreationMethod;
41
42 import java.lang.reflect.InvocationTargetException;
43 import java.util.ArrayList;
44 import java.util.Collection;
45 import java.util.List;
46 import java.util.Map;
47 import java.util.Objects;
48 import java.util.Optional;
49 import java.util.stream.Collectors;
50
51 /**
52  * Created by Talio on 11/29/2016.
53  */
54 public class HealingManagerImpl implements HealingManager {
55
56   private static final String HEALERS_BY_ENTITY_TYPE_FILE = "entityHealingConfiguration.json";
57   private static final String HEALING_USER_SUFFIX = "_healer";
58   private static final String PUBLIC_USER = "public";
59
60   private HealingDao healingDao;
61   private VersioningManager versioningManager;
62
63   public HealingManagerImpl(VersioningManager versioningManager, HealingDao healingDao) {
64     this.versioningManager = versioningManager;
65     this.healingDao = healingDao;
66   }
67
68   @Override
69   public Optional<Version> healItemVersion(String itemId, Version version, ItemType itemType,
70                                            boolean force) {
71     String user = getUser();
72     if (force || isPrivateHealingNeededByFlag(itemId, version.getId(), user)) {
73       version = versioningManager.get(itemId, version);
74       Version origVersion = version;
75       if (version.getStatus() == VersionStatus.Certified) {
76         Optional<Version> newVersion = createNewVersion(itemId, version);
77         if (!newVersion.isPresent()) {
78           // do NOT turn off flag here (in thought of saving version calculate performance next
79           // time) because maybe next time the next version will be available (due to deletion of
80           // the taken one)
81           return Optional.empty();
82         }
83         version = newVersion.get();
84       }
85
86       doHeal(itemId, version, origVersion, itemType, user, force);
87       return Optional.of(version);
88     }
89     return Optional.empty();
90   }
91
92   private void doHeal(String itemId, Version version, Version origVersion,
93                       ItemType itemType, String user, boolean force) {
94     Optional<String> privateFailureMessages =
95         healPrivate(itemId, version, origVersion, getItemHealers(itemType), user);
96
97     Optional<String> publicFailureMessages =
98         force || origVersion.getStatus() == VersionStatus.Certified ||
99             isPublicHealingNeededByFlag(itemId, origVersion.getId())
100             ? healPublic(itemId, version, origVersion, getItemHealers(itemType), user)
101             : Optional.empty();
102
103     if (privateFailureMessages.isPresent() || publicFailureMessages.isPresent()) {
104       throw new CoreException(new ErrorCode.ErrorCodeBuilder().withMessage(
105           publicFailureMessages.orElse("") + " " + privateFailureMessages.orElse(""))
106           .build());
107     }
108   }
109
110   private Optional<String> healPrivate(String itemId, Version version, Version origVersion,
111                                        Map<String, Map<String, String>> itemHealers, String user) {
112     if (origVersion.getStatus() != VersionStatus.Certified) {
113       itemHealers.remove(HealerType.structure.name());
114     }
115
116     Optional<String> privateHealingFailureMessages = executeHealers(itemId, version, itemHealers);
117     markAsHealed(itemId, origVersion.getId(), user);
118     return privateHealingFailureMessages;
119   }
120
121   private Optional<String> healPublic(String itemId, Version version, Version origVersion,
122                                       Map<String, Map<String, String>> itemHealers, String user) {
123     Optional<String> healingFailureMessages = origVersion.getStatus() == VersionStatus.Certified
124         ? Optional.empty()
125         : healPublic(itemId, version, itemHealers, user);
126
127     markAsHealed(itemId, origVersion.getId(), PUBLIC_USER);
128     return healingFailureMessages;
129   }
130
131   private Optional<String> healPublic(String itemId, Version version,
132                                       Map<String, Map<String, String>> itemHealers, String user) {
133     SessionContext context =
134         SessionContextProviderFactory.getInstance().createInterface().get();
135     SessionContextProviderFactory.getInstance().createInterface().create(user
136         + HEALING_USER_SUFFIX,context.getTenant());
137
138     versioningManager.sync(itemId, version);
139
140     Optional<String> healingFailureMessages = executeHealers(itemId, version, itemHealers);
141     Version publicVersion = versioningManager.get(itemId, version);
142
143     if (Objects.nonNull(publicVersion.getState()) && publicVersion.getState().isDirty()) {
144       versioningManager.publish(itemId, version, "Healing vsp");
145     }
146
147     SessionContextProviderFactory.getInstance().createInterface().create(user, context.getTenant());
148     return healingFailureMessages;
149   }
150
151   private Optional<Version> createNewVersion(String itemId, Version version) {
152     Version newVersion = new Version();
153     newVersion.setBaseId(version.getId());
154     try {
155       return Optional.of(versioningManager.create(itemId, newVersion, VersionCreationMethod.major));
156     } catch (Exception e) {
157       return Optional.empty();
158     }
159   }
160
161   private Optional<String> executeHealers(String itemId, Version version,
162                                           Map<String, Map<String, String>> itemHealers) {
163     List<String> healers = itemHealers.values().stream()
164         .map(Map::values)
165         .flatMap(Collection::stream)
166         .collect(Collectors.toList());
167
168     List<String> healingFailureMessages = new ArrayList<>();
169     for (String implClassName : healers) {
170       executeHealer(itemId, version, implClassName, healingFailureMessages);
171     }
172
173     return healingFailureMessages.isEmpty()
174         ? Optional.empty()
175         : Optional.of(CommonMethods.listToSeparatedString(healingFailureMessages, '\n'));
176   }
177
178
179   private Object executeHealer(String itemId, Version version, String healerClassName,
180                                List<String> healingFailureMessages) {
181     Healer healer;
182     try {
183       healer = getHealerImplInstance(healerClassName);
184     } catch (Exception e) {
185       healingFailureMessages
186           .add(String.format(Messages.CANT_LOAD_HEALING_CLASS.getErrorMessage(),
187               healerClassName));
188       return null;
189     }
190
191     try {
192       return healer.heal(itemId, version);
193     } catch (Exception e) {
194       healingFailureMessages.add(e.getMessage() + " ,healer name :" + healerClassName);
195     }
196     return null;
197   }
198
199   private boolean isPrivateHealingNeededByFlag(String itemId, String version, String user) {
200     Optional<Boolean> userHealingFlag = getHealingFlag(itemId, version, user);
201     return userHealingFlag.orElseGet(() -> isPublicHealingNeededByFlag(itemId, version));
202   }
203
204   private boolean isPublicHealingNeededByFlag(String itemId, String versionId) {
205     Optional<Boolean> publicHealingFlag = getHealingFlag(itemId, versionId, PUBLIC_USER);
206     return publicHealingFlag.isPresent() && publicHealingFlag.get();
207   }
208
209   private Optional<Boolean> getHealingFlag(String itemId, String version, String user) {
210     return healingDao.getItemHealingFlag(user, itemId, version);
211   }
212
213   private void markAsHealed(String itemId, String versionId, String user) {
214     healingDao.setItemHealingFlag(false, user, itemId, versionId);
215   }
216
217   private Map<String, Map<String, String>> getItemHealers(ItemType itemType) {
218     Map healingConfig = FileUtils
219         .readViaInputStream(HEALERS_BY_ENTITY_TYPE_FILE,
220             stream -> JsonUtil.json2Object(stream, Map.class));
221     return (Map<String, Map<String, String>>) healingConfig.get(itemType.name());
222   }
223
224   private Healer getHealerImplInstance(String implClassName)
225       throws InstantiationException, IllegalAccessException, InvocationTargetException,
226       NoSuchMethodException, ClassNotFoundException {
227     return (Healer) Class.forName(implClassName).getConstructor().newInstance();
228   }
229
230   private String getUser() {
231     return SessionContextProviderFactory.getInstance().createInterface().get().getUser()
232         .getUserId();
233   }
234 }