cps-ncmp-service/src/main/java/org/onap/cps/ncmp/api/impl/operations/DmiDataOperations.java

   1 /*
   2  *  ============LICENSE_START=======================================================
   3  *  Copyright (C) 2021-2023 Nordix Foundation
   4  *  Modifications Copyright (C) 2022 Bell Canada
   5  *  ================================================================================
   6  *  Licensed under the Apache License, Version 2.0 (the "License");
   7  *  you may not use this file except in compliance with the License.
   8  *  You may obtain a copy of the License at
   9  *
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  *  Unless required by applicable law or agreed to in writing, software
  13  *  distributed under the License is distributed on an "AS IS" BASIS,
  14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *  See the License for the specific language governing permissions and
  16  *  limitations under the License.
  17  *
  18  *  SPDX-License-Identifier: Apache-2.0
  19  *  ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.cps.ncmp.api.impl.operations;
  23
  24 import static org.onap.cps.ncmp.api.NcmpResponseStatus.DMI_SERVICE_NOT_RESPONDING;
  25 import static org.onap.cps.ncmp.api.NcmpResponseStatus.UNABLE_TO_READ_RESOURCE_DATA;
  26 import static org.onap.cps.ncmp.api.impl.operations.DatastoreType.PASSTHROUGH_RUNNING;
  27 import static org.onap.cps.ncmp.api.impl.operations.OperationType.READ;
  28
  29 import io.micrometer.core.annotation.Timed;
  30 import java.util.Collection;
  31 import java.util.List;
  32 import java.util.Map;
  33 import java.util.Set;
  34 import java.util.stream.Collectors;
  35 import lombok.extern.slf4j.Slf4j;
  36 import org.onap.cps.ncmp.api.NcmpResponseStatus;
  37 import org.onap.cps.ncmp.api.impl.client.DmiRestClient;
  38 import org.onap.cps.ncmp.api.impl.config.NcmpConfiguration;
  39 import org.onap.cps.ncmp.api.impl.exception.HttpClientRequestException;
  40 import org.onap.cps.ncmp.api.impl.executor.TaskExecutor;
  41 import org.onap.cps.ncmp.api.impl.inventory.CmHandleState;
  42 import org.onap.cps.ncmp.api.impl.inventory.InventoryPersistence;
  43 import org.onap.cps.ncmp.api.impl.utils.DmiServiceUrlBuilder;
  44 import org.onap.cps.ncmp.api.impl.utils.data.operation.ResourceDataOperationRequestUtils;
  45 import org.onap.cps.ncmp.api.impl.yangmodels.YangModelCmHandle;
  46 import org.onap.cps.ncmp.api.models.DataOperationRequest;
  47 import org.onap.cps.spi.exceptions.CpsException;
  48 import org.onap.cps.utils.JsonObjectMapper;
  49 import org.springframework.http.ResponseEntity;
  50 import org.springframework.stereotype.Component;
  51 import org.springframework.util.LinkedMultiValueMap;
  52 import org.springframework.util.MultiValueMap;
  53 import org.springframework.web.util.UriComponentsBuilder;
  54
  55 /**
  56  * Operations class for DMI data.
  57  */
  58 @Component
  59 @Slf4j
  60 public class DmiDataOperations extends DmiOperations {
  61
  62     private static final long DEFAULT_ASYNC_TASK_EXECUTOR_TIMEOUT_IN_MILLISECONDS = 30000L;
  63
  64     public DmiDataOperations(final InventoryPersistence inventoryPersistence,
  65                              final JsonObjectMapper jsonObjectMapper,
  66                              final NcmpConfiguration.DmiProperties dmiProperties,
  67                              final DmiRestClient dmiRestClient,
  68                              final DmiServiceUrlBuilder dmiServiceUrlBuilder) {
  69         super(inventoryPersistence, jsonObjectMapper, dmiProperties, dmiRestClient, dmiServiceUrlBuilder);
  70     }
  71
  72     /**
  73      * This method fetches the resource data from operational data store for given cm handle
  74      * identifier on given resource using dmi client.
  75      *
  76      * @param dataStoreName       name of data store
  77      * @param cmHandleId          network resource identifier
  78      * @param resourceId          resource identifier
  79      * @param optionsParamInQuery options query
  80      * @param topicParamInQuery   topic name for (triggering) async responses
  81      * @param requestId           requestId for async responses
  82      * @param authorization       contents of Authorization header, or null if not present
  83      * @return {@code ResponseEntity} response entity
  84      */
  85     @Timed(value = "cps.ncmp.dmi.get",
  86             description = "Time taken to fetch the resource data from operational data store for given cm handle "
  87                     + "identifier on given resource using dmi client")
  88     public ResponseEntity<Object> getResourceDataFromDmi(final String dataStoreName,
  89                                                          final String cmHandleId,
  90                                                          final String resourceId,
  91                                                          final String optionsParamInQuery,
  92                                                          final String topicParamInQuery,
  93                                                          final String requestId,
  94                                                          final String authorization) {
  95         final YangModelCmHandle yangModelCmHandle = getYangModelCmHandle(cmHandleId);
  96         final CmHandleState cmHandleState = yangModelCmHandle.getCompositeState().getCmHandleState();
  97         validateIfCmHandleStateReady(yangModelCmHandle, cmHandleState);
  98         final String jsonRequestBody = getDmiRequestBody(READ, requestId, null, null,
  99                 yangModelCmHandle);
 100         final String dmiResourceDataUrl = getDmiRequestUrl(dataStoreName, cmHandleId, resourceId, optionsParamInQuery,
 101                 topicParamInQuery, yangModelCmHandle.resolveDmiServiceName(RequiredDmiService.DATA));
 102         return dmiRestClient.postOperationWithJsonData(dmiResourceDataUrl, jsonRequestBody, READ, authorization);
 103     }
 104
 105     /**
 106      * This method fetches all the resource data from operational data store for given cm handle
 107      * identifier using dmi client.
 108      *
 109      * @param dataStoreName data store name
 110      * @param cmHandleId    network resource identifier
 111      * @param requestId     requestId for async responses
 112      * @return {@code ResponseEntity} response entity
 113      */
 114     public ResponseEntity<Object> getResourceDataFromDmi(final String dataStoreName,
 115                                                          final String cmHandleId,
 116                                                          final String requestId) {
 117         final YangModelCmHandle yangModelCmHandle = getYangModelCmHandle(cmHandleId);
 118         final String jsonRequestBody = getDmiRequestBody(READ, requestId, null, null,
 119                 yangModelCmHandle);
 120         final String dmiResourceDataUrl = getDmiRequestUrl(dataStoreName, cmHandleId, "/",
 121                 null, null,
 122                 yangModelCmHandle.resolveDmiServiceName(RequiredDmiService.DATA));
 123         final CmHandleState cmHandleState = yangModelCmHandle.getCompositeState().getCmHandleState();
 124         validateIfCmHandleStateReady(yangModelCmHandle, cmHandleState);
 125         return dmiRestClient.postOperationWithJsonData(dmiResourceDataUrl, jsonRequestBody, READ, null);
 126     }
 127
 128     /**
 129      * This method requests the resource data by data store for given list of cm handles using dmi client.
 130      * The data wil be returned as message on the topic specified.
 131      *
 132      * @param topicParamInQuery        topic name for (triggering) async responses
 133      * @param dataOperationRequest     data operation request to execute operations
 134      * @param requestId                requestId for as a response
 135      */
 136     public void requestResourceDataFromDmi(final String topicParamInQuery,
 137                                            final DataOperationRequest dataOperationRequest,
 138                                            final String requestId)  {
 139
 140         final Set<String> cmHandlesIds
 141                 = getDistinctCmHandleIdsFromDataOperationRequest(dataOperationRequest);
 142
 143         final Collection<YangModelCmHandle> yangModelCmHandles
 144                 = inventoryPersistence.getYangModelCmHandles(cmHandlesIds);
 145
 146         final Map<String, List<DmiDataOperation>> operationsOutPerDmiServiceName
 147                 = ResourceDataOperationRequestUtils.processPerDefinitionInDataOperationsRequest(topicParamInQuery,
 148                 requestId, dataOperationRequest, yangModelCmHandles);
 149
 150         buildDataOperationRequestUrlAndSendToDmiService(topicParamInQuery, requestId, operationsOutPerDmiServiceName);
 151     }
 152
 153     /**
 154      * This method creates the resource data from pass-through running data store for given cm handle
 155      * identifier on given resource using dmi client.
 156      *
 157      * @param cmHandleId    network resource identifier
 158      * @param resourceId    resource identifier
 159      * @param operationType operation enum
 160      * @param requestData   the request data
 161      * @param dataType      data type
 162      * @param authorization contents of Authorization header, or null if not present
 163      * @return {@code ResponseEntity} response entity
 164      */
 165     public ResponseEntity<Object> writeResourceDataPassThroughRunningFromDmi(final String cmHandleId,
 166                                                                              final String resourceId,
 167                                                                              final OperationType operationType,
 168                                                                              final String requestData,
 169                                                                              final String dataType,
 170                                                                              final String authorization) {
 171         final YangModelCmHandle yangModelCmHandle = getYangModelCmHandle(cmHandleId);
 172         final String jsonRequestBody = getDmiRequestBody(operationType, null, requestData, dataType,
 173                 yangModelCmHandle);
 174         final String dmiUrl = getDmiRequestUrl(PASSTHROUGH_RUNNING.getDatastoreName(), cmHandleId, resourceId,
 175                 null, null,
 176                 yangModelCmHandle.resolveDmiServiceName(RequiredDmiService.DATA));
 177         final CmHandleState cmHandleState = yangModelCmHandle.getCompositeState().getCmHandleState();
 178         validateIfCmHandleStateReady(yangModelCmHandle, cmHandleState);
 179         return dmiRestClient.postOperationWithJsonData(dmiUrl, jsonRequestBody, operationType, authorization);
 180     }
 181
 182     private YangModelCmHandle getYangModelCmHandle(final String cmHandleId) {
 183         return inventoryPersistence.getYangModelCmHandle(cmHandleId);
 184     }
 185
 186     private String getDmiRequestBody(final OperationType operationType,
 187                                      final String requestId,
 188                                      final String requestData,
 189                                      final String dataType,
 190                                      final YangModelCmHandle yangModelCmHandle) {
 191         final DmiRequestBody dmiRequestBody = DmiRequestBody.builder()
 192                 .operationType(operationType)
 193                 .requestId(requestId)
 194                 .data(requestData)
 195                 .dataType(dataType)
 196                 .build();
 197         dmiRequestBody.asDmiProperties(yangModelCmHandle.getDmiProperties());
 198         return jsonObjectMapper.asJsonString(dmiRequestBody);
 199     }
 200
 201     private String getDmiRequestUrl(final String dataStoreName,
 202                                     final String cmHandleId,
 203                                     final String resourceId,
 204                                     final String optionsParamInQuery,
 205                                     final String topicParamInQuery,
 206                                     final String dmiServiceName) {
 207         return dmiServiceUrlBuilder.getDmiDatastoreUrl(
 208                 dmiServiceUrlBuilder.populateQueryParams(resourceId, optionsParamInQuery,
 209                         topicParamInQuery), dmiServiceUrlBuilder.populateUriVariables(dataStoreName, dmiServiceName,
 210                         cmHandleId));
 211     }
 212
 213     private String getDmiServiceDataOperationRequestUrl(final String dmiServiceName,
 214                                                         final String topicParamInQuery,
 215                                                         final String requestId) {
 216         final MultiValueMap<String, String> dataOperationRequestQueryParams = dmiServiceUrlBuilder
 217                 .getDataOperationRequestQueryParams(topicParamInQuery, requestId);
 218         return dmiServiceUrlBuilder.getDataOperationRequestUrl(dataOperationRequestQueryParams,
 219                 dmiServiceUrlBuilder.populateDataOperationRequestUriVariables(dmiServiceName));
 220     }
 221
 222     private void validateIfCmHandleStateReady(final YangModelCmHandle yangModelCmHandle,
 223                                               final CmHandleState cmHandleState) {
 224         if (cmHandleState != CmHandleState.READY) {
 225             throw new CpsException("State mismatch exception.", "Cm-Handle not in READY state. "
 226                     + "cm handle state is "
 227                     + yangModelCmHandle.getCompositeState().getCmHandleState());
 228         }
 229     }
 230
 231     private static Set<String> getDistinctCmHandleIdsFromDataOperationRequest(final DataOperationRequest
 232                                                                               dataOperationRequest) {
 233         return dataOperationRequest.getDataOperationDefinitions().stream()
 234                 .flatMap(dataOperationDefinition ->
 235                         dataOperationDefinition.getCmHandleIds().stream()).collect(Collectors.toSet());
 236     }
 237
 238     private void buildDataOperationRequestUrlAndSendToDmiService(final String topicParamInQuery,
 239                                                                  final String requestId,
 240                                                                  final Map<String, List<DmiDataOperation>>
 241                                                                 groupsOutPerDmiServiceName) {
 242
 243         groupsOutPerDmiServiceName.forEach((dmiServiceName, dmiDataOperationRequestBodies) -> {
 244             final String dmiDataOperationResourceUrl =
 245                     getDmiServiceDataOperationRequestUrl(dmiServiceName, topicParamInQuery, requestId);
 246             sendDataOperationRequestToDmiService(dmiDataOperationResourceUrl, dmiDataOperationRequestBodies);
 247         });
 248     }
 249
 250     private void sendDataOperationRequestToDmiService(final String dataOperationResourceUrl,
 251                                                       final List<DmiDataOperation> dmiDataOperationRequestBodies) {
 252         final DmiDataOperationRequest dmiDataOperationRequest = DmiDataOperationRequest.builder()
 253                 .operations(dmiDataOperationRequestBodies).build();
 254         final String dmiDataOperationRequestAsJsonString =
 255                 jsonObjectMapper.asJsonString(dmiDataOperationRequest);
 256         TaskExecutor.executeTask(() -> dmiRestClient.postOperationWithJsonData(dataOperationResourceUrl,
 257                                 dmiDataOperationRequestAsJsonString, READ, null),
 258                         DEFAULT_ASYNC_TASK_EXECUTOR_TIMEOUT_IN_MILLISECONDS)
 259                 .whenCompleteAsync((response, throwable) -> handleTaskCompletionException(throwable,
 260                         dataOperationResourceUrl, dmiDataOperationRequestBodies));
 261     }
 262
 263     private void handleTaskCompletionException(final Throwable throwable,
 264                                                final String dataOperationResourceUrl,
 265                                                final List<DmiDataOperation> dmiDataOperationRequestBodies) {
 266         if (throwable != null) {
 267             final MultiValueMap<String, String> dataOperationResourceUrlParameters =
 268                     UriComponentsBuilder.fromUriString(dataOperationResourceUrl).build().getQueryParams();
 269             final String topicName = dataOperationResourceUrlParameters.get("topic").get(0);
 270             final String requestId = dataOperationResourceUrlParameters.get("requestId").get(0);
 271
 272             final MultiValueMap<DmiDataOperation, Map<NcmpResponseStatus, List<String>>>
 273                     cmHandleIdsPerResponseCodesPerOperation = new LinkedMultiValueMap<>();
 274
 275             dmiDataOperationRequestBodies.forEach(dmiDataOperationRequestBody -> {
 276                 final List<String> cmHandleIds = dmiDataOperationRequestBody.getCmHandles().stream()
 277                         .map(CmHandle::getId).toList();
 278                 if (throwable.getCause() instanceof HttpClientRequestException) {
 279                     cmHandleIdsPerResponseCodesPerOperation.add(dmiDataOperationRequestBody,
 280                             Map.of(UNABLE_TO_READ_RESOURCE_DATA, cmHandleIds));
 281                 } else {
 282                     cmHandleIdsPerResponseCodesPerOperation.add(dmiDataOperationRequestBody,
 283                             Map.of(DMI_SERVICE_NOT_RESPONDING, cmHandleIds));
 284                 }
 285             });
 286             ResourceDataOperationRequestUtils.publishErrorMessageToClientTopic(topicName, requestId,
 287                     cmHandleIdsPerResponseCodesPerOperation);
 288         }
 289     }
 290 }