Add unit test for cloud bean.
[aai/esr-server.git] / esr-mgr / src / main / java / org / onap / aai / esr / wrapper / VimManagerWrapper.java
1 /**
2  * Copyright 2017 ZTE 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 package org.onap.aai.esr.wrapper;
17
18 import java.util.ArrayList;
19
20 import javax.ws.rs.core.Response;
21
22 import org.onap.aai.esr.entity.aai.CloudRegionDetail;
23 import org.onap.aai.esr.entity.aai.CloudRegionList;
24 import org.onap.aai.esr.entity.aai.EsrSystemInfo;
25 import org.onap.aai.esr.entity.rest.VimRegisterInfo;
26 import org.onap.aai.esr.entity.rest.VimRegisterResponse;
27 import org.onap.aai.esr.externalservice.aai.CloudRegionProxy;
28 import org.onap.aai.esr.util.ExtsysUtil;
29 import org.onap.aai.esr.util.VimManagerUtil;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32
33 import com.google.gson.Gson;
34
35
36 public class VimManagerWrapper {
37
38   private static VimManagerWrapper vimManagerWrapper;
39   private static final Logger LOG = LoggerFactory.getLogger(VimManagerWrapper.class);
40
41   /**
42    * get VimManagerWrapper instance.
43    * 
44    * @return vim manager wrapper instance
45    */
46   public static VimManagerWrapper getInstance() {
47     if (vimManagerWrapper == null) {
48       vimManagerWrapper = new VimManagerWrapper();
49     }
50     return vimManagerWrapper;
51
52   }
53
54   public Response registerVim(VimRegisterInfo vimRegisterInfo) {
55     LOG.info(
56         "Start register VIM, input VIM info is: " + ExtsysUtil.objectToString(vimRegisterInfo));
57     CloudRegionDetail cloudRegion = new CloudRegionDetail();
58     VimRegisterResponse result = new VimRegisterResponse();
59     cloudRegion = VimManagerUtil.vimRegisterInfo2CloudRegion(vimRegisterInfo);
60     String cloudOwner = vimRegisterInfo.getCloudOwner();
61     String cloudRegionId = vimRegisterInfo.getCloudRegionId();
62     try {
63       CloudRegionProxy.registerVim(cloudOwner, cloudRegionId, cloudRegion);
64       result.setCloudOwner(cloudOwner);
65       result.setCloudRegionId(cloudRegionId);
66       return Response.ok(result).build();
67     } catch (Exception error) {
68       error.printStackTrace();
69       LOG.error("Register VIM failed !" + error.getMessage());
70       return Response.serverError().build();
71     }
72   }
73
74   public Response updateVim(String cloudOwner, String cloudRegionId,VimRegisterInfo vimRegisterInfo) {
75     LOG.info("Start update VIM, input VIM info is: " + ExtsysUtil.objectToString(vimRegisterInfo));
76     CloudRegionDetail cloudRegionDetail = new CloudRegionDetail();
77     VimRegisterResponse result = new VimRegisterResponse();
78
79     cloudRegionDetail = getVimUpdateInfo(vimRegisterInfo);
80     try {
81       CloudRegionProxy.registerVim(cloudOwner, cloudRegionId, cloudRegionDetail);
82       result.setCloudOwner(cloudOwner);
83       result.setCloudRegionId(cloudRegionId);
84       return Response.ok(result).build();
85     } catch (Exception e) {
86       e.printStackTrace();
87       LOG.error("Update VIM failed !" + e.getMessage());
88       return Response.serverError().build();
89     }
90   }
91
92   public Response queryVimListDetails() {
93     ArrayList<VimRegisterInfo> vimRegisterInfos = new ArrayList<VimRegisterInfo>();
94     CloudRegionList cloudRegionList = new CloudRegionList();
95     try {
96       String aaiVimList = CloudRegionProxy.qureyVimList();
97       cloudRegionList = new Gson().fromJson(aaiVimList, CloudRegionList.class);
98       vimRegisterInfos = getVimDetailList(cloudRegionList);
99       return Response.ok(vimRegisterInfos).build();
100     } catch (Exception error) {
101       error.printStackTrace();
102       LOG.error("Query vim list details failed !" + error.getMessage());
103       return Response.serverError().build();
104     }
105
106   }
107
108   public Response queryVimById(String cloudOwner, String cloudRegionId) {
109     VimRegisterInfo vim = new VimRegisterInfo();
110     CloudRegionDetail cloudRegionDetail = new CloudRegionDetail();
111     try {
112       String cloudRegionstr = CloudRegionProxy.queryVimDetail(cloudOwner, cloudRegionId);
113       LOG.info("Response from AAI by query VIM: " + cloudRegionstr);
114       cloudRegionDetail = new Gson().fromJson(cloudRegionstr, CloudRegionDetail.class);
115       vim = VimManagerUtil.cloudRegion2VimRegisterInfo(cloudRegionDetail);
116       return Response.ok(vim).build();
117     } catch (Exception e) {
118       e.printStackTrace();
119       return Response.serverError().build();
120     }
121
122   }
123
124   private ArrayList<VimRegisterInfo> getVimDetailList(CloudRegionList cloudRegionList) {
125     ArrayList<VimRegisterInfo> vimRegisterInfos = new ArrayList<VimRegisterInfo>();
126     VimRegisterInfo vimRegisterInfo = new VimRegisterInfo();
127     int cloudRegionNum = cloudRegionList.getCloudRegion().size();
128     for (int i = 0; i < cloudRegionNum; i++) {
129       String cloudOwner = cloudRegionList.getCloudRegion().get(i).getCloudOwner();
130       String cloudRegionId = cloudRegionList.getCloudRegion().get(i).getCloudRegionId();
131       vimRegisterInfo = getVimDetail(cloudOwner, cloudRegionId);
132       vimRegisterInfos.add(vimRegisterInfo);
133     }
134     return vimRegisterInfos;
135   }
136
137   private VimRegisterInfo getVimDetail(String cloudOwner, String cloudRegionId) {
138     CloudRegionDetail cloudRegionDetail = new CloudRegionDetail();
139     VimRegisterInfo registeredVimInfo = new VimRegisterInfo();
140     try {
141       String cloudRegionstr = CloudRegionProxy.queryVimDetail(cloudOwner, cloudRegionId);
142       cloudRegionDetail = new Gson().fromJson(cloudRegionstr, CloudRegionDetail.class);
143       registeredVimInfo = VimManagerUtil.cloudRegion2VimRegisterInfo(cloudRegionDetail);
144     } catch (Exception error) {
145       error.printStackTrace();
146       LOG.error("query VIM detail failed ! cloud-owner = " + cloudOwner + ", cloud-region-id = "
147           + cloudRegionId + error.getMessage());
148     }
149     return registeredVimInfo;
150   }
151
152   private CloudRegionDetail getOriginalCloudRegion(String cloudOwner, String cloudRegionId) {
153     CloudRegionDetail cloudRegionDetail = new CloudRegionDetail();
154     try {
155       String cloudRegionstr = CloudRegionProxy.queryVimDetail(cloudOwner, cloudRegionId);
156       cloudRegionDetail = new Gson().fromJson(cloudRegionstr, CloudRegionDetail.class);
157       return cloudRegionDetail;
158     } catch (Exception error) {
159       error.printStackTrace();
160       LOG.error("query VIM detail failed ! cloud-owner = " + cloudOwner + ", cloud-region-id = "
161           + cloudRegionId + error.getMessage());
162       return null;
163     }
164   }
165   
166   private CloudRegionDetail getVimUpdateInfo(VimRegisterInfo vimRegisterInfo) {
167     CloudRegionDetail cloudRegionDetail = new CloudRegionDetail();
168     CloudRegionDetail originalCloudRegionDetail = new CloudRegionDetail();
169     EsrSystemInfo originalSystemInfo = new EsrSystemInfo();
170     String cloudOwner = vimRegisterInfo.getCloudOwner();
171     String cloudRegionId = vimRegisterInfo.getCloudRegionId();
172     originalCloudRegionDetail = getOriginalCloudRegion(cloudOwner, cloudRegionId);
173     String resourceVersion = originalCloudRegionDetail.getResourceVersion();
174     cloudRegionDetail = VimManagerUtil.vimRegisterInfo2CloudRegion(vimRegisterInfo);
175     if(resourceVersion != null) {
176       cloudRegionDetail.setResourceVersion(resourceVersion);
177       originalSystemInfo = originalCloudRegionDetail.getEsrSystemInfoList().getEsrSystemInfo().get(0);
178       cloudRegionDetail.getEsrSystemInfoList().getEsrSystemInfo().get(0).setEsrSystemInfoId(originalSystemInfo.getEsrSystemInfoId());
179       cloudRegionDetail.getEsrSystemInfoList().getEsrSystemInfo().get(0).setResouceVersion(originalSystemInfo.getResouceVersion());
180     }
181     return cloudRegionDetail;
182   }
183
184   public Response delVim(String cloudOwner, String cloudRegionId) {
185     // TODO
186     return Response.noContent().build();
187   }
188 }