exchange with GNPy to check path feasibility

[transportpce.git] / pce / src / main / java / org / opendaylight / transportpce / pce / gnpy / ConnectToGnpyServer.java

diff --git a/pce/src/main/java/org/opendaylight/transportpce/pce/gnpy/ConnectToGnpyServer.java b/pce/src/main/java/org/opendaylight/transportpce/pce/gnpy/ConnectToGnpyServer.java
new file mode 100644 (file)
index 0000000..6a6a0d8
--- /dev/null
+++ b/pce/src/main/java/org/opendaylight/transportpce/pce/gnpy/ConnectToGnpyServer.java
@@ -0,0 +1,111 @@
+/*
+ * Copyright © 2019 Orange, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.transportpce.pce.gnpy;
+
+import com.google.common.io.CharStreams;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.io.OutputStream;
+import java.net.HttpURLConnection;
+import java.net.MalformedURLException;
+import java.net.URL;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Class to connect to GNPy tool.
+ *
+ * @author Ahmed Triki ( ahmed.triki@orange.com )
+ *
+ */
+
+public class ConnectToGnpyServer {
+
+    private static final Logger LOG = LoggerFactory.getLogger(ConnectToGnpyServer.class);
+    static final String URL_GNPY = "http://127.0.0.1:5000/gnpy/api/v1.0/files";
+
+    public String gnpyCnx(String jsonTxt) {
+        String jsonRespTxt = null;
+        try {
+            URL url = new URL(URL_GNPY);
+            String userCredentials = "gnpy:gnpy";
+            String basicAuth = "Basic " + new String(java.util.Base64.getEncoder().encode(userCredentials.getBytes()));
+
+            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
+            conn.setDoOutput(true);
+            conn.setRequestMethod("POST");
+            conn.setRequestProperty("Authorization", basicAuth);
+            conn.setRequestProperty("Content-Type", "application/json");
+
+            // Send the request to the GNPy
+            OutputStream os = conn.getOutputStream();
+            os.write(jsonTxt.getBytes());
+            os.flush();
+            if (conn.getResponseCode() != HttpURLConnection.HTTP_CREATED) {
+                throw new RuntimeException("Failed : HTTP error code : " + conn.getResponseCode());
+            }
+            InputStreamReader response = new InputStreamReader((conn.getInputStream()));
+            if (response != null) {
+                jsonRespTxt = null;
+                try {
+                    jsonRespTxt = CharStreams.toString(response);
+                } catch (IOException e1) {
+                    LOG.warn("Could not read characters of GNPy response: {}", e1.getMessage());
+                }
+            }
+            conn.disconnect();
+        } catch (MalformedURLException e) {
+            LOG.warn("Exception : Malformed GNPy URL");
+        } catch (IOException e) {
+            LOG.warn("IOException when connecting to GNPy server: {}", e.getMessage());
+        }
+        return jsonRespTxt;
+    }
+
+    public boolean isGnpyURLExist() {
+        boolean exist = false;
+        try {
+            URL url = new URL(URL_GNPY);
+            String userCredentials = "gnpy:gnpy";
+            String basicAuth = "Basic " + new String(java.util.Base64.getEncoder().encode(userCredentials.getBytes()));
+            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
+            conn.setDoOutput(true);
+            conn.setRequestMethod("HEAD");
+            conn.setRequestProperty("Authorization", basicAuth);
+            conn.setRequestProperty("Content-Type", "application/json");
+            conn.connect();
+            if (conn.getResponseCode() == HttpURLConnection.HTTP_OK) {
+                exist = true;
+            }
+            conn.disconnect();
+        }
+        catch (IOException e) {
+            LOG.warn("Could not connect to GNPy server");
+        }
+        return exist;
+    }
+
+    public String readResponse(InputStreamReader response) {
+        String output = null;
+        BufferedReader br = new BufferedReader(response);
+        String line;
+        StringBuilder sb = new StringBuilder();
+        try {
+            while ((line = br.readLine()) != null) {
+                sb.append(line);
+            }
+            output = sb.toString();
+        } catch (IOException e) {
+            LOG.warn("IOException when reading GNPy response: {}", e.getMessage());
+        }
+        return output;
+    }
+}