Fixed issue with host-disconnect-host

Resolvd #74 simply by adding the timeout for when the user can click the host button again.
KyleRAnderson · Jan 20, 2019 · f1d0f9a · f1d0f9a
1 parent 3ad5acc
commit f1d0f9a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 7 deletions.
diff --git a/ICS4UC_RST/src/menu/MainMenu.java b/ICS4UC_RST/src/menu/MainMenu.java
@@ -139,6 +139,7 @@ public class MainMenu extends Application {
     public MainMenu() {
         currentInstance = this;
         PartyHandler.setIncomingMessageListener(this::messageReceived);
+        PartyHandler.setOnReceiverClosed(this::receiverClosed);
 
         StringBuilder builder = new StringBuilder(MENU_HELP_TEXT);
         for (Game game : games) {
@@ -147,6 +148,18 @@ public MainMenu() {
         helpText = builder.toString();
     }
 
+    /**
+     * Called when the party's receiving thread is closed.
+     */
+    private void receiverClosed() {
+        hostMenuItem.setDisable(false);
+        hostMenuItem.setConnected(false);
+        connectMenuItem.setConnected(false);
+        connectMenuItem.setDisable(false);
+        hostMenuItem.setActive(false);
+        connectMenuItem.setActive(false);
+    }
+
     /**
      * Gets the current instance of the main menu.
      *
@@ -406,18 +419,14 @@ private void disconnect(boolean notifyOtherClient) {
             connectTask.cancel(true);
         }
 
-        if (notifyOtherClient) {
+        if (notifyOtherClient && PartyHandler.isConnected()) {
             NetworkMessage message = new NetworkMessage(HostStatus.DISCONNECTING);
             sendNetworkMessage(message);
+            hostMenuItem.setDisable(true);
         } else {
             PartyHandler.disconnect();
+            receiverClosed();
         }
-        hostMenuItem.setConnected(false);
-        connectMenuItem.setConnected(false);
-        hostMenuItem.setDisable(false);
-        connectMenuItem.setDisable(false);
-        hostMenuItem.setActive(false);
-        connectMenuItem.setActive(false);
     }
 
     /**

diff --git a/ICS4UC_RST/src/network/party/PartyHandler.java b/ICS4UC_RST/src/network/party/PartyHandler.java
@@ -31,6 +31,7 @@ public class PartyHandler {
     private static ReceiverTask incomingTask;
     private static SenderTask outgoingTask;
     private static Consumer<ReceivedDataEvent> incomingListener;
+    private static Runnable receiverClosedListener;
 
     /**
      * Begins a party session with the user at the given IP address.
@@ -230,6 +231,18 @@ public static ExecutorService createFixedTimeoutExecutorService(
      */
     private static void receiverClosed() {
         disconnect();
+        if (receiverClosedListener != null) {
+            receiverClosedListener.run();
+        }
+    }
+
+    /**
+     * Sets an action to be performed when the receiving thread is closed.
+     *
+     * @param action The action to run.
+     */
+    public static void setOnReceiverClosed(Runnable action) {
+        receiverClosedListener = action;
     }
 
     /**