[SDC-29] rebase continue work to align source
[sdc.git] / asdctool / src / main / java / org / openecomp / sdc / asdctool / impl / migration / v1707 / jsonmodel / ComponentMigration.java
1 package org.openecomp.sdc.asdctool.impl.migration.v1707.jsonmodel;
2
3 import fj.data.Either;
4 import org.openecomp.sdc.asdctool.impl.migration.v1707.jsonmodel.relations.RequirementsCapabilitiesMigrationService;
5 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
6 import org.openecomp.sdc.be.model.Component;
7 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
8 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
9 import org.openecomp.sdc.be.model.operations.migration.MigrationMalformedDataLogger;
10 import org.slf4j.Logger;
11 import org.slf4j.LoggerFactory;
12
13 import javax.annotation.Resource;
14
15 import java.util.List;
16 import java.util.stream.Collectors;
17
18 import static org.openecomp.sdc.asdctool.impl.migration.v1707.MigrationUtils.handleError;
19
20 public abstract class ComponentMigration <T extends Component> extends JsonModelMigration<T> {
21
22     private static Logger LOGGER = LoggerFactory.getLogger(ComponentMigration.class);
23
24     @Resource(name = "tosca-operation-facade")
25     private ToscaOperationFacade toscaOperations;
26
27     @Resource(name = "req-cap-mig-service")
28     RequirementsCapabilitiesMigrationService<T> requirementsCapabilitiesMigrationService;
29
30     @Resource(name = "invariant-uuid-resolver")
31     private InvariantUUIDResolver <T> invariantUUIDResolver;
32
33     @Override
34     boolean save(T element) {
35         LOGGER.debug(String.format("creating component %s in new graph", element.getName()));
36         return toscaOperations.createToscaComponent(element)
37                 .either(savedNode -> true,
38                         err -> handleError(String.format("failed to create component %s with id %s. reason: %s", element.getName(), element.getUniqueId(), err.name())));
39
40     }
41
42     @Override
43     Either<T, StorageOperationStatus> getElementFromNewGraph(T element) {
44         LOGGER.debug(String.format("checking if component %s already exists on new graph", element.getName()));
45         return toscaOperations.getToscaElement(element.getUniqueId(), JsonParseFlagEnum.ParseMetadata);
46     }
47
48     @Override
49     public StorageOperationStatus getNotFoundErrorStatus() {
50         return StorageOperationStatus.NOT_FOUND;
51     }
52
53     @Override
54     void doPreMigrationOperation(List<T> elements) {
55         setMissingInvariantUids(elements);
56     }
57
58     //some invariants uids are missing in production
59     private void setMissingInvariantUids(List<T> components) {
60         List<T> missingInvariantCmpts = getComponentsWithNoInvariantUUIDs(components);
61         for (T missingInvariantCmpt : missingInvariantCmpts) {
62             missingInvariantCmpt.setInvariantUUID(invariantUUIDResolver.resolveInvariantUUID(components, missingInvariantCmpt));
63         }
64     }
65
66     private List<T> getComponentsWithNoInvariantUUIDs(List<T> components) {
67         List<T> cmptsWithoutInvariant = components.stream().filter(c -> c.getInvariantUUID() == null).collect(Collectors.toList());
68         if (!cmptsWithoutInvariant.isEmpty()) {
69             cmptsWithoutInvariant.forEach(cmpt -> MigrationMalformedDataLogger.logMalformedDataMsg(String.format("component %s is missing invariant uuid", cmpt.getUniqueId())));
70         }
71         return cmptsWithoutInvariant;
72     }
73
74
75 }