blob: d22236e9fbf6580c92a06a8fc7f22f104c4977f3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
require 'benchmark_driver/output/simple'
# This replicates the legacy benchmark/driver.rb behavior.
class BenchmarkDriver::Output::Driver < BenchmarkDriver::Output::Simple
def initialize(*)
super
@stdout = $stdout
@strio = StringIO.new
$stdout = IOMultiplexer.new(@stdout, @strio)
end
def with_benchmark(*)
super
ensure
logfile = "bmlog-#{Time.now.strftime('%Y%m%d-%H%M%S')}.#{$$}.log"
puts "\nLog file: #{logfile}"
$stdout = @stdout
File.write(logfile, @strio.tap(&:rewind).read)
end
class IOMultiplexer
def initialize(io1, io2)
@io1 = io1
@io2 = io2
end
[:write, :sync, :sync=, :puts, :print, :flush].each do |method|
define_method(method) do |*args|
@io1.send(method, *args)
@io2.send(method, *args)
end
end
end
private_constant :IOMultiplexer
end
|