【Benchmark】代码执行效率测试脚本

张咚咚

脚本如下，设置可随意更改，打印完美输出。

图片预览

1. #encoding:utf-8
   
2. #==============================================================================
   
3. # ■ Benchmark v1.0 By QQEat
   
4. #------------------------------------------------------------------------------
   
5. # 　基准测试(优雅 + 完美强迫症打印)
   
6. #==============================================================================
   
7. #
   
8. #  # 打印代码块执行时间(count 默认 1, 可在下面设置更改)
   
9. #  Benchmark.measure(count){|r| ... }
   
10. #
    
11. #  # 测试代码快效率
    
12. #  Benchmark.measure(10_000) { [*1..1000].shuffle }
    
13. #
    
14. #  # 汇报打印
    
15. #  Benchmark.measure do |x|
    
16. #    a = (1..100).to_a.shuffle!
    
17. #    x.report('Array#sort', 100000)  { a.sort  }
    
18. #    x.report('Array#sort!', 100000) { a.sort! }
    
19. #    10.times{ x.report{ a.sort! } }
    
20. #  end
    
21. #
    
22. 
    
23. module Benchmark
    
24.   #--------------------------------------------------------------------------
    
25.   # ● 常量
    
26.   #--------------------------------------------------------------------------
    
27.   ITERATIONS = 1 # 默认执行次数
    
28.   MARGIN = 3 # 列间距
    
29.   SPLIT_LINE = true # 是否显示列分割线
    
30.   COLS = [
    
31.     [7, '#', :R],
    
32.     [10, 'user', :R],   # 用户时间
    
33.     [10, 'system', :R], # 系统时间
    
34.     [10, 'real', :R],   # 真实用时
    
35.     [10, 'avg', :R],    # 平均用时
    
36.     [10, 'times', :R],  # 执行次数
    
37.   ].map.with_index{|v, i|
    
38.     [v[0], v[1].send(v[2] == :L ? :ljust : :rjust, v[0]) + ' ' * MARGIN, v[2]]
    
39.   } # [列宽, 列名, 对齐方向]
    
40.   
    
41.   #==============================================================================
    
42.   # ■ Reporter
    
43.   #==============================================================================
    
44.   class Reporter
    
45.     #--------------------------------------------------------------------------
    
46.     # ● 初始化
    
47.     #--------------------------------------------------------------------------
    
48.     def initialize
    
49.       @total = [0, 0, 0, 0]
    
50.     end
    
51.     #--------------------------------------------------------------------------
    
52.     # ● 汇报执行(标签名, 执行次数)
    
53.     #--------------------------------------------------------------------------
    
54.     def report(label = nil, iterate = ITERATIONS)
    
55.       result = Benchmark.send(:time, iterate) { yield }
    
56.       result << result[-1] / iterate
    
57.       Benchmark.send(:output, label, *result.map{|v| '%0.6f' % v }, iterate)
    
58.       @total.map!.with_index { |t, i| t += result[i].to_f }
    
59.     end
    
60.   end
    
61.   #--------------------------------------------------------------------------
    
62.   # ● measure 测量脚本执行时间
    
63.   #--------------------------------------------------------------------------
    
64.   def self.measure(iterate = ITERATIONS, &block)
    
65.     @row = 0
    
66.     width = COLS.inject(0){|a,b| a+b[1].length } / 2 - 5
    
67.     puts "#{'-'*width} BENCHMARK #{'-'*width}"
    
68.     puts COLS.map{|v| v[1] }.join
    
69.     reporter = Reporter.new
    
70.     if block.arity == 0
    
71.       result = time(iterate) { yield }
    
72.     else
    
73.       yield reporter
    
74.       result = reporter.instance_variable_get(:@total)
    
75.     end
    
76.     print "\n"
    
77.     output('Total', *result.map{|v| '%0.6f' % v }, '')
    
78.     print "\n"
    
79.   end
    
80.   
    
81.   class << self
    
82.     private
    
83.     #--------------------------------------------------------------------------
    
84.     # ● 执行脚本(执行次数)
    
85.     #   return：[用户时间, 系统时间, 真实时间]
    
86.     #--------------------------------------------------------------------------
    
87.     def time(iterate = ITERATIONS)
    
88.       t = Process.times
    
89.       initial = [t.utime, t.stime, Time.now]
    
90.       iterate.times { yield }
    
91.       t2 = Process.times
    
92.       [t2.utime, t2.stime, Time.now].map!.with_index do |time, index|
    
93.         time - initial[index]
    
94.       end
    
95.     end
    
96.     #--------------------------------------------------------------------------
    
97.     # ● 打印输出
    
98.     #--------------------------------------------------------------------------
    
99.     def output(*args)
    
100.       args[-1] = args[-1].to_s.gsub(/(?=(?!\b)(\d{3})+$)/, ',')
     
101.       (args[0] ||= @row.to_s) << ':'
     
102.       args.map!.with_index{|v,i| v.to_s.scan(/.{1,#{COLS[i][0]}}/) }
     
103.       row = args.max_by{|v| v.length }.length
     
104.       row.times do |i|
     
105.         bottom = i == row-1
     
106.         args.each_with_index do |arr, n|
     
107.           begin_ = n == 0
     
108.           end_ = n == args.length - 1
     
109.           col = COLS[n]
     
110.           diff = args[0].length - arr.length
     
111.           diff = 0 if diff < 0
     
112.           str = (i < diff ? '' : arr[i-diff]) || ''
     
113.           str = str.send(col[2] == :L ? :ljust : :rjust, col[0])
     
114.           if SPLIT_LINE and !end_
     
115.             hmar = (MARGIN - 1) / 2
     
116.             str << ' ' * hmar << '|' << ' ' * (MARGIN - hmar - 1)
     
117.           else
     
118.             str << ' ' * MARGIN
     
119.           end
     
120.           str << "\n" if end_
     
121.           print str
     
122.         end
     
123.       end
     
124.       @row += 1
     
125.     end
     
126.   end
     
127. end

复制代码

百里_飞柳

2302年了，居然还能看到rgss3的基础组件