Refactor: Update RouteCommand.java

Author: SauravBhowmick

src/main/java/com/mycmd/commands/RouteCommand.java

@@ -5,6 +5,7 @@
 import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStreamReader;
+import java.util.concurrent.TimeUnit;
 
 /**
  * Displays and modifies entries in the local IP routing table.
@@ -45,20 +46,22 @@ public void execute(String[] args, ShellContext context) throws IOException {
       }
 
       Process process = pb.start();
-      BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
-      BufferedReader errorReader =
-          new BufferedReader(new InputStreamReader(process.getErrorStream()));
+      try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
+      BufferedReader errorReader = new BufferedReader(new InputStreamReader(process.getErrorStream()))) {
+        String line;
+          while ((line = reader.readLine()) != null) {
+            System.out.println(line);
+          }
 
-      String line;
-      while ((line = reader.readLine()) != null) {
-        System.out.println(line);
+          while ((line = errorReader.readLine()) != null) {
+            System.err.println(line);
+          }
       }
 
-      while ((line = errorReader.readLine()) != null) {
-        System.err.println(line);
-      }
-
-      process.waitFor();
+        if (!process.waitFor(30, TimeUnit.SECONDS)) {
+          process.destroyForcibly();
+          System.out.println("Command timed out.");
+        }
 
     } catch (Exception e) {
       System.out.println("Error executing route: " + e.getMessage());