2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 package org.onap.aai.modelloader.config;
22 import java.util.Date;
23 import java.util.Timer;
24 import java.util.TimerTask;
26 import javax.annotation.PreDestroy;
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;
42 @ConditionalOnProperty(value = "ml.distribution.connection.enabled", havingValue = "true", matchIfMissing = true)
43 public class DistributionClientStartupConfig {
45 private static final Logger logger = LoggerFactory.getInstance().getLogger(DistributionClientStartupConfig.class);
47 private final IDistributionClient distributionClient;
48 private final ModelLoaderConfig config;
49 private final EventCallback eventCallback;
51 public DistributionClientStartupConfig(IDistributionClient distributionClient, ModelLoaderConfig config,
52 EventCallback eventCallback) {
53 this.distributionClient = distributionClient;
55 this.eventCallback = eventCallback;
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);
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");
72 String errorMsg = "Failed to start distribution client: " + startResult.getDistributionMessageResult();
73 logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
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);
81 String errorMsg = "Failed to initialize distribution client: " + initResult.getDistributionMessageResult();
82 logger.error(ModelLoaderMsgs.ASDC_CONNECTION_ERROR, errorMsg);
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);
92 public void destroy() {
93 logger.info(ModelLoaderMsgs.STOPPING_CLIENT);
94 if (distributionClient != null) {
95 distributionClient.stop();