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

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

}