1,獲取指定類型文件
def getFile(file:File): Array[File] ={
val files = file.listFiles().filter(! _.isDirectory)
.filter(t => t.toString.endsWith(".csv")) //此處讀取.txt and .csv文件
files ++ file.listFiles().filter(_.isDirectory).flatMap(getFile)
}
2,創建文件 指定文件夾,獲取文件夾下的類型文件,再導入數據
val path = new File("D:/test/")
getFile(path).foreach(f=> {
val sourceFile_city1 = Source.fromFile(f)
Data2SQL_citydataframe_color_result.connsqlserver(sourceFile_city1)
})
