From d867a4b7a229826e60bc53c251032bc6dc7cac4e Mon Sep 17 00:00:00 2001
From: Alistair McIntyre <ajm412@uclive.ac.nz>
Date: Thu, 17 Aug 2017 14:53:54 +1200
Subject: [PATCH] Fixed bug, should fix disconnection issues. tags : #issue[47]

---
 src/main/java/seng302/visualiser/ClientToServerThread.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/main/java/seng302/visualiser/ClientToServerThread.java b/src/main/java/seng302/visualiser/ClientToServerThread.java
index 3d6152d8..57256760 100644
--- a/src/main/java/seng302/visualiser/ClientToServerThread.java
+++ b/src/main/java/seng302/visualiser/ClientToServerThread.java
@@ -338,6 +338,8 @@ public class ClientToServerThread implements Runnable {
             closeSocket();
         }
         if (currentByte == -1) {
+            notifyDisconnectListeners("Cannot read from server.");
+            closeSocket();
             logger.warn("InputStream reach end of stream", 1);
         }
         return currentByte;