cps-ncmp-rest/src/main/java/org/onap/cps/ncmp/rest/executor/CpsNcmpTaskExecutor.java

   1 /*
   2  *  ============LICENSE_START=======================================================
   3  *  Copyright (C) 2022-2024 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
   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.rest.executor;
  22
  23 import static java.util.concurrent.TimeUnit.MILLISECONDS;
  24
  25 import java.util.concurrent.CompletableFuture;
  26 import java.util.function.BiConsumer;
  27 import java.util.function.Supplier;
  28 import lombok.extern.slf4j.Slf4j;
  29 import org.springframework.stereotype.Service;
  30
  31 @Slf4j
  32 @Service
  33 public class CpsNcmpTaskExecutor {
  34
  35     /**
  36      * Execute a task asynchronously, and invoke completion handler when done.
  37      *
  38      * @param taskSupplier functional method is get() task needed to be executed asynchronously
  39      * @param taskCompletionHandler the action to perform on task completion or error
  40      * @param timeOutInMillis the time-out value in milliseconds
  41      */
  42     public void executeTaskWithErrorHandling(final Supplier<Object> taskSupplier,
  43                                              final BiConsumer<Object, Throwable> taskCompletionHandler,
  44                                              final long timeOutInMillis) {
  45         CompletableFuture.supplyAsync(taskSupplier)
  46                 .orTimeout(timeOutInMillis, MILLISECONDS)
  47                 .whenCompleteAsync(taskCompletionHandler);
  48     }
  49
  50     /**
  51      * Execute a task asynchronously.
  52      *
  53      * @param taskSupplier functional method is get() task needed to be executed asynchronously
  54      * @param timeOutInMillis the time-out value in milliseconds
  55      */
  56     public void executeTask(final Supplier<Object> taskSupplier, final long timeOutInMillis) {
  57         executeTaskWithErrorHandling(taskSupplier, (taskResult, throwable) -> handleTaskCompletion(throwable),
  58                 timeOutInMillis);
  59     }
  60
  61     private void handleTaskCompletion(final Throwable throwable) {
  62         if (throwable == null) {
  63             log.info("Async task completed successfully.");
  64         } else {
  65             log.error("Async task failed. caused by : {}", throwable.toString());
  66         }
  67     }
  68 }
  69
  70
  71