Fixed bug when connecting to client immediately after registering it

iguanastin · Dec 5, 2017 · e304eaa · e304eaa
1 parent b4fb1a5
commit e304eaa
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/main/java/edu/colostate/cs/cs414/hnefatafl/client/gui/RegisterController.java b/src/main/java/edu/colostate/cs/cs414/hnefatafl/client/gui/RegisterController.java
@@ -76,7 +76,14 @@ public void cancelRegisterOnAction(ActionEvent event) {
 
     private void closeRegisterWindow() {
         Platform.runLater(() -> {
-            if (client != null) client.removeRegisterListener(this);
+            if (client != null) {
+                client.removeRegisterListener(this);
+                try {
+                    client.disconnect();
+                } catch (IOException e) {
+                    logger.error("Error disconnecting client", e);
+                }
+            }
             registerUsernameField.getScene().getWindow().hide();
         });
     }