Upgrade SDC from Titan to Janus Graph
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / be / components / merge / instance / ComponentInstanceArtifactsMerge.java
1 package org.openecomp.sdc.be.components.merge.instance;
2
3 import fj.data.Either;
4 import org.openecomp.sdc.be.components.impl.ArtifactsBusinessLogic;
5 import org.openecomp.sdc.be.model.*;
6 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.ToscaOperationFacade;
7 import org.openecomp.sdc.common.api.Constants;
8 import org.openecomp.sdc.exception.ResponseFormat;
9 import org.springframework.beans.factory.annotation.Autowired;
10
11 import java.util.HashMap;
12 import java.util.Map;
13 import java.util.Optional;
14 import java.util.stream.Collectors;
15
16 /**
17  * Created by chaya on 9/20/2017.
18  */
19 @org.springframework.stereotype.Component("ComponentInstanceArtifactsMerge")
20 public class ComponentInstanceArtifactsMerge implements ComponentInstanceMergeInterface {
21
22     @Autowired
23     ToscaOperationFacade toscaOperationFacade;
24
25     @Autowired
26     ArtifactsBusinessLogic artifactsBusinessLogic;
27
28     @Override
29     public void saveDataBeforeMerge(DataForMergeHolder dataHolder, Component containerComponent, ComponentInstance currentResourceInstance, Component originComponent) {
30         Map<String, ArtifactDefinition> componentInstancesDeploymentArtifacts = currentResourceInstance.safeGetDeploymentArtifacts();
31         Map<String, ArtifactDefinition> originalComponentDeploymentArtifacts = originComponent.getDeploymentArtifacts();
32         Map<String, ArtifactDefinition> deploymentArtifactsCreatedOnTheInstance = componentInstancesDeploymentArtifacts.entrySet()
33                 .stream()
34                 .filter(i -> !originalComponentDeploymentArtifacts.containsKey(i.getKey()))
35                 .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
36
37         dataHolder.setOrigComponentDeploymentArtifactsCreatedOnTheInstance(deploymentArtifactsCreatedOnTheInstance);
38
39         Map<String, ArtifactDefinition> componentInstancesInformationalArtifacts = currentResourceInstance.safeGetArtifacts();
40         Map<String, ArtifactDefinition> originalComponentInformationalArtifacts = originComponent.getArtifacts();
41         Map<String, ArtifactDefinition> informationalArtifactsCreatedOnTheInstance = componentInstancesInformationalArtifacts.entrySet()
42                 .stream()
43                 .filter(i -> !originalComponentInformationalArtifacts.containsKey(i.getKey()))
44                 .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
45         dataHolder.setOrigComponentInformationalArtifactsCreatedOnTheInstance(informationalArtifactsCreatedOnTheInstance);
46     }
47
48     private void addEsIdToArtifactJson(Map<String, Object> artifactJson, String origEsId) {
49         artifactJson.put(Constants.ARTIFACT_ES_ID, origEsId);
50     }
51
52     @Override
53     public Either<Component, ResponseFormat> mergeDataAfterCreate(User user, DataForMergeHolder dataHolder, Component updatedContainerComponent, String newInstanceId) {
54         Map<String, ArtifactDefinition> origInstanceDeploymentArtifactsCreatedOnTheInstance = dataHolder.getOrigComponentDeploymentArtifactsCreatedOnTheInstance();
55         Map<String, ArtifactDefinition> currentInstanceDeploymentArtifacts = updatedContainerComponent.safeGetComponentInstanceDeploymentArtifacts(newInstanceId);
56         Map<String, ArtifactDefinition> filteredDeploymentArtifactsToAdd = Optional.ofNullable(origInstanceDeploymentArtifactsCreatedOnTheInstance).orElse(new HashMap<>()).entrySet().stream()
57                 .filter(artifact -> noArtifactWithTheSameLabel(artifact.getValue().getArtifactLabel(), currentInstanceDeploymentArtifacts))
58                 .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
59         Map<String, ArtifactDefinition> origInstanceInformationalArtifactsCreatedOnTheInstance = dataHolder.getOrigComponentInformationalArtifactsCreatedOnTheInstance();
60         Map<String, ArtifactDefinition> currentInstanceInformationalArtifacts = updatedContainerComponent.safeGetComponentInstanceInformationalArtifacts(newInstanceId);
61         Map<String, ArtifactDefinition> filteredInformationalArtifactsToAdd = Optional.ofNullable(origInstanceInformationalArtifactsCreatedOnTheInstance).orElse(new HashMap<>()).entrySet().stream()
62                 .filter(artifact -> noArtifactWithTheSameLabel(artifact.getValue().getArtifactLabel(), currentInstanceInformationalArtifacts))
63                 .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
64         Map<String, ArtifactDefinition> allFilteredArtifactsToAdd = new HashMap<>();
65         allFilteredArtifactsToAdd.putAll(filteredDeploymentArtifactsToAdd);
66         allFilteredArtifactsToAdd.putAll(filteredInformationalArtifactsToAdd);
67
68         for (Map.Entry<String, ArtifactDefinition> currentArtifactDefinition :  allFilteredArtifactsToAdd.entrySet()) {
69             Map<String, Object> jsonForUpdateArtifact = artifactsBusinessLogic.buildJsonForUpdateArtifact(
70                     currentArtifactDefinition.getValue().getUniqueId(), 
71                     currentArtifactDefinition.getValue().getArtifactName(),
72                     currentArtifactDefinition.getValue().getArtifactType(), 
73                     currentArtifactDefinition.getValue().getArtifactGroupType(),
74                     currentArtifactDefinition.getValue().getArtifactLabel(), 
75                     currentArtifactDefinition.getValue().getArtifactDisplayName(),
76                     currentArtifactDefinition.getValue().getDescription(), 
77                     currentArtifactDefinition.getValue().getPayloadData(),
78                     null, currentArtifactDefinition.getValue().getListHeatParameters());
79             addEsIdToArtifactJson(jsonForUpdateArtifact, currentArtifactDefinition.getValue().getEsId());
80             Either<Either<ArtifactDefinition, Operation>, ResponseFormat> uploadArtifactToService =
81                     artifactsBusinessLogic.updateResourceInstanceArtifactNoContent(newInstanceId, updatedContainerComponent,
82                             user, jsonForUpdateArtifact, artifactsBusinessLogic.new ArtifactOperationInfo(
83                                     false, false, ArtifactsBusinessLogic.ArtifactOperationEnum.LINK), currentArtifactDefinition.getValue());
84             if (uploadArtifactToService.isRight()) {
85                 return Either.right(uploadArtifactToService.right().value());
86             }
87         }
88         return Either.left(updatedContainerComponent);
89     }
90
91     private boolean noArtifactWithTheSameLabel(String artifactLabel, Map<String, ArtifactDefinition> currDeploymentArtifacts) {
92         for (Map.Entry<String, ArtifactDefinition> artifact : currDeploymentArtifacts.entrySet()) {
93             if (artifact.getValue().getArtifactLabel().equals(artifactLabel)) {
94                 return false;
95             }
96         }
97         return true;
98     }
99 }