WordsCount

import java.io.File
import scala.io.Source

class WordsCount{
  def subdirs(dir: File): Iterator[File] = {
    val children = dir.listFiles.filter(_.isFile)
    children.toIterator
  }

  def getWordNum(file: Iterator[File]):scala.collection.mutable.Map[String, Int]={
    val wordsMap = scala.collection.mutable.Map[String, Int]()
      file.foreach(file => Source.fromFile(file).getLines().foreach(line=>line.split(" ").foreach(
        word=>{
          if (wordsMap.contains(word)){
            wordsMap(word)+=1
          }
          else{
            wordsMap(word)=1
          }
        }
      )))
    wordsMap
  }

}
object WordCount {
  def main(args: Array[String]): Unit = {
    val wc = new WordsCount()
    val dir = new File("E:\\scala")
    val wordsMap = wc.getWordNum(wc.subdirs(dir))
    for ((k, v) <- wordsMap)
    {
      println(k+":"+v)
    }
  }
}