97 lines
2.2 KiB
Ruby
97 lines
2.2 KiB
Ruby
#!/usr/bin/env ruby
|
|
$LOAD_PATH.unshift(File.expand_path("../../lib", __FILE__))
|
|
|
|
def set_mongo_driver_mode(mode)
|
|
case mode
|
|
when :c
|
|
ENV.delete('TEST_MODE')
|
|
ENV['C_EXT'] = 'TRUE'
|
|
when :ruby
|
|
ENV['TEST_MODE'] = 'TRUE'
|
|
ENV.delete('C_EXT')
|
|
else
|
|
raise 'mode must be :c or :ruby'
|
|
end
|
|
ENV['MONGO_DRIVER_MODE'] = mode.to_s
|
|
end
|
|
|
|
$mode = ARGV[0].to_sym if ARGV[0]
|
|
set_mongo_driver_mode($mode || :c)
|
|
|
|
require 'rubygems'
|
|
require 'mongo'
|
|
require 'benchmark'
|
|
require 'ruby-prof'
|
|
require 'perftools'
|
|
|
|
def array_size_fixnum(base, power)
|
|
n = base ** power
|
|
return [n, {n.to_s => Array.new(n, n)}]
|
|
end
|
|
|
|
def array_size_fixnum(base, power)
|
|
n = base ** power
|
|
return [n, {n.to_s => Array.new(n, n)}]
|
|
end
|
|
|
|
def insert(coll, h)
|
|
h.delete(:_id) # delete :_id to insert
|
|
coll.insert(h) # note that insert stores :_id in h and subsequent inserts are updates
|
|
end
|
|
|
|
def benchmark(iterations)
|
|
btms = Benchmark.measure do
|
|
(0...iterations).each do
|
|
yield
|
|
end
|
|
end
|
|
utime = btms.utime
|
|
p ({'ops' => (iterations.to_f / utime.to_f).round(1)})
|
|
end
|
|
|
|
def ruby_prof(iterations)
|
|
RubyProf.start
|
|
puts Benchmark.measure {
|
|
iterations.times { yield }
|
|
}
|
|
result = RubyProf.stop
|
|
|
|
# Print a flat profile to text
|
|
printer = RubyProf::FlatPrinter.new(result)
|
|
printer.print(STDOUT)
|
|
|
|
# Print a graph profile to text
|
|
printer = RubyProf::GraphPrinter.new(result)
|
|
printer.print(STDOUT, {})
|
|
end
|
|
|
|
def perftools(iterations)
|
|
profile_file_name = '/tmp/profile_array.perftools'
|
|
PerfTools::CpuProfiler.start(profile_file_name) do
|
|
iterations.times { yield }
|
|
end
|
|
cmd = "pprof.rb --ignore=IO --text \"#{profile_file_name}\""
|
|
system(cmd)
|
|
end
|
|
|
|
conn = Mongo::Connection.new
|
|
|
|
db = conn['benchmark']
|
|
coll = db['profile']
|
|
|
|
coll.remove
|
|
#puts "coll.count: #{coll.count}"
|
|
|
|
base = 2
|
|
power = 6
|
|
n, doc = array_size_fixnum(base, power)
|
|
p ({'generator' => 'array_size_fixnum', 'operation' => 'insert', 'base' => base, 'power' => power})
|
|
|
|
benchmark(10000) { insert(coll, doc)} # valgrind --tool=callgrind ruby bench/profile_array.rb; callgrind_annotate ...
|
|
#ruby_prof(1000) { insert(coll, doc) }
|
|
#perftools(10000) { insert(coll, doc) }
|
|
|
|
#puts "coll.count: #{coll.count}"
|
|
coll.remove
|
|
|