统计目录下面的文件所有单词

1.作业：统计目录下面的文件所有单词

2.代码实现
package
com.jn.scala.wangjialin.copy

import
java.io.File

import
scala.io.Source

/**

* Created by admin on 2016/1/9.

*/

object
WordCounter {

//存储单词和个数

var
map
=
Map.
empty[String,
Int]

def
main(args: Array[String
]) {

val
files2 = subFileNames(
new
File(
"d:/test"))

for(i <- files2){

readFile(i)

println(i)

}

map.foreach(f =>println(f))

}

/**

* 递归调用获取所有文件的全路径

* @param dir


*
@return


*/

def
subFileNames(dir: File):Iterator
[File]={

val
d = dir.listFiles.filter(_.isDirectory)

val
f = dir.listFiles.filter(_.isFile).toIterator

f ++ d.toIterator.flatMap(subFileNames(_))

}

/**

* 获取文件的每一个行

* @param file


*/

def
readFile(file: File){

val
f = Source.fromFile(file)

for
(line <- f.getLines()){

count(line)

}

}

/**

* 进行计算

* @param line


*/

def
count(line:
String
) = {

for
(word <- line.split(" ")){

if
(map.contains(word))

map
+= (word -> (map(word)+
1))

else


map
+= (word ->
1)

}

}
}