gerrit.onap Code Review - sdc.git/blob

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