From bac2fa8e7e6286d64b50cdf6b7fd32f958080ceb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kjetil=20=C3=98rbekk?= <kjetil.orbekk@gmail.com>
Date: Mon, 6 Feb 2012 13:44:04 +0100
Subject: Add a class that holds server information.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

– Provides access to the context and port, used in testing.
---
 .../main/java/com/orbekk/same/SameController.java  | 13 ++----
 .../java/com/orbekk/same/http/ServerBuilder.java   |  6 +--
 .../java/com/orbekk/same/http/ServerContainer.java | 51 ++++++++++++++++++++++
 3 files changed, 58 insertions(+), 12 deletions(-)
 create mode 100644 same/src/main/java/com/orbekk/same/http/ServerContainer.java

(limited to 'same/src/main/java')

diff --git a/same/src/main/java/com/orbekk/same/SameController.java b/same/src/main/java/com/orbekk/same/SameController.java
index 4c923d9..2faf669 100644
--- a/same/src/main/java/com/orbekk/same/SameController.java
+++ b/same/src/main/java/com/orbekk/same/SameController.java
@@ -1,25 +1,20 @@
 package com.orbekk.same;
 
-import org.eclipse.jetty.server.Handler;
-import org.eclipse.jetty.server.Server;
-import org.eclipse.jetty.server.handler.HandlerList;
-import org.eclipse.jetty.servlet.ServletContextHandler;
-import org.eclipse.jetty.servlet.ServletHolder;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import com.googlecode.jsonrpc4j.JsonRpcServer;
 import com.orbekk.net.HttpUtil;
 import com.orbekk.paxos.PaxosService;
 import com.orbekk.paxos.PaxosServiceImpl;
 import com.orbekk.same.config.Configuration;
 import com.orbekk.same.http.ServerBuilder;
+import com.orbekk.same.http.ServerContainer;
 import com.orbekk.same.http.StateServlet;
 
 public class SameController {
     private Logger logger = LoggerFactory.getLogger(getClass());
     private int port;
-    private Server server;
+    private ServerContainer server;
     private MasterServiceImpl master;
     private ClientServiceImpl client;
     private PaxosServiceImpl paxos;
@@ -41,7 +36,7 @@ public class SameController {
         ClientServiceImpl client = new ClientServiceImpl(state, connections);
         PaxosServiceImpl paxos = new PaxosServiceImpl("");
         
-        Server server = new ServerBuilder(port)
+        ServerContainer server = new ServerBuilder(port)
                 .withServlet(new StateServlet(), "/_/state")
                 .withService(client.getService(), ClientService.class)
                 .withService(master, MasterService.class)
@@ -55,7 +50,7 @@ public class SameController {
     
     public SameController(
             int port,
-            Server server,
+            ServerContainer server,
             MasterServiceImpl master,
             ClientServiceImpl client,
             PaxosServiceImpl paxos) {
diff --git a/same/src/main/java/com/orbekk/same/http/ServerBuilder.java b/same/src/main/java/com/orbekk/same/http/ServerBuilder.java
index a582283..ff29e43 100644
--- a/same/src/main/java/com/orbekk/same/http/ServerBuilder.java
+++ b/same/src/main/java/com/orbekk/same/http/ServerBuilder.java
@@ -32,9 +32,9 @@ public class ServerBuilder {
         return withServlet(new RpcServlet(server), pathSpec);
     }
     
-    public Server build() {
-        Server server = new Server(port);
-        server.setHandler(getServletContextHandler());
+    public ServerContainer build() {
+        ServerContainer server = ServerContainer.create(port); 
+        server.setContext(getServletContextHandler());
         return server;
     }
 
diff --git a/same/src/main/java/com/orbekk/same/http/ServerContainer.java b/same/src/main/java/com/orbekk/same/http/ServerContainer.java
new file mode 100644
index 0000000..cd4dc4d
--- /dev/null
+++ b/same/src/main/java/com/orbekk/same/http/ServerContainer.java
@@ -0,0 +1,51 @@
+package com.orbekk.same.http;
+
+import org.eclipse.jetty.server.Server;
+import org.eclipse.jetty.servlet.ServletContextHandler;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class ServerContainer {
+    Logger logger = LoggerFactory.getLogger(getClass());
+    Server server;
+    int port;
+    ServletContextHandler context = null;
+    
+    public ServerContainer(Server server, int port, ServletContextHandler context) {
+        this.server = server;
+        this.port = port;
+        this.context = context;
+    }
+    
+    public static ServerContainer create(int port) {
+        Server server = new Server(port);
+        return new ServerContainer(server, port, null);
+    }
+    
+    public void setContext(ServletContextHandler context) {
+        server.setHandler(context);
+        this.context = context;
+    }
+    
+    public int getPort() {
+        if (port == 0) {
+            return server.getConnectors()[0].getLocalPort();
+        } else {
+            return port;
+        }
+    }
+    
+    public void start() throws Exception {
+        server.start();
+        logger.info("Started server on port {}", getPort());
+    }
+ 
+    public void stop() throws Exception {
+        server.stop();
+        logger.info("Server stopped.");
+    }
+    
+    public void join() throws InterruptedException {
+        server.join();
+    }
+}
-- 
cgit v1.2.3