Add unit test for ems management.
[aai/esr-server.git] / esr-mgr / src / main / java / org / onap / aai / esr / wrapper / EmsManagerWrapper.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 import java.util.List;
20 import javax.ws.rs.core.Response;
21 import org.onap.aai.esr.entity.rest.EmsRegisterInfo;
22 import org.onap.aai.esr.exception.ExceptionUtil;
23 import org.onap.aai.esr.exception.ExtsysException;
24 import org.onap.aai.esr.externalservice.aai.ExternalSystemProxy;
25 import org.onap.aai.esr.util.EmsManagerUtil;
26 import org.onap.aai.esr.entity.aai.EsrEmsDetail;
27 import org.onap.aai.esr.entity.aai.EsrEmsList;
28 import org.onap.aai.esr.entity.aai.EsrSystemInfo;
29 import org.onap.aai.esr.entity.rest.CommonRegisterResponse;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32 import com.google.gson.Gson;
33
34 public class EmsManagerWrapper {
35     private static EmsManagerWrapper emsManagerWrapper;
36     private static final Logger LOG = LoggerFactory.getLogger(EmsManagerWrapper.class);
37     private static EmsManagerUtil emsManagerUtil = new EmsManagerUtil();
38
39     /**
40      * get VnfmManagerWrapper instance.
41      * 
42      * @return ems manager wrapper instance
43      */
44     public static EmsManagerWrapper getInstance() {
45         if (emsManagerWrapper == null) {
46             emsManagerWrapper = new EmsManagerWrapper();
47         }
48         return emsManagerWrapper;
49     }
50
51     public Response registerEms(EmsRegisterInfo emsRegisterInfo) {
52         CommonRegisterResponse result = new CommonRegisterResponse();
53         EsrEmsDetail esrEmsDetail = new EsrEmsDetail();
54         esrEmsDetail = emsManagerUtil.emsRegisterInfo2EsrEms(emsRegisterInfo);
55         String emsId = esrEmsDetail.getEmsId();
56         try {
57             ExternalSystemProxy.registerEms(emsId, esrEmsDetail);
58             result.setId(emsId);
59             return Response.ok(result).build();
60         } catch (ExtsysException e) {
61             LOG.error("Register EMS failed !", e);
62             throw ExceptionUtil.buildExceptionResponse(e.getMessage());
63         }
64     }
65
66     public Response updateEms(EmsRegisterInfo emsRegisterInfo, String emsId) {
67         CommonRegisterResponse result = new CommonRegisterResponse();
68         EsrEmsDetail esrEmsDetail = new EsrEmsDetail();
69         esrEmsDetail = getNewEsrEmsDetail(emsRegisterInfo, emsId);
70         try {
71             ExternalSystemProxy.registerEms(emsId, esrEmsDetail);
72             result.setId(emsId);
73             return Response.ok(result).build();
74         } catch (ExtsysException e) {
75             LOG.error("Update VNFM failed !", e);
76             throw ExceptionUtil.buildExceptionResponse(e.getMessage());
77         }
78     }
79
80     public Response queryEmsList() {
81         List<EmsRegisterInfo> emsList = new ArrayList<>();
82         EsrEmsList esrEms = new EsrEmsList();
83         try {
84             String esrEmsStr = ExternalSystemProxy.queryEmsList();
85             esrEms = new Gson().fromJson(esrEmsStr, EsrEmsList.class);
86             LOG.info("Response from AAI by query EMS list: " + esrEms);
87             emsList = getEmsDetailList(esrEms);
88             return Response.ok(emsList).build();
89         } catch (ExtsysException e) {
90             LOG.error("Query EMS list failed !", e);
91             return Response.ok(emsList).build();
92         }
93     }
94
95     public Response queryEmsById(String emsId) {
96         EmsRegisterInfo ems = queryEmsDetail(emsId);
97         if (ems != null) {
98             return Response.ok(ems).build();
99         } else {
100             return Response.ok(ems).build();
101         }
102     }
103
104     public Response delEms(String emsId) {
105         EsrEmsDetail esrEmsDetail = new EsrEmsDetail();
106         esrEmsDetail = queryEsrEmsDetail(emsId);
107         String resourceVersion = esrEmsDetail.getResourceVersion();
108         try {
109             ExternalSystemProxy.deleteEms(emsId, resourceVersion);
110             return Response.noContent().build();
111         } catch (ExtsysException e) {
112             LOG.error("Delete EMS from A&AI failed! EMS ID: " + emsId + "resouce-version:" + resourceVersion, e);
113             throw ExceptionUtil.buildExceptionResponse(e.getMessage());
114         }
115     }
116
117     private EmsRegisterInfo queryEmsDetail(String emsId) {
118         EmsRegisterInfo emsRegisterInfo = new EmsRegisterInfo();
119         EsrEmsDetail esrEmsDetail = new EsrEmsDetail();
120         try {
121             String esrEmsStr = ExternalSystemProxy.queryEmsDetail(emsId);
122             LOG.info("Response from AAI by query EMS: " + esrEmsStr);
123             esrEmsDetail = new Gson().fromJson(esrEmsStr, EsrEmsDetail.class);
124             emsRegisterInfo = emsManagerUtil.EsrEms2EmsRegisterInfo(esrEmsDetail);
125         } catch (ExtsysException e) {
126             LOG.error("Query EMS detail failed! EMS ID: " + emsId, e);
127         }
128         return emsRegisterInfo;
129     }
130
131     private List<EmsRegisterInfo> getEmsDetailList(EsrEmsList esrEms) {
132         List<EmsRegisterInfo> emsInfoList = new ArrayList<>();
133         EmsRegisterInfo emsInfo = new EmsRegisterInfo();
134         for (int i = 0; i < esrEms.getEsrEms().size(); i++) {
135             String emsId = esrEms.getEsrEms().get(i).getEmsId();
136             emsInfo = queryEmsDetail(emsId);
137             if (emsInfo != null) {
138                 emsInfoList.add(emsInfo);
139             }
140         }
141         return emsInfoList;
142     }
143
144     private EsrEmsDetail queryEsrEmsDetail(String emsId) {
145         EsrEmsDetail esrEmsDetail = new EsrEmsDetail();
146         try {
147             String esrEmsStr = ExternalSystemProxy.queryEmsDetail(emsId);
148             LOG.info("Response from AAI by query EMS: " + esrEmsStr);
149             esrEmsDetail = new Gson().fromJson(esrEmsStr, EsrEmsDetail.class);
150         } catch (ExtsysException e) {
151             LOG.error("Query EMS detail failed! EMS ID: " + emsId, e);
152         }
153         return esrEmsDetail;
154     }
155
156     private EsrEmsDetail getNewEsrEmsDetail(EmsRegisterInfo emsRegisterInfo, String emsId) {
157         EsrEmsDetail esrEmsDetail = new EsrEmsDetail();
158         List<EsrSystemInfo> newEsrSysInfoList = new ArrayList<>();
159         EsrSystemInfo newEsrSystemInfo = new EsrSystemInfo();
160         EsrEmsDetail oriEsrEmsDetail = new EsrEmsDetail();
161         List<EsrSystemInfo> oriEsrSysInfoList = new ArrayList<>();
162         EsrSystemInfo originalEsrSystemInfo = new EsrSystemInfo();
163
164         oriEsrEmsDetail = queryEsrEmsDetail(emsId);
165         esrEmsDetail = emsManagerUtil.emsRegisterInfo2EsrEms(emsRegisterInfo);
166         String emsResourceVersion = oriEsrEmsDetail.getResourceVersion();
167         esrEmsDetail.setResourceVersion(emsResourceVersion);
168         esrEmsDetail.setEmsId(emsId);
169         newEsrSysInfoList = esrEmsDetail.getEsrSystemInfoList().getEsrSystemInfo();
170         oriEsrSysInfoList = oriEsrEmsDetail.getEsrSystemInfoList().getEsrSystemInfo();
171         for (int i = 0; i < oriEsrSysInfoList.size(); i++) {
172             originalEsrSystemInfo = oriEsrSysInfoList.get(i);
173             for (int j = 0; j < newEsrSysInfoList.size(); j++) {
174                 newEsrSystemInfo = newEsrSysInfoList.get(j);
175                 if (originalEsrSystemInfo.getSystemType().equals(newEsrSystemInfo.getSystemType())) {
176                     esrEmsDetail.getEsrSystemInfoList().getEsrSystemInfo().get(j)
177                             .setResouceVersion(originalEsrSystemInfo.getResouceVersion());
178                     esrEmsDetail.getEsrSystemInfoList().getEsrSystemInfo().get(j)
179                             .setEsrSystemInfoId(originalEsrSystemInfo.getEsrSystemInfoId());
180                     break;
181                 }
182             }
183         }
184         return esrEmsDetail;
185     }
186 }