# encoding: utf-8module ProjectCodeStatistics module FetchFile def recursion_fetch(dir=nil) dir = @base_dir if dir == nil Dir.foreach(dir).each do |file_dir| dir_path = File.join(dir,file_dir) if File.directory?(dir_path) then recursion_fetch(dir_path) unless file_dir =~ /^\./ else if [/\.java$/,/\.jsp$/,/\.js$/,/\.xml$/].any?{|reg| reg=~file_dir} @all_file_with_row_number[dir_path.gsub(@base_dir,"")] = File.new(dir_path).readlines.size end end end end end class Calculate include FetchFile attr_reader :all_file_with_row_number def initialize(dir) @base_dir, @all_file_with_row_number = dir, {} recursion_fetch end # 分类统计所有文件数量 def file_total_numbers() row_numbers(:xml,:js,:jsp,:java,@all_file_with_row_number) end # 分类统计新加入的文件代码行总量的 def add_file_row_numbers(other) add_file_keys = @all_file_with_row_number.keys - other.all_file_with_row_number.keys add_file_hash = @all_file_with_row_number.find_all{|key,value| add_file_keys.include? key} row_numbers(:xml,:js,:jsp,:java,add_file_hash) end # 分类统计新加入的各种类型文件总量 def compare_add_file_count(other) add_file_keys = @all_file_with_row_number.keys - other.all_file_with_row_number.keys file_numbers(:xml,:js,:jsp,:java,add_file_keys) end # 分类统计修改过的文件 def compare_each_modify_file(other) file_hash = @all_file_with_row_number calculated_hash =other.all_file_with_row_number.find_all.inject({}) do |hash,item| hash[item[0]] = (file_hash[item[0]] - item[1]) if file_hash.include?(item[0]) && file_hash[item[0]] != item[1] hash end row_numbers(:xml,:js,:jsp,:java,calculated_hash) end def file_numbers(*parameter) last = parameter.reverse!.shift parameter.reverse.inject([]) do |total,type| regexp = eval('/' + type.to_s + '$/') total.push(last.find_all{|key,value| regexp =~ key}.size) end end def row_numbers(*parameter) last = parameter.reverse!.shift parameter.reverse.inject([]) do |total,type| regexp = eval('/' + type.to_s + '$/') total.push last.find_all{|key,value| regexp =~ key}.inject(0){|sum,item| sum + item[1] } end end end class Statistics def initialize(new_project,old_project,new_desc="新工程",old_desc="老工程") @new_project = Calculate.new(new_project) @old_project = Calculate.new(old_project) @new_desc,@old_desc = new_desc,old_desc end def print old_total = @old_project.file_total_numbers new_total = @new_project.file_total_numbers puts '系统代码统计' puts "#{@old_desc}代码总量统计，总计：#{old_total.reduce(:+)}行， java：#{old_total[3]}行,jsp: #{old_total[2]}行,js: #{old_total[1]}行,sql: #{old_total[0]}行 " puts "#{@new_desc}代码总量统计，总计：#{new_total.reduce(:+)}行，java：#{new_total[3]}行,jsp: #{new_total[2]}行,js: #{new_total[1]}行,sql: #{new_total[0]}行 " puts "新增总量 #{new_total.reduce(:+) - old_total.reduce(:+)}行" add_files = @new_project.compare_add_file_count(@old_project) new_rows = @new_project.add_file_row_numbers(@old_project) modify_rows = @new_project.compare_each_modify_file(@old_project) total_change = (new_rows.zip modify_rows).inject([]) {|n,a|n<<a.reduce(:+)} puts puts "新增加的文件数量，java：#{add_files[3]}个，jsp: #{add_files[2]}个，js: #{add_files[1]}个， sql: #{add_files[0]}个 " puts puts "新增加的文件代码行，java：#{new_rows[3]}行，jsp: #{new_rows[2]}行，js: #{new_rows[1]}行， sql: #{new_rows[0]}行 " puts "新增加的代码行， java：#{modify_rows[3]}行，jsp: #{modify_rows[2]}行，js: #{modify_rows[1]}行， sql: #{modify_rows[0]}行 " puts "新增加的代码行总计，java：#{total_change[3]}行，jsp: #{total_change[2]}行，js: #{total_change[1]}行， sql: #{total_change[0]}行 " puts "新增总量 #{new_rows.reduce(:+) + modify_rows.reduce(:+)}行" end endends = ProjectCodeStatistics::Statistics.new('D:\boss\Code\udrm\src','D:\boss\Code\digital20110726\src')s.print
坚守自己的原则，世界上的诱-惑很多，
