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) })