Improve unit-test coverage
[vfc/nfvo/driver/vnfm/svnfm.git] / nokia / vnfmdriver / vfcadaptorservice / vfcadaptor / src / main / java / org / onap / vfc / nfvo / driver / vnfm / svnfm / init / CbamNofiticationSubscription.java
1 /*
2  * Copyright 2016-2017, Nokia Corporation
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.onap.vfc.nfvo.driver.vnfm.svnfm.init;
18
19 import java.io.IOException;
20 import java.util.List;
21
22 import org.apache.http.client.ClientProtocolException;
23 import org.onap.vfc.nfvo.driver.vnfm.svnfm.cbam.bo.CBAMCreateSubscriptionRequest;
24 import org.onap.vfc.nfvo.driver.vnfm.svnfm.cbam.bo.CBAMCreateSubscriptionResponse;
25 import org.onap.vfc.nfvo.driver.vnfm.svnfm.cbam.bo.entity.Subscription;
26 import org.onap.vfc.nfvo.driver.vnfm.svnfm.cbam.inf.CbamMgmrInf;
27 import org.onap.vfc.nfvo.driver.vnfm.svnfm.common.bo.AdaptorEnv;
28 import org.onap.vfc.nfvo.driver.vnfm.svnfm.db.bean.VnfmSubscriptionInfo;
29 import org.onap.vfc.nfvo.driver.vnfm.svnfm.db.mapper.VnfmSubscriptionsMapper;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32 import org.springframework.beans.factory.annotation.Autowired;
33 import org.springframework.boot.ApplicationArguments;
34 import org.springframework.boot.ApplicationRunner;
35 import org.springframework.core.annotation.Order;
36 import org.springframework.stereotype.Component;
37
38 //@Order(3)
39 //@Component
40 public class CbamNofiticationSubscription implements ApplicationRunner {
41         private static final Logger logger = LoggerFactory.getLogger(CbamNofiticationSubscription.class);
42
43         @Autowired
44         AdaptorEnv adaptorEnv;
45         
46         @Autowired
47         private CbamMgmrInf cbamMgmr;
48         
49         @Autowired
50         private VnfmSubscriptionsMapper subscriptionsMapper;
51         
52         @Override
53         public void run(ApplicationArguments args){
54                 boolean subscribed = false;
55                 List<VnfmSubscriptionInfo> allSubscripions = subscriptionsMapper.getAll();
56                 if(allSubscripions != null && !allSubscripions.isEmpty())
57                 {
58                         for(VnfmSubscriptionInfo subscriptionInfo : allSubscripions)
59                         {
60                                 try {
61                                         Subscription subscription = cbamMgmr.getSubscription(subscriptionInfo.getId());
62                                         if(subscription != null)
63                                         {
64                                                 subscribed = true;
65                                                 logger.info("CBAM Notification has already been subscribed with id = " + subscriptionInfo.getId());
66                                         }
67                                         else
68                                         {
69                                                 subscriptionsMapper.delete(subscriptionInfo.getId());
70                                         }
71                                 } catch (Exception e) {
72                                         logger.error("Query or delete subscription error.", e);
73                                 }
74                         }
75                 }
76                 
77                 if(!subscribed)
78                 {
79                         CBAMCreateSubscriptionRequest subscriptionRequest = new CBAMCreateSubscriptionRequest();
80                         subscriptionRequest.setCallbackUrl(adaptorEnv.getDriverApiUriFront() + "/api/nokiavnfmdriver/v1/notifications");
81                         try {
82                                 CBAMCreateSubscriptionResponse createSubscription = cbamMgmr.createSubscription(subscriptionRequest);
83                                 if(createSubscription != null)
84                                 {
85                                         subscriptionsMapper.insert(createSubscription.getId());
86                                         logger.info("CBAM Notification is successfully subscribed with id = " + createSubscription.getId());
87                                 }
88                         } catch (Exception e) {
89                                 logger.error("Subscribe notification error.", e);
90                         }
91                 }
92         }
93 }