cps-service/src/main/java/org/onap/cps/config/AsyncConfig.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * Copyright (c) 2021 Bell Canada.
   4  * Modifications Copyright (C) 2022 Nordix Foundation.
   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.config;
  23
  24 import jakarta.validation.constraints.Min;
  25 import java.util.concurrent.ThreadPoolExecutor;
  26 import lombok.Setter;
  27 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
  28 import org.springframework.boot.context.properties.ConfigurationProperties;
  29 import org.springframework.context.annotation.Bean;
  30 import org.springframework.context.annotation.Configuration;
  31 import org.springframework.core.task.TaskExecutor;
  32 import org.springframework.scheduling.annotation.EnableAsync;
  33 import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
  34 import org.springframework.validation.annotation.Validated;
  35
  36 @EnableAsync
  37 @Configuration
  38 @ConditionalOnProperty(name = "notification.enabled", havingValue = "true", matchIfMissing = true)
  39 @ConfigurationProperties("notification.async.executor")
  40 @Validated
  41 @Setter
  42 public class AsyncConfig {
  43
  44     @Min(0)
  45     private int corePoolSize = 2;
  46     @Min(2)
  47     private int maxPoolSize = 10;
  48     @Min(0)
  49     private int queueCapacity = Integer.MAX_VALUE;
  50     private boolean waitForTasksToCompleteOnShutdown = true;
  51     private String threadNamePrefix = "Async-";
  52
  53     /**
  54      * Creates TaskExecutor for processing data-updated events.
  55      *
  56      * @return TaskExecutor
  57      */
  58     @Bean("notificationExecutor")
  59     public TaskExecutor getThreadAsyncExecutorForNotification() {
  60         final ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
  61         executor.setCorePoolSize(corePoolSize);
  62         executor.setMaxPoolSize(maxPoolSize);
  63         executor.setQueueCapacity(queueCapacity);
  64         executor.setWaitForTasksToCompleteOnShutdown(waitForTasksToCompleteOnShutdown);
  65         executor.setKeepAliveSeconds(60);
  66         executor.setThreadNamePrefix(threadNamePrefix);
  67         executor.setRejectedExecutionHandler(new ThreadPoolExecutor.DiscardPolicy());
  68         executor.initialize();
  69         return executor;
  70     }
  71
  72 }