Realize the query VNFM list API.

[aai/esr-server.git] / esr-mgr / src / main / java / org / onap / aai / esr / wrapper / VnfmManagerWrapper.java

diff --git a/esr-mgr/src/main/java/org/onap/aai/esr/wrapper/VnfmManagerWrapper.java b/esr-mgr/src/main/java/org/onap/aai/esr/wrapper/VnfmManagerWrapper.java
index 7749c26..5127371 100644 (file)
--- a/esr-mgr/src/main/java/org/onap/aai/esr/wrapper/VnfmManagerWrapper.java
+++ b/esr-mgr/src/main/java/org/onap/aai/esr/wrapper/VnfmManagerWrapper.java
@@ -20,6 +20,7 @@ import java.util.ArrayList;
 import javax.ws.rs.core.Response;
 
 import org.onap.aai.esr.entity.aai.EsrVnfmDetail;
+import org.onap.aai.esr.entity.aai.EsrVnfmList;
 import org.onap.aai.esr.entity.rest.CommonRegisterResponse;
 import org.onap.aai.esr.entity.rest.VnfmRegisterInfo;
 import org.onap.aai.esr.externalservice.aai.ExternalSystemProxy;
@@ -27,6 +28,8 @@ import org.onap.aai.esr.util.VnfmManagerUtil;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import com.google.gson.Gson;
+
 public class VnfmManagerWrapper {
   private static VnfmManagerWrapper vnfmManagerWrapper;
   private static final Logger LOG = LoggerFactory.getLogger(VnfmManagerWrapper.class);
@@ -64,18 +67,59 @@ public class VnfmManagerWrapper {
   
   public Response queryVnfmList() {
     ArrayList<VnfmRegisterInfo> vnfmList = new ArrayList<VnfmRegisterInfo>();
-    //TODO
-    return Response.ok(vnfmList).build();
+    EsrVnfmList esrVnfm = new EsrVnfmList();
+    try {
+      String esrVnfmStr = ExternalSystemProxy.queryVnfmList();
+      esrVnfm = new Gson().fromJson(esrVnfmStr, EsrVnfmList.class);
+      vnfmList = getVnfmDetailList(esrVnfm);
+      return Response.ok(vnfmList).build();
+    } catch (Exception e) {
+      LOG.error("Query VNFM list failed !");
+      return Response.serverError().build();
+    }
+    
   }
   
   public Response queryVnfmById(String vnfmId) {
     VnfmRegisterInfo vnfm = new VnfmRegisterInfo();
-    //TODO
-    return Response.ok(vnfm).build();
+    vnfm = queryVnfmDetail(vnfmId);
+    if (vnfm != null) {
+      return Response.ok(vnfm).build();
+    } else {
+      return Response.serverError().build();
+    }
   }
   
   public Response delVnfm(String vnfmId) {
     //TODO
     return Response.noContent().build();
   }
+  
+  private VnfmRegisterInfo queryVnfmDetail(String vnfmId) {
+    VnfmRegisterInfo vnfm = new VnfmRegisterInfo();
+    EsrVnfmDetail esrVnfmDetail = new EsrVnfmDetail();
+    try {
+      String esrVnfmstr = ExternalSystemProxy.queryVnfmDetail(vnfmId);
+      LOG.info("Response from AAI by query VNFM: " + esrVnfmstr);
+      esrVnfmDetail = new Gson().fromJson(esrVnfmstr, EsrVnfmDetail.class);
+      vnfm = VnfmManagerUtil.esrVnfm2VnfmRegisterInfo(esrVnfmDetail);
+      return vnfm;
+    } catch (Exception e) {
+      LOG.error("Query VNFM detail failed! VNFM ID: " + vnfmId, e.getMessage());
+      return null;
+    }
+  }
+  
+  private ArrayList<VnfmRegisterInfo> getVnfmDetailList(EsrVnfmList esrVnfm) {
+    ArrayList<VnfmRegisterInfo> vnfmInfoList = new ArrayList<VnfmRegisterInfo>();
+    VnfmRegisterInfo vnfmInfo = new VnfmRegisterInfo();
+    for (int i=0; i<esrVnfm.getEsrVnfm().size(); i++) {
+      String vnfmId = esrVnfm.getEsrVnfm().get(i).getVnfmId();
+      vnfmInfo = queryVnfmDetail(vnfmId);
+      if(vnfmInfo != null) {
+        vnfmInfoList.add(vnfmInfo);
+      }
+    }
+    return vnfmInfoList;
+  }
 }