Added timeout to call also cancelNow() path in case something hangs badly

1 files changed, 4 insertions, 1 deletions

diff --git a/src/main/java/com/it_jaros/jns/App.java b/src/main/java/com/it_jaros/jns/App.java
index 7fb2415..322f950 100644
--- a/src/main/java/com/it_jaros/jns/App.java
+++ b/src/main/java/com/it_jaros/jns/App.java
@@ -6,6 +6,7 @@ import com.it_jaros.jns.scan.Scanner;
 
 import java.io.IOException;
 import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
 
 public class App {
 
@@ -80,7 +81,9 @@ public class App {
     private static void shutdown(Scanner scanner) {
         try {
             scanner.cancel();
-            shutdownLatch.await();
+            if (!shutdownLatch.await(5, TimeUnit.SECONDS)) {
+                scanner.cancelNow();
+            }
         } catch (InterruptedException e) {
             Thread.currentThread().interrupt();
         } catch (Exception e) {