[SDC-29] rebase continue work to align source
[sdc.git] / asdctool / src / main / java / org / openecomp / sdc / asdctool / impl / migration / v1707 / jsonmodel / ComponentMigration.java
index c9212f9..63eb887 100644 (file)
@@ -1,20 +1,20 @@
 package org.openecomp.sdc.asdctool.impl.migration.v1707.jsonmodel;
 
-import fj.Function;
 import fj.data.Either;
-import org.openecomp.sdc.asdctool.impl.migration.v1707.MigrationUtils;
-import org.openecomp.sdc.asdctool.impl.migration.v1707.jsonmodel.relations.FulfilledCapabilitiesMigrationService;
-import org.openecomp.sdc.asdctool.impl.migration.v1707.jsonmodel.relations.FulfilledRequirementsMigrationService;
 import org.openecomp.sdc.asdctool.impl.migration.v1707.jsonmodel.relations.RequirementsCapabilitiesMigrationService;
 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
 import org.openecomp.sdc.be.model.Component;
 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
+import org.openecomp.sdc.be.model.operations.migration.MigrationMalformedDataLogger;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import javax.annotation.Resource;
 
+import java.util.List;
+import java.util.stream.Collectors;
+
 import static org.openecomp.sdc.asdctool.impl.migration.v1707.MigrationUtils.handleError;
 
 public abstract class ComponentMigration <T extends Component> extends JsonModelMigration<T> {
@@ -27,10 +27,15 @@ public abstract class ComponentMigration <T extends Component> extends JsonModel
     @Resource(name = "req-cap-mig-service")
     RequirementsCapabilitiesMigrationService<T> requirementsCapabilitiesMigrationService;
 
+    @Resource(name = "invariant-uuid-resolver")
+    private InvariantUUIDResolver <T> invariantUUIDResolver;
+
     @Override
-    Either<T, StorageOperationStatus> save(T element) {
+    boolean save(T element) {
         LOGGER.debug(String.format("creating component %s in new graph", element.getName()));
-        return toscaOperations.createToscaComponent(element).right().map(err -> handleError(err, String.format("failed to create component %s.", element.getName())));
+        return toscaOperations.createToscaComponent(element)
+                .either(savedNode -> true,
+                        err -> handleError(String.format("failed to create component %s with id %s. reason: %s", element.getName(), element.getUniqueId(), err.name())));
 
     }
 
@@ -45,4 +50,26 @@ public abstract class ComponentMigration <T extends Component> extends JsonModel
         return StorageOperationStatus.NOT_FOUND;
     }
 
+    @Override
+    void doPreMigrationOperation(List<T> elements) {
+        setMissingInvariantUids(elements);
+    }
+
+    //some invariants uids are missing in production
+    private void setMissingInvariantUids(List<T> components) {
+        List<T> missingInvariantCmpts = getComponentsWithNoInvariantUUIDs(components);
+        for (T missingInvariantCmpt : missingInvariantCmpts) {
+            missingInvariantCmpt.setInvariantUUID(invariantUUIDResolver.resolveInvariantUUID(components, missingInvariantCmpt));
+        }
+    }
+
+    private List<T> getComponentsWithNoInvariantUUIDs(List<T> components) {
+        List<T> cmptsWithoutInvariant = components.stream().filter(c -> c.getInvariantUUID() == null).collect(Collectors.toList());
+        if (!cmptsWithoutInvariant.isEmpty()) {
+            cmptsWithoutInvariant.forEach(cmpt -> MigrationMalformedDataLogger.logMalformedDataMsg(String.format("component %s is missing invariant uuid", cmpt.getUniqueId())));
+        }
+        return cmptsWithoutInvariant;
+    }
+
+
 }