src/main/java/org/onap/aai/modelloader/config/DistributionClientStartupConfig.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2024 Deutsche Telekom AG Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *       http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.aai.modelloader.config;
  21
  22 import java.util.Date;
  23 import java.util.Timer;
  24 import java.util.TimerTask;
  25
  26 import javax.annotation.PreDestroy;
  27
  28 import org.onap.aai.cl.api.Logger;
  29 import org.onap.aai.cl.eelf.LoggerFactory;
  30 import org.onap.aai.modelloader.notification.EventCallback;
  31 import org.onap.aai.modelloader.service.ModelLoaderMsgs;
  32 import org.onap.aai.modelloader.service.SdcConnectionJob;
  33 import org.onap.sdc.api.IDistributionClient;
  34 import org.onap.sdc.api.results.IDistributionClientResult;
  35 import org.onap.sdc.utils.DistributionActionResultEnum;
  36 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
  37 import org.springframework.boot.context.event.ApplicationReadyEvent;
  38 import org.springframework.context.annotation.Configuration;
  39 import org.springframework.context.event.EventListener;
  40
  41 @Configuration
  42 @ConditionalOnProperty(value = "ml.distribution.connection.enabled", havingValue = "true", matchIfMissing = true)
  43 public class DistributionClientStartupConfig {
  44
  45     private static final Logger logger = LoggerFactory.getInstance().getLogger(DistributionClientStartupConfig.class);
  46
  47     private final IDistributionClient distributionClient;
  48     private final ModelLoaderConfig config;
  49     private final EventCallback eventCallback;
  50
  51     public DistributionClientStartupConfig(IDistributionClient distributionClient, ModelLoaderConfig config,
  52             EventCallback eventCallback) {
  53         this.distributionClient = distributionClient;
  54         this.config = config;
  55         this.eventCallback = eventCallback;
  56     }
  57
  58     @EventListener(ApplicationReadyEvent.class)
  59     protected void initSdcClient() {
  60         // Initialize distribution client
  61         logger.debug(ModelLoaderMsgs.INITIALIZING, "Initializing distribution client...");
  62         IDistributionClientResult initResult = null;
  63         initResult = distributionClient.init(config, eventCallback);
  64
  65         if (initResult.getDistributionActionResult() == DistributionActionResultEnum.SUCCESS) {
  66             // Start distribution client
  67             logger.debug(ModelLoaderMsgs.INITIALIZING, "Starting distribution client...");
  68             IDistributionClientResult startResult = distributionClient.start();
  69             if (startResult.getDistributionActionResult() == DistributionActionResultEnum.SUCCESS) {
  70                 logger.info(ModelLoaderMsgs.INITIALIZING, "Connection to SDC established");
  71             } else {
  72                 String errorMsg = "Failed to start distribution client: " + startResult.getDistributionMessageResult();
  73                 logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
  74
  75                 // Kick off a timer to retry the SDC connection
  76                 Timer timer = new Timer();
  77                 TimerTask task = new SdcConnectionJob(distributionClient, config, eventCallback, timer);
  78                 timer.schedule(task, new Date(), 60000);
  79             }
  80         } else {
  81             String errorMsg = "Failed to initialize distribution client: " + initResult.getDistributionMessageResult();
  82             logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
  83
  84             // Kick off a timer to retry the SDC connection
  85             Timer timer = new Timer();
  86             TimerTask task = new SdcConnectionJob(distributionClient, config, eventCallback, timer);
  87             timer.schedule(task, new Date(), 60000);
  88         }
  89     }
  90
  91     @PreDestroy
  92     public void destroy() {
  93         logger.info(ModelLoaderMsgs.STOPPING_CLIENT);
  94         if (distributionClient != null) {
  95             distributionClient.stop();
  96         }
  97     }
  98 }