gerrit.onap Code Review - cps.git/blob

   1 /*
   2  *  ============LICENSE_START=======================================================
   3  *  Copyright (C) 2024-2025 OpenInfra Foundation Europe. All rights reserved.
   4  *  ================================================================================
   5  *  Licensed under the Apache License, Version 2.0 (the "License");
   6  *  you may not use this file except in compliance with the License.
   7  *  You may obtain a copy of the License at
   8  *
   9  *        http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  *  Unless required by applicable law or agreed to in writing, software
  12  *  distributed under the License is distributed on an "AS IS" BASIS,
  13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  *  See the License for the specific language governing permissions and
  15  *  limitations under the License.
  16  *
  17  *  SPDX-License-Identifier: Apache-2.0
  18  *  ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.cps.ncmp.impl.cmnotificationsubscription.cache;
  22
  23 import static org.onap.cps.ncmp.impl.cmnotificationsubscription.models.CmSubscriptionStatus.PENDING;
  24
  25 import java.util.ArrayList;
  26 import java.util.Collection;
  27 import java.util.HashMap;
  28 import java.util.HashSet;
  29 import java.util.List;
  30 import java.util.Map;
  31 import java.util.Set;
  32 import java.util.stream.Collectors;
  33 import lombok.RequiredArgsConstructor;
  34 import org.onap.cps.ncmp.api.data.models.DatastoreType;
  35 import org.onap.cps.ncmp.impl.cmnotificationsubscription.models.CmSubscriptionStatus;
  36 import org.onap.cps.ncmp.impl.cmnotificationsubscription.models.DmiCmSubscriptionDetails;
  37 import org.onap.cps.ncmp.impl.cmnotificationsubscription.models.DmiCmSubscriptionPredicate;
  38 import org.onap.cps.ncmp.impl.cmnotificationsubscription.utils.CmDataJobSubscriptionPersistenceService;
  39 import org.onap.cps.ncmp.impl.cmnotificationsubscription_1_0_0.client_to_ncmp.Predicate;
  40 import org.onap.cps.ncmp.impl.inventory.InventoryPersistence;
  41 import org.onap.cps.ncmp.impl.inventory.models.YangModelCmHandle;
  42 import org.springframework.stereotype.Component;
  43
  44 @Component
  45 @RequiredArgsConstructor
  46 public class DmiCacheHandler {
  47
  48     private final CmDataJobSubscriptionPersistenceService cmDataJobSubscriptionPersistenceService;
  49     private final Map<String, Map<String, DmiCmSubscriptionDetails>> cmNotificationSubscriptionCache;
  50     private final InventoryPersistence inventoryPersistence;
  51
  52     /**
  53      * Adds subscription to the subscription cache.
  54      *
  55      * @param subscriptionId    subscription id
  56      * @param predicates        subscription request predicates
  57      */
  58     public void add(final String subscriptionId, final List<Predicate> predicates) {
  59         cmNotificationSubscriptionCache.put(subscriptionId, createDmiSubscriptionsPerDmi(predicates));
  60     }
  61
  62     /**
  63      * Adds subscription to the subscription cache.
  64      *
  65      * @param subscriptionId subscription id
  66      * @param dmiCmSubscriptionDetailsPerDmi map of dmi cm notification subscription details per dmi
  67      */
  68     public void add(final String subscriptionId,
  69                     final Map<String, DmiCmSubscriptionDetails>
  70                             dmiCmSubscriptionDetailsPerDmi) {
  71         cmNotificationSubscriptionCache.put(subscriptionId, dmiCmSubscriptionDetailsPerDmi);
  72     }
  73
  74     /**
  75      * Get cm notification subscription cache entry via subscription id.
  76      *
  77      * @param subscriptionId    subscription id
  78      * @return map of dmi cm notification subscriptions per dmi
  79      */
  80     public Map<String, DmiCmSubscriptionDetails> get(final String subscriptionId) {
  81         return cmNotificationSubscriptionCache.get(subscriptionId);
  82     }
  83
  84
  85     /**
  86      * Remove cache entries with CmNotificationSubscriptionStatus ACCEPTED/REJECTED via subscription id.
  87      *
  88      * @param subscriptionId subscription id as key in CM notification Subscription cache.
  89      */
  90     public void removeAcceptedAndRejectedDmiSubscriptionEntries(final String subscriptionId) {
  91         final Map<String, DmiCmSubscriptionDetails> dmiSubscriptionsPerDmi =
  92                 cmNotificationSubscriptionCache.get(subscriptionId);
  93         final Map<String, DmiCmSubscriptionDetails> updatedDmiSubscriptionsPerDmi =
  94                 dmiSubscriptionsPerDmi.entrySet().stream()
  95                         .filter(dmiCmNotificationSubscription -> !isAcceptedOrRejected(
  96                                 dmiCmNotificationSubscription.getValue()))
  97                         .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
  98         cmNotificationSubscriptionCache.put(subscriptionId, updatedDmiSubscriptionsPerDmi);
  99     }
 100
 101     /**
 102      *  Creates map of subscription details per DMI.
 103      *
 104      * @param predicates    CM Subscription Create Request Predicates
 105      * @return              Map of DmiCmNotificationSubscription per DMI plugin
 106      */
 107     public Map<String, DmiCmSubscriptionDetails> createDmiSubscriptionsPerDmi(
 108             final List<Predicate> predicates) {
 109         final Map<String, DmiCmSubscriptionDetails> dmiSubscriptionsPerDmi =
 110                 new HashMap<>();
 111         for (final Predicate requestPredicate : predicates) {
 112             final List<String> targetFilter = requestPredicate.getTargetFilter();
 113             final DatastoreType datastoreType = DatastoreType.fromDatastoreName(
 114                     requestPredicate.getScopeFilter().getDatastore().toString());
 115             final Set<String> xpaths = new HashSet<>(requestPredicate.getScopeFilter().getXpathFilter());
 116             final Map<String, Set<String>> targetCmHandlesByDmiMap = groupTargetCmHandleIdsByDmi(targetFilter);
 117             for (final Map.Entry<String, Set<String>> targetCmHandlesByDmi: targetCmHandlesByDmiMap.entrySet()) {
 118                 final DmiCmSubscriptionPredicate dmiCmSubscriptionPredicate =
 119                         new DmiCmSubscriptionPredicate(targetCmHandlesByDmi.getValue(),
 120                                 datastoreType, xpaths);
 121                 updateDmiSubscriptionDetailsPerDmi(targetCmHandlesByDmi.getKey(),
 122                         dmiCmSubscriptionPredicate,
 123                         dmiSubscriptionsPerDmi);
 124             }
 125         }
 126         return dmiSubscriptionsPerDmi;
 127     }
 128
 129     /**
 130      *  Update status in map of subscription details per DMI.
 131      *
 132      * @param subscriptionId    String of subscription Id
 133      * @param dmiServiceName    String of dmiServiceName
 134      * @param status            String of status
 135      *
 136      */
 137     public void updateDmiSubscriptionStatus(final String subscriptionId, final String dmiServiceName,
 138                                             final CmSubscriptionStatus status) {
 139         final Map<String, DmiCmSubscriptionDetails> dmiSubscriptionsPerDmi =
 140                 cmNotificationSubscriptionCache.get(subscriptionId);
 141         dmiSubscriptionsPerDmi.get(dmiServiceName).setCmSubscriptionStatus(status);
 142         cmNotificationSubscriptionCache.put(subscriptionId, dmiSubscriptionsPerDmi);
 143     }
 144
 145     /**
 146      *  Persist map of subscription details per DMI.
 147      *
 148      * @param subscriptionId    String of subscription Id
 149      * @param dmiServiceName    String of dmiServiceName
 150      *
 151      */
 152     public void persistIntoDatabasePerDmi(final String subscriptionId, final String dmiServiceName) {
 153         final List<DmiCmSubscriptionPredicate> dmiCmSubscriptionPredicates =
 154                 cmNotificationSubscriptionCache.get(subscriptionId).get(dmiServiceName)
 155                         .getDmiCmSubscriptionPredicates();
 156         for (final DmiCmSubscriptionPredicate dmiCmSubscriptionPredicate : dmiCmSubscriptionPredicates) {
 157             final DatastoreType datastoreType = dmiCmSubscriptionPredicate.getDatastoreType();
 158             final Set<String> cmHandles = dmiCmSubscriptionPredicate.getTargetCmHandleIds();
 159             for (final String cmHandle: cmHandles) {
 160                 cmDataJobSubscriptionPersistenceService.addSubscription(datastoreType.getDatastoreName(),
 161                         cmHandle, subscriptionId);
 162             }
 163         }
 164     }
 165
 166     /**
 167      *  Remove subscription from database per DMI service name.
 168      *
 169      * @param subscriptionId    String of subscription id
 170      * @param dmiServiceName    String of dmiServiceName
 171      *
 172      */
 173     public void removeFromDatabase(final String subscriptionId, final String dmiServiceName) {
 174         final List<DmiCmSubscriptionPredicate> dmiCmSubscriptionPredicates =
 175                 cmNotificationSubscriptionCache.get(subscriptionId).get(dmiServiceName)
 176                         .getDmiCmSubscriptionPredicates();
 177         for (final DmiCmSubscriptionPredicate dmiCmSubscriptionPredicate : dmiCmSubscriptionPredicates) {
 178             final DatastoreType datastoreType = dmiCmSubscriptionPredicate.getDatastoreType();
 179             final Set<String> cmHandles = dmiCmSubscriptionPredicate.getTargetCmHandleIds();
 180             for (final String cmHandle: cmHandles) {
 181                 cmDataJobSubscriptionPersistenceService.removeSubscription(datastoreType.getDatastoreName(),
 182                         cmHandle, subscriptionId);
 183             }
 184         }
 185     }
 186
 187     private void updateDmiSubscriptionDetailsPerDmi(
 188             final String dmiServiceName,
 189             final DmiCmSubscriptionPredicate dmiCmSubscriptionPredicate,
 190             final Map<String, DmiCmSubscriptionDetails> dmiSubscriptionsPerDmi) {
 191         if (dmiSubscriptionsPerDmi.containsKey(dmiServiceName)) {
 192             dmiSubscriptionsPerDmi.get(dmiServiceName)
 193                     .getDmiCmSubscriptionPredicates().add(dmiCmSubscriptionPredicate);
 194         } else {
 195             dmiSubscriptionsPerDmi.put(dmiServiceName,
 196                     new DmiCmSubscriptionDetails(
 197                             new ArrayList<>(List.of(dmiCmSubscriptionPredicate)),
 198                             PENDING));
 199         }
 200     }
 201
 202     private Map<String, Set<String>> groupTargetCmHandleIdsByDmi(final List<String> targetCmHandleIds) {
 203         final Map<String, Set<String>> targetCmHandlesByDmiServiceNames = new HashMap<>();
 204         final Collection<YangModelCmHandle> yangModelCmHandles =
 205                 inventoryPersistence.getYangModelCmHandles(targetCmHandleIds);
 206
 207         for (final YangModelCmHandle yangModelCmHandle : yangModelCmHandles) {
 208             final String dmiServiceName = yangModelCmHandle.getDmiServiceName();
 209             targetCmHandlesByDmiServiceNames.putIfAbsent(dmiServiceName, new HashSet<>());
 210             targetCmHandlesByDmiServiceNames.get(dmiServiceName).add(yangModelCmHandle.getId());
 211         }
 212         return targetCmHandlesByDmiServiceNames;
 213     }
 214
 215     private boolean isAcceptedOrRejected(final DmiCmSubscriptionDetails dmiCmSubscription) {
 216         return dmiCmSubscription.getCmSubscriptionStatus().toString().equals("ACCEPTED")
 217                 || dmiCmSubscription.getCmSubscriptionStatus().toString().equals("REJECTED");
 218     }
 219 }