adding some sample servlets

Author: nicegraham

src/main/java/com/seleniumgrid2api/StartGrid.java

@@ -5,9 +5,10 @@
 public class StartGrid {
 
   public static void main(String[] args) throws Exception {
-    String[] a = {"-port", "4455",
+    String[] a = {"-port", "4444",
         "-host", "localhost",
-        "-role", "hub"};
+        "-role", "hub",
+        "-servlets", "com.seleniumgrid2api.servlet.AllProxiesJsonServlet,com.seleniumgrid2api.servlet.ProxyStatusJsonServlet" };
     GridLauncher.main(a);
   }
 }

src/main/java/com/seleniumgrid2api/servlet/AllProxiesJsonServlet.java

@@ -0,0 +1,71 @@
+package com.seleniumgrid2api.servlet;
+
+
+import org.json.JSONArray;
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.openqa.grid.common.exception.GridException;
+import org.openqa.grid.internal.ProxySet;
+import org.openqa.grid.internal.Registry;
+import org.openqa.grid.internal.RemoteProxy;
+import org.openqa.grid.web.servlet.RegistryBasedServlet;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.util.Iterator;
+
+public class AllProxiesJsonServlet extends RegistryBasedServlet {
+
+  private static final long serialVersionUID = -1975392591408983229L;
+
+  public AllProxiesJsonServlet() {
+    this(null);
+  }
+
+  public AllProxiesJsonServlet(Registry registry) {
+    super(registry);
+  }
+
+  @Override
+  protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
+    doPost(req, resp);
+  }
+
+  @Override
+  protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
+    process(req, resp);
+
+  }
+
+  protected void process(HttpServletRequest request, HttpServletResponse response) throws IOException {
+    response.setContentType("text/json");
+    response.setCharacterEncoding("UTF-8");
+    response.setStatus(200);
+    JSONObject res;
+    try {
+      res = getResponse();
+      response.getWriter().print(res.toString(4));
+      response.getWriter().close();
+    } catch (JSONException e) {
+      throw new GridException(e.getMessage());
+    }
+
+  }
+
+  private JSONObject getResponse() throws IOException, JSONException {
+    JSONObject requestJSON = new JSONObject();
+    ProxySet proxies = this.getRegistry().getAllProxies();
+    Iterator<RemoteProxy> iterator = proxies.iterator();
+    JSONArray p = new JSONArray();
+    while (iterator.hasNext()) {
+      RemoteProxy eachProxy = iterator.next();
+      p.put(eachProxy.getOriginalRegistrationRequest().getAssociatedJSON());
+    }
+    requestJSON.put("Proxies", p);
+
+    return requestJSON;
+  }
+
+}

src/main/java/com/seleniumgrid2api/servlet/ProxyStatusJsonServlet.java

@@ -0,0 +1,76 @@
+package com.seleniumgrid2api.servlet;
+
+import org.json.JSONArray;
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.openqa.grid.common.exception.GridException;
+import org.openqa.grid.internal.ProxySet;
+import org.openqa.grid.internal.Registry;
+import org.openqa.grid.internal.RemoteProxy;
+import org.openqa.grid.web.servlet.RegistryBasedServlet;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.util.Iterator;
+
+public class ProxyStatusJsonServlet extends RegistryBasedServlet {
+
+  private static final long serialVersionUID = -1975392591408983229L;
+
+  public ProxyStatusJsonServlet() {
+    this(null);
+  }
+
+  public ProxyStatusJsonServlet(Registry registry) {
+    super(registry);
+  }
+
+  @Override
+  protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
+    doPost(req, resp);
+  }
+
+  @Override
+  protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
+    process(req, resp);
+
+  }
+
+  protected void process(HttpServletRequest request, HttpServletResponse response) throws IOException {
+    response.setContentType("text/json");
+    response.setCharacterEncoding("UTF-8");
+    response.setStatus(200);
+    JSONObject res;
+    try {
+      res = getResponse();
+      response.getWriter().print(res.toString(4));
+      response.getWriter().close();
+    } catch (JSONException e) {
+      throw new GridException(e.getMessage());
+    }
+
+  }
+
+  private JSONObject getResponse() throws IOException, JSONException {
+    JSONObject requestJSON = new JSONObject();
+    ProxySet proxies = this.getRegistry().getAllProxies();
+    Iterator<RemoteProxy> iterator = proxies.iterator();
+    JSONArray busyProxies = new JSONArray();
+    JSONArray freeProxies = new JSONArray();
+    while (iterator.hasNext()) {
+      RemoteProxy eachProxy = iterator.next();
+      if (eachProxy.isBusy()) {
+        busyProxies.put(eachProxy.getOriginalRegistrationRequest().getAssociatedJSON());
+      } else {
+        freeProxies.put(eachProxy.getOriginalRegistrationRequest().getAssociatedJSON());
+      }
+    }
+    requestJSON.put("BusyProxies", busyProxies);
+    requestJSON.put("FreeProxies", freeProxies);
+
+    return requestJSON;
+  }
+
+}