added input checking for scala

luke5542 · Aug 6, 2015 · b80dc57 · b80dc57 · dan-c-underwood · Aug 6, 2015
1 parent eaaa900
commit b80dc57
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/sieve-of-eratosthenes/dlang/source/app.d b/sieve-of-eratosthenes/dlang/source/app.d
@@ -16,6 +16,7 @@ void main(char[][] args)
     }
     catch(Exception e)
     {
+        writeln("Invalid input. Expected a single number.");
         exit(1);
     }
 

diff --git a/sieve-of-eratosthenes/scala/sieve_scala.scala b/sieve-of-eratosthenes/scala/sieve_scala.scala
@@ -2,7 +2,15 @@ import Array._
 
 object sieve_scala {
   def main(args: Array[String]) {
-    var max = Integer.parseInt(args(0));
+    var max = 0;
+    try {
+        max = Integer.parseInt(args(0));
+    } catch {
+        case ex: NumberFormatException => {
+            println("Invalid input. Expected a single number.");
+            System.exit(1);
+        }
+    }
 
     val start = System.currentTimeMillis;