Merge pull request #5 from google/relay-integration

Finish Integration With Relay

Author: vaage

src/codeu/chat/RelayMain.java

@@ -16,10 +16,10 @@
 
 import java.io.IOException;
 
-import codeu.chat.common.Hub;
 import codeu.chat.relay.Server;
 import codeu.chat.relay.ServerFrontEnd;
 import codeu.chat.util.Logger;
+import codeu.chat.util.Timeline;
 import codeu.chat.util.connections.Connection;
 import codeu.chat.util.connections.ConnectionSource;
 import codeu.chat.util.connections.ServerConnectionSource;
@@ -56,39 +56,39 @@ public static void main(String[] args) {
   private static void startRelay(ConnectionSource source) {
 
     final Server relay = new Server(1024, 16);
-
     LOG.info("Relay object created.");
 
-    // TODO: Load team information
-
     final ServerFrontEnd frontEnd = new ServerFrontEnd(relay);
-
     LOG.info("Relay front end object created.");
 
-    LOG.info("Starting relay main loop...");
-
-    final Runnable hub = new Hub(source, new Hub.Handler() {
-
-      @Override
-      public void handle(Connection connection) throws Exception {
-
-        frontEnd.handleConnection(connection);
+    final Timeline timeline = new Timeline();
+    LOG.info("Relay timeline created.");
 
-      }
-
-      @Override
-      public void onException(Exception ex) {
+    // TODO: Load team information
 
-        System.out.println("ERROR: front end failed to handle connection. Check log for details.");
-        LOG.error(ex, "Exception handling connection.");
+    LOG.info("Starting relay main loop...");
 
+    while (true) {
+      try {
+
+        LOG.info("Established connection...");
+        final Connection connection = source.connect();
+        LOG.info("Connection established.");
+
+        timeline.scheduleNow(new Runnable() {
+          @Override
+          public void run() {
+            try {
+              frontEnd.handleConnection(connection);
+            } catch (Exception ex) {
+              LOG.error(ex, "Exception handling connection.");
+            }
+          }
+        });
+
+      } catch (IOException ex) {
+        LOG.error(ex, "Failed to establish connection.");
       }
-    });
-
-    LOG.info("Starting hub...");
-
-    hub.run();
-
-    LOG.info("Hub exited.");
+    }
   }
 }

src/codeu/chat/ServerMain.java

@@ -17,7 +17,6 @@
 
 import java.io.IOException;
 
-import codeu.chat.common.Hub;
 import codeu.chat.common.Relay;
 import codeu.chat.common.Secret;
 import codeu.chat.common.Uuid;
@@ -83,30 +82,21 @@ private static void runServer(Uuid id,
 
     final Server server = new Server(id, secret, relay);
 
-    LOG.info("Server object created.");
+    LOG.info("Created server.");
 
-    final Runnable hub = new Hub(serverSource, new Hub.Handler() {
+    while (true) {
 
-      @Override
-      public void handle(Connection connection) throws Exception {
+      try {
 
-        server.handleConnection(connection);
-
-      }
+        LOG.info("Established connection...");
+        final Connection connection = serverSource.connect();
+        LOG.info("Connection established.");
 
-      @Override
-      public void onException(Exception ex) {
-
-        System.out.println("ERROR: Exception during server tick. Check log for details.");
-        LOG.error(ex, "Exception during server tick.");
+        server.handleConnection(connection);
 
+      } catch (IOException ex) {
+        LOG.error(ex, "Failed to establish connection.");
       }
-    });
-
-    LOG.info("Starting hub...");
-
-    hub.run();
-
-    LOG.info("Hub exited.");
+    }
   }
 }