Try with if-else

apache · Apr 27, 2017 · b5c9839 · b5c9839
1 parent 0f55c49
commit b5c9839
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/core/src/main/scala/org/apache/spark/util/Utils.scala b/core/src/main/scala/org/apache/spark/util/Utils.scala
@@ -740,9 +740,13 @@ private[spark] object Utils extends Logging {
    * always return a single directory.
    */
   def getLocalDir(conf: SparkConf): String = {
-    getOrCreateLocalRootDirs(conf).headOption.getOrElse {
-      val dirPaths = getConfiguredLocalDirs(conf)
-      throw new IOException(s"Failed to get a temp directory under [${dirPaths.mkString(",")}].")
+    val localRootDirs = getOrCreateLocalRootDirs(conf)
+    if (localRootDirs.nonEmpty) {
+      localRootDirs(0)
+    } else {
+      val configuredLocalDirs = getConfiguredLocalDirs(conf)
+      throw new IOException(
+        s"Failed to get a temp directory under [${configuredLocalDirs.mkString(",")}].")
     }
   }