Add unit test for vnfm manage API.
[aai/esr-server.git] / esr-mgr / src / main / java / org / onap / aai / esr / wrapper / VnfmManagerWrapper.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.EsrSystemInfo;
23 import org.onap.aai.esr.entity.aai.EsrVnfmDetail;
24 import org.onap.aai.esr.entity.aai.EsrVnfmList;
25 import org.onap.aai.esr.entity.rest.CommonRegisterResponse;
26 import org.onap.aai.esr.entity.rest.VnfmRegisterInfo;
27 import org.onap.aai.esr.exception.ExceptionUtil;
28 import org.onap.aai.esr.exception.ExtsysException;
29 import org.onap.aai.esr.externalservice.aai.ExternalSystemProxy;
30 import org.onap.aai.esr.util.VnfmManagerUtil;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33
34 import com.google.gson.Gson;
35
36 public class VnfmManagerWrapper {
37   private static VnfmManagerWrapper vnfmManagerWrapper;
38   private static final Logger LOG = LoggerFactory.getLogger(VnfmManagerWrapper.class);
39
40   private static VnfmManagerUtil vnfmManagerUtil = new VnfmManagerUtil();
41   
42   /**
43    * get VnfmManagerWrapper instance.
44    * 
45    * @return vnfm manager wrapper instance
46    */
47   public static VnfmManagerWrapper getInstance() {
48     if (vnfmManagerWrapper == null) {
49       vnfmManagerWrapper = new VnfmManagerWrapper();
50     }
51     return vnfmManagerWrapper;
52   }
53
54   public Response registerVnfm(VnfmRegisterInfo vnfm) {
55     CommonRegisterResponse result = new CommonRegisterResponse();
56     EsrVnfmDetail esrVnfmDetail = vnfmManagerUtil.vnfmRegisterInfo2EsrVnfm(vnfm);
57     String vnfmId = esrVnfmDetail.getVnfmId();
58     try {
59       ExternalSystemProxy.registerVnfm(vnfmId, esrVnfmDetail);
60       result.setId(vnfmId);
61       return Response.ok(result).build();
62     } catch (ExtsysException e) {
63       LOG.error("Register VNFM failed !" , e);
64       throw ExceptionUtil.buildExceptionResponse(e.getMessage());
65     }
66   }
67
68   public Response updateVnfm(VnfmRegisterInfo vnfm, String vnfmId) {
69     CommonRegisterResponse result = new CommonRegisterResponse();
70     EsrVnfmDetail originalEsrVnfmDetail = queryEsrVnfmDetail(vnfmId);
71     EsrVnfmDetail esrVnfmDetail = vnfmManagerUtil.vnfmRegisterInfo2EsrVnfm(vnfm);
72     String resourceVersion = getResourceVersion(vnfmId);
73     esrVnfmDetail.setResourceVersion(resourceVersion);
74     esrVnfmDetail.setVnfmId(vnfmId);
75     EsrSystemInfo originalEsrSystemInfo = originalEsrVnfmDetail.getEsrSystemInfoList().getEsrSystemInfo().get(0);
76     esrVnfmDetail.getEsrSystemInfoList().getEsrSystemInfo().get(0)
77         .setEsrSystemInfoId(originalEsrSystemInfo.getEsrSystemInfoId());
78     esrVnfmDetail.getEsrSystemInfoList().getEsrSystemInfo().get(0)
79         .setResouceVersion(originalEsrSystemInfo.getResouceVersion());
80     try {
81       ExternalSystemProxy.registerVnfm(vnfmId, esrVnfmDetail);
82       result.setId(vnfmId);
83       return Response.ok(result).build();
84     } catch (ExtsysException e) {
85       LOG.error("Update VNFM failed !", e);
86       throw ExceptionUtil.buildExceptionResponse(e.getMessage());
87     }
88   }
89
90   public Response queryVnfmList() {
91     ArrayList<VnfmRegisterInfo> vnfmList = new ArrayList<VnfmRegisterInfo>();
92     EsrVnfmList esrVnfm = new EsrVnfmList();
93     try {
94       String esrVnfmStr = ExternalSystemProxy.queryVnfmList();
95       esrVnfm = new Gson().fromJson(esrVnfmStr, EsrVnfmList.class);
96       LOG.info("Response from AAI by query VNFM list: " + esrVnfm);
97       vnfmList = getVnfmDetailList(esrVnfm);
98       return Response.ok(vnfmList).build();
99     } catch (ExtsysException e) {
100       LOG.error("Query VNFM list failed !", e);
101       return Response.ok(vnfmList).build();
102     }
103   }
104
105   public Response queryVnfmById(String vnfmId) {
106     VnfmRegisterInfo vnfm = queryVnfmDetail(vnfmId);
107     if (vnfm != null) {
108       return Response.ok(vnfm).build();
109     } else {
110       return Response.ok(vnfm).build();
111     }
112   }
113
114   public Response delVnfm(String vnfmId) {
115     String resourceVersion = getResourceVersion(vnfmId);
116     try {
117       ExternalSystemProxy.deleteVnfm(vnfmId, resourceVersion);
118       return Response.noContent().build();
119     } catch (ExtsysException e) {
120       LOG.error(
121           "Delete VNFM from A&AI failed! VNFM ID: " + vnfmId + "resouce-version:" + resourceVersion,
122           e);
123       throw ExceptionUtil.buildExceptionResponse(e.getMessage());
124     }
125   }
126
127   private VnfmRegisterInfo queryVnfmDetail(String vnfmId) {
128     VnfmRegisterInfo vnfm = new VnfmRegisterInfo();
129     EsrVnfmDetail esrVnfmDetail = new EsrVnfmDetail();
130     try {
131       String esrVnfmstr = ExternalSystemProxy.queryVnfmDetail(vnfmId);
132       LOG.info("Response from AAI by query VNFM: " + esrVnfmstr);
133       esrVnfmDetail = new Gson().fromJson(esrVnfmstr, EsrVnfmDetail.class);
134       vnfm = vnfmManagerUtil.esrVnfm2VnfmRegisterInfo(esrVnfmDetail);
135     } catch (ExtsysException e) {
136       LOG.error("Query VNFM detail failed! VNFM ID: " + vnfmId, e);
137     }
138     return vnfm;
139   }
140
141   private ArrayList<VnfmRegisterInfo> getVnfmDetailList(EsrVnfmList esrVnfm) {
142     ArrayList<VnfmRegisterInfo> vnfmInfoList = new ArrayList<VnfmRegisterInfo>();
143     for (int i = 0; i < esrVnfm.getEsrVnfm().size(); i++) {
144       String vnfmId = esrVnfm.getEsrVnfm().get(i).getVnfmId();
145       VnfmRegisterInfo vnfmInfo = queryVnfmDetail(vnfmId);
146       if (vnfmInfo != null) {
147         vnfmInfoList.add(vnfmInfo);
148       }
149     }
150     return vnfmInfoList;
151   }
152   
153   private String getResourceVersion (String vnfmId) {
154     String resourceVersion = null;
155     EsrVnfmDetail esrVnfmDetail = queryEsrVnfmDetail(vnfmId);
156     if (esrVnfmDetail != null && esrVnfmDetail.getResourceVersion() != null) {
157       resourceVersion = esrVnfmDetail.getResourceVersion();
158     }
159     return resourceVersion;
160   }
161   
162   private EsrVnfmDetail queryEsrVnfmDetail (String vnfmId) {
163     EsrVnfmDetail esrVnfmDetail = new EsrVnfmDetail();
164     try {
165       String esrVnfmstr = ExternalSystemProxy.queryVnfmDetail(vnfmId);
166       LOG.info("Response from AAI by query VNFM: " + esrVnfmstr);
167       esrVnfmDetail = new Gson().fromJson(esrVnfmstr, EsrVnfmDetail.class);
168     } catch (ExtsysException e) {
169       LOG.error("Query VNFM detail failed! VNFM ID: " + vnfmId, e);
170       throw ExceptionUtil.buildExceptionResponse(e.getMessage());
171     }
172     return esrVnfmDetail;
173   }
174 }