Add Utils method to exit application after a timeout

chrishantha · chrishantha · commit 6215dcebcf4e · 2018-07-05T14:18:29.000+05:30
diff --git a/base/src/main/java/com/github/chrishantha/sample/base/Utils.java b/base/src/main/java/com/github/chrishantha/sample/base/Utils.java
@@ -0,0 +1,27 @@
+package com.github.chrishantha.sample.base;
+
+import java.util.Timer;
+import java.util.TimerTask;
+
+/**
+ * Some basic utilities for sample applications
+ */
+public class Utils {
+
+    /**
+     * Exit application after a given number of seconds.
+     *
+     * @param timeout Timeout in seconds
+     */
+    public static void exitApplication(int timeout) {
+        if (timeout > 0) {
+            final Timer timer = new Timer();
+            timer.schedule(new TimerTask() {
+                public void run() {
+                    System.out.println("Exiting now...");
+                    System.exit(0);
+                }
+            }, timeout * 1000);
+        }
+    }
+}
diff --git a/highcpu/src/main/java/com/github/chrishantha/sample/highcpu/HighCpuApplication.java b/highcpu/src/main/java/com/github/chrishantha/sample/highcpu/HighCpuApplication.java
@@ -17,9 +17,7 @@
 
 import com.beust.jcommander.Parameter;
 import com.github.chrishantha.sample.base.SampleApplication;
-
-import java.util.Timer;
-import java.util.TimerTask;
+import com.github.chrishantha.sample.base.Utils;
 
 public class HighCpuApplication implements SampleApplication {
 
@@ -53,15 +51,7 @@ public class HighCpuApplication implements SampleApplication {
     @Override
     public void start() {
         System.out.println("Starting Application...");
-        if (exitTimeoutInSeconds > 0) {
-            final Timer timer = new Timer();
-            timer.schedule(new TimerTask() {
-                public void run() {
-                    System.out.println("Exiting now...");
-                    System.exit(0);
-                }
-            }, exitTimeoutInSeconds * 1000);
-        }
+        Utils.exitApplication(exitTimeoutInSeconds);
         if (runHashing) {
             for (int i = 0; i < hashingWorkers; i++) {
                 startThread(i, "Hashing", new HashingWorker(hashDataLength, hashingAlgorithm));
