catalog-be/src/main/java/org/openecomp/sdc/be/components/merge/capability/SimpleCapabilityResolver.java

   1 package org.openecomp.sdc.be.components.merge.capability;
   2
   3 import org.apache.commons.collections.CollectionUtils;
   4 import org.openecomp.sdc.be.components.merge.utils.MergeInstanceUtils;
   5 import org.openecomp.sdc.be.dao.utils.MapUtil;
   6 import org.openecomp.sdc.be.model.CapabilityDefinition;
   7 import org.openecomp.sdc.be.model.Component;
   8 import org.openecomp.sdc.be.model.ComponentInstance;
   9
  10 import java.util.*;
  11 import java.util.function.Predicate;
  12 import java.util.stream.Collectors;
  13
  14 @org.springframework.stereotype.Component
  15 public class SimpleCapabilityResolver implements CapabilityResolver {
  16
  17     private MergeInstanceUtils mergeInstanceUtils;
  18
  19     public SimpleCapabilityResolver(MergeInstanceUtils mergeInstanceUtils) {
  20         this.mergeInstanceUtils = mergeInstanceUtils;
  21     }
  22
  23     @Override
  24     public Map<CapabilityDefinition, CapabilityDefinition> resolvePrevCapToNewCapability(Component container, Component prevInstanceOrigNode, String cmptInstanceId, List<CapabilityDefinition> prevCapabilities) {
  25         Map<String, List<CapabilityDefinition>> newCapabilitiesByType = resolveInstanceCapabilities(container, cmptInstanceId).getCapabilities();
  26         Map<String, String> oldCapOwnerToNewOwner = mapOldToNewCapabilitiesOwnerIds(container, prevInstanceOrigNode, cmptInstanceId, prevCapabilities);
  27         return mapOldToNewCapabilities(prevCapabilities, newCapabilitiesByType, oldCapOwnerToNewOwner);
  28     }
  29
  30     @Override
  31     public Map<CapabilityDefinition, CapabilityDefinition> resolvePrevCapIdToNewCapability(ComponentInstance oldInstance, ComponentInstance currInstance) {
  32         Map<String, List<CapabilityDefinition>> newCapabilitiesByType = currInstance.getCapabilities();
  33         Map<String, String> oldCapOwnerToNewOwner = mergeInstanceUtils.mapOldToNewCapabilitiesOwnerIds(oldInstance, currInstance);
  34         List<CapabilityDefinition> prevCapabilities = oldInstance.getCapabilities().values().stream().flatMap(Collection::stream).collect(Collectors.toList());
  35         return mapOldToNewCapabilities(prevCapabilities, newCapabilitiesByType, oldCapOwnerToNewOwner);
  36     }
  37
  38     private Map<CapabilityDefinition, CapabilityDefinition> mapOldToNewCapabilities(List<CapabilityDefinition> prevCapabilities, Map<String, List<CapabilityDefinition>> newCapabilitiesByType, Map<String, String> oldCapOwnerToNewOwner) {
  39         Map<CapabilityDefinition, CapabilityDefinition> oldToNewCapability = prevCapabilities
  40                 .stream()
  41                 .collect(HashMap::new,
  42                         (resultMap, prevCap) -> mapOldToNewCapability(newCapabilitiesByType, oldCapOwnerToNewOwner, resultMap, prevCap),
  43                         HashMap::putAll);
  44         removeNotFoundNewCapabilities(oldToNewCapability);
  45         return oldToNewCapability;
  46     }
  47
  48     private CapabilityDefinition mapOldToNewCapability(Map<String, List<CapabilityDefinition>> newCapabilitiesByType, Map<String, String> oldCapOwnerToNewOwner, Map<CapabilityDefinition, CapabilityDefinition> resultMap, CapabilityDefinition prevCap) {
  49         return resultMap.put(prevCap, findCurrCapability(newCapabilitiesByType, prevCap, oldCapOwnerToNewOwner.get(prevCap.getOwnerId())));
  50     }
  51
  52     private Map<String, String> mapOldToNewCapabilitiesOwnerIds(Component container, Component origInstanceNode, String cmptInstanceId, List<CapabilityDefinition> prevCapabilities) {
  53         List<String> prevCapOwnerIds = prevCapabilities.stream().map(CapabilityDefinition::getOwnerId).distinct().collect(Collectors.toList());
  54         return mergeInstanceUtils.mapOldToNewCapabilitiesOwnerIds(container, origInstanceNode, cmptInstanceId, prevCapOwnerIds);
  55     }
  56
  57     private void removeNotFoundNewCapabilities(Map<CapabilityDefinition, CapabilityDefinition> oldToNewCapMap) {
  58         oldToNewCapMap.values().removeIf(Objects::isNull);
  59     }
  60
  61     private ComponentInstance resolveInstanceCapabilities(Component capabilityOwnerContainer, String cmptInstanceId) {
  62         return MapUtil.toMap(capabilityOwnerContainer.getComponentInstances(), ComponentInstance::getUniqueId).get(cmptInstanceId);
  63     }
  64
  65
  66     private CapabilityDefinition findCurrCapability(Map<String, List<CapabilityDefinition>> capabilitiesByType, CapabilityDefinition oldCap, String newCapOwnerId) {
  67         List<CapabilityDefinition> newCapOfType = capabilitiesByType.get(oldCap.getType());
  68         if (newCapOwnerId == null || CollectionUtils.isEmpty(newCapOfType)) {
  69             return null;
  70         }
  71         return newCapOfType.stream().filter(sameNameAndOwner(oldCap.getName(), newCapOwnerId))
  72                                              .findFirst().orElse(null);
  73
  74     }
  75
  76     private Predicate<CapabilityDefinition> sameNameAndOwner(String capName, String newCapOwnerId) {
  77         return newCap -> newCap.getName().equals(capName) && newCap.getOwnerId().equals(newCapOwnerId);
  78     }
  79
  80
  81
  82 }