fix: client disconnect and engine stopiing

Author: mkutay

src/api/WebSocketMessageHandler.java

@@ -42,6 +42,11 @@ public void handleMessage(String message) {
       return;
     }
 
+    if (engine != null) {
+      logger.warn("Simulation is already running. Stopping the current simulation before starting a new one.");
+      stopEngine();
+    }
+
     // Get and set the simulation data:
     SimulationData data = null;
     try {
@@ -57,7 +62,7 @@ public void handleMessage(String message) {
     engine.start();
   }
 
-  private void stopEngine() {
+  public void stopEngine() {
     if (engine != null) {
       engine.stop();
     }

src/api/WebSocketServer.java

@@ -179,7 +179,7 @@ private void handleRead(SelectionKey key) {
         }
       }
 
-    } catch (IOException e) {
+    } catch (Exception e) {
       logger.warn("Error reading from client", e);
       clients.remove(clientChannel);
       try {
@@ -188,6 +188,11 @@ private void handleRead(SelectionKey key) {
         // Ignore
       }
       key.cancel();
+    } finally {
+      if (clients.isEmpty()) {
+        logger.info("No clients connected, stopping");
+        messageHandler.stopEngine();
+      }
     }
   }
 

src/graphics/RenderPanelWeb.java

@@ -74,7 +74,9 @@ public void drawTransparentRect(int x, int y, int width, int height, Color color
 	 */
 	public void update() {
 		double nowTime = System.nanoTime();
-		logger.info("Time since last tick: " + (nowTime - lastTick) * 1e-6 + " ms");
+		if ((nowTime - lastTick) * 1e-6 > 50) {
+			logger.info("Time since last tick: " + (nowTime - lastTick) * 1e-6 + " ms");
+		}
 		lastTick = nowTime;
 
 		Gson g = new Gson();