2  *  ============LICENSE_START=======================================================
 
   3  *  Copyright (C) 2023 Nordix Foundation
 
   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
 
   9  *        http://www.apache.org/licenses/LICENSE-2.0
 
  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.
 
  17  *  SPDX-License-Identifier: Apache-2.0
 
  18  * ============LICENSE_END=========================================================
 
  21 package org.onap.cps.ncmp.api.impl.utils.data.operation;
 
  23 import io.cloudevents.CloudEvent;
 
  24 import java.util.ArrayList;
 
  25 import java.util.Collection;
 
  26 import java.util.Collections;
 
  27 import java.util.HashMap;
 
  28 import java.util.List;
 
  31 import java.util.stream.Collectors;
 
  32 import lombok.AccessLevel;
 
  33 import lombok.NoArgsConstructor;
 
  34 import lombok.extern.slf4j.Slf4j;
 
  35 import org.onap.cps.ncmp.api.NcmpEventResponseCode;
 
  36 import org.onap.cps.ncmp.api.impl.events.EventsPublisher;
 
  37 import org.onap.cps.ncmp.api.impl.operations.CmHandle;
 
  38 import org.onap.cps.ncmp.api.impl.operations.DmiDataOperation;
 
  39 import org.onap.cps.ncmp.api.impl.utils.DmiServiceNameOrganizer;
 
  40 import org.onap.cps.ncmp.api.impl.utils.context.CpsApplicationContext;
 
  41 import org.onap.cps.ncmp.api.impl.yangmodels.YangModelCmHandle;
 
  42 import org.onap.cps.ncmp.api.inventory.CmHandleState;
 
  43 import org.onap.cps.ncmp.api.models.DataOperationDefinition;
 
  44 import org.onap.cps.ncmp.api.models.DataOperationRequest;
 
  45 import org.springframework.scheduling.annotation.Async;
 
  46 import org.springframework.util.LinkedMultiValueMap;
 
  47 import org.springframework.util.MultiValueMap;
 
  50 @NoArgsConstructor(access = AccessLevel.PRIVATE)
 
  51 public class ResourceDataOperationRequestUtils {
 
  53     private static final String UNKNOWN_SERVICE_NAME = null;
 
  56      * Create a list of DMI data operation per DMI service (name).
 
  58      * @param topicParamInQuery      client given topic
 
  59      * @param requestId              unique identifier per request
 
  60      * @param dataOperationRequestIn incoming data operation request details
 
  61      * @param yangModelCmHandles     involved cm handles represented as YangModelCmHandle (incl. metadata)
 
  62      * @return {@code Map<String, List<DmiBatchOperation>>} Create a list of DMI batch operation per DMI service (name).
 
  64     public static Map<String, List<DmiDataOperation>> processPerDefinitionInDataOperationsRequest(
 
  65             final String topicParamInQuery,
 
  66             final String requestId,
 
  67             final DataOperationRequest dataOperationRequestIn,
 
  68             final Collection<YangModelCmHandle> yangModelCmHandles) {
 
  70         final Map<String, List<DmiDataOperation>> dmiDataOperationsOutPerDmiServiceName = new HashMap<>();
 
  71         final MultiValueMap<String, Map<NcmpEventResponseCode, List<String>>> cmHandleIdsPerOperationIdPerResponseCode
 
  72                 = new LinkedMultiValueMap<>();
 
  73         final Set<String> nonReadyCmHandleIdsLookup = filterAndGetNonReadyCmHandleIds(yangModelCmHandles);
 
  75         final Map<String, Map<String, Map<String, String>>> dmiPropertiesPerCmHandleIdPerServiceName =
 
  76                 DmiServiceNameOrganizer.getDmiPropertiesPerCmHandleIdPerServiceName(yangModelCmHandles);
 
  78         final Map<String, String> dmiServiceNamesPerCmHandleId =
 
  79                 getDmiServiceNamesPerCmHandleId(dmiPropertiesPerCmHandleIdPerServiceName);
 
  81         for (final DataOperationDefinition dataOperationDefinitionIn :
 
  82                 dataOperationRequestIn.getDataOperationDefinitions()) {
 
  83             final List<String> nonExistingCmHandleIds = new ArrayList<>();
 
  84             final List<String> nonReadyCmHandleIds = new ArrayList<>();
 
  85             for (final String cmHandleId : dataOperationDefinitionIn.getCmHandleIds()) {
 
  86                 if (nonReadyCmHandleIdsLookup.contains(cmHandleId)) {
 
  87                     nonReadyCmHandleIds.add(cmHandleId);
 
  89                     final String dmiServiceName = dmiServiceNamesPerCmHandleId.get(cmHandleId);
 
  90                     final Map<String, String> cmHandleIdProperties
 
  91                             = dmiPropertiesPerCmHandleIdPerServiceName.get(dmiServiceName).get(cmHandleId);
 
  92                     if (cmHandleIdProperties == null) {
 
  93                         nonExistingCmHandleIds.add(cmHandleId);
 
  95                         final DmiDataOperation dmiBatchOperationOut = getOrAddDmiBatchOperation(dmiServiceName,
 
  96                                 dataOperationDefinitionIn, dmiDataOperationsOutPerDmiServiceName);
 
  97                         final CmHandle cmHandle = CmHandle.buildCmHandleWithProperties(cmHandleId,
 
  98                                 cmHandleIdProperties);
 
  99                         dmiBatchOperationOut.getCmHandles().add(cmHandle);
 
 103             populateCmHandleIdsPerOperationIdPerResponseCode(cmHandleIdsPerOperationIdPerResponseCode,
 
 104                     dataOperationDefinitionIn.getOperationId(), NcmpEventResponseCode.CODE_100, nonExistingCmHandleIds);
 
 105             populateCmHandleIdsPerOperationIdPerResponseCode(cmHandleIdsPerOperationIdPerResponseCode,
 
 106                     dataOperationDefinitionIn.getOperationId(), NcmpEventResponseCode.CODE_101, nonReadyCmHandleIds);
 
 108         if (!cmHandleIdsPerOperationIdPerResponseCode.isEmpty()) {
 
 109             publishErrorMessageToClientTopic(topicParamInQuery, requestId, cmHandleIdsPerOperationIdPerResponseCode);
 
 111         return dmiDataOperationsOutPerDmiServiceName;
 
 115     private static void publishErrorMessageToClientTopic(final String clientTopic,
 
 116                                                          final String requestId,
 
 117                                                          final MultiValueMap<String,
 
 118                                                                  Map<NcmpEventResponseCode, List<String>>>
 
 119                                                                  cmHandleIdsPerOperationIdPerResponseCode) {
 
 120         final CloudEvent dataOperationCloudEvent = DataOperationEventCreator.createDataOperationEvent(clientTopic,
 
 121                 requestId, cmHandleIdsPerOperationIdPerResponseCode);
 
 122         final EventsPublisher<CloudEvent> eventsPublisher = CpsApplicationContext.getCpsBean(EventsPublisher.class);
 
 123         eventsPublisher.publishCloudEvent(clientTopic, requestId, dataOperationCloudEvent);
 
 126     private static Map<String, String> getDmiServiceNamesPerCmHandleId(
 
 127             final Map<String, Map<String, Map<String, String>>> dmiDmiPropertiesPerCmHandleIdPerServiceName) {
 
 128         final Map<String, String> dmiServiceNamesPerCmHandleId = new HashMap<>();
 
 129         for (final Map.Entry<String, Map<String, Map<String, String>>> dmiDmiPropertiesEntry
 
 130                 : dmiDmiPropertiesPerCmHandleIdPerServiceName.entrySet()) {
 
 131             final String dmiServiceName = dmiDmiPropertiesEntry.getKey();
 
 132             final Set<String> cmHandleIds = dmiDmiPropertiesPerCmHandleIdPerServiceName.get(dmiServiceName).keySet();
 
 133             for (final String cmHandleId : cmHandleIds) {
 
 134                 dmiServiceNamesPerCmHandleId.put(cmHandleId, dmiServiceName);
 
 137         dmiDmiPropertiesPerCmHandleIdPerServiceName.put(UNKNOWN_SERVICE_NAME, Collections.emptyMap());
 
 138         return dmiServiceNamesPerCmHandleId;
 
 141     private static DmiDataOperation getOrAddDmiBatchOperation(final String dmiServiceName,
 
 142                                                                final DataOperationDefinition
 
 143                                                                        dataOperationDefinitionIn,
 
 144                                                                final Map<String, List<DmiDataOperation>>
 
 145                                                                        dmiBatchOperationsOutPerDmiServiceName) {
 
 146         dmiBatchOperationsOutPerDmiServiceName
 
 147                 .computeIfAbsent(dmiServiceName, dmiServiceNameAsKey -> new ArrayList<>());
 
 148         final List<DmiDataOperation> dmiBatchOperationsOut
 
 149                 = dmiBatchOperationsOutPerDmiServiceName.get(dmiServiceName);
 
 150         final boolean isNewOperation = dmiBatchOperationsOut.isEmpty()
 
 151                 || !dmiBatchOperationsOut.get(dmiBatchOperationsOut.size() - 1).getOperationId()
 
 152                 .equals(dataOperationDefinitionIn.getOperationId());
 
 153         if (isNewOperation) {
 
 154             final DmiDataOperation newDmiBatchOperationOut =
 
 155                     DmiDataOperation.buildDmiDataOperationRequestBodyWithoutCmHandles(dataOperationDefinitionIn);
 
 156             dmiBatchOperationsOut.add(newDmiBatchOperationOut);
 
 157             return newDmiBatchOperationOut;
 
 159         return dmiBatchOperationsOut.get(dmiBatchOperationsOut.size() - 1);
 
 162     private static Set<String> filterAndGetNonReadyCmHandleIds(final Collection<YangModelCmHandle> yangModelCmHandles) {
 
 163         return yangModelCmHandles.stream()
 
 164                 .filter(yangModelCmHandle -> yangModelCmHandle.getCompositeState().getCmHandleState()
 
 165                         != CmHandleState.READY).map(YangModelCmHandle::getId).collect(Collectors.toSet());
 
 168     private static void populateCmHandleIdsPerOperationIdPerResponseCode(final MultiValueMap<String,
 
 169             Map<NcmpEventResponseCode, List<String>>> cmHandleIdsPerOperationIdByResponseCode,
 
 170                                                                         final String operationId,
 
 171                                                                         final NcmpEventResponseCode
 
 172                                                                                 ncmpEventResponseCode,
 
 173                                                                         final List<String> cmHandleIds) {
 
 174         if (!cmHandleIds.isEmpty()) {
 
 175             cmHandleIdsPerOperationIdByResponseCode.add(operationId, Map.of(ncmpEventResponseCode, cmHandleIds));