changed logic

andersonkmi · Nov 24, 2019 · 7cf0f9b · 7cf0f9b
1 parent 3bc3e59
commit 7cf0f9b
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/src/main/scala/org/codecraftlabs/spark/utils/FileUtil.scala b/src/main/scala/org/codecraftlabs/spark/utils/FileUtil.scala
@@ -7,19 +7,14 @@ object FileUtil {
   def getListOfFiles(dir: String): List[String] = {
     val root = new File(dir)
 
-    if (root.exists && root.isDirectory) {
-      getRecursiveListOfFiles(root).map(item => item.getAbsolutePath).filter(item => item.endsWith(".csv") || item.endsWith(".parquet")).toList
-    } else {
-      List[String]()
-    }
+    if (!(root.exists && root.isDirectory)) List.empty[String] else getRecursiveListOfFiles(root).map(item => item.getAbsolutePath).filter(item => item.endsWith(".csv") || item.endsWith(".parquet")).toList
   }
 
   def getRecursiveListOfFiles(dir: File): Array[File] = {
     val these = dir.listFiles
     these ++ these.filter(_.isDirectory).flatMap(getRecursiveListOfFiles)
   }
 
-
   def buildFilePath(folder: String, fileName: String): String = {
     s"$folder" + getDefault.getSeparator + fileName
   }