mongo-ruby-driver/bench/profile_array.rb

83 lines
1.8 KiB
Ruby
Raw Normal View History

#!/usr/bin/env ruby
$LOAD_PATH.unshift(File.expand_path("../../lib", __FILE__))
2012-04-16 21:28:48 +00:00
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]
2012-04-17 13:25:11 +00:00
#p (ARGV[0] && ARGV[0].to_sym || :c)
2012-04-16 21:28:48 +00:00
set_mongo_driver_mode($mode || :c)
require 'rubygems'
require 'mongo'
require 'benchmark'
require 'ruby-prof'
2012-04-16 21:28:48 +00:00
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
2012-04-16 21:28:48 +00:00
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
2012-04-16 21:28:48 +00:00
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}"
n, doc = array_size_fixnum(2, 6)
2012-04-16 21:28:48 +00:00
ruby_prof(1000) { insert(coll, doc) }
#perftools(10000) { insert(coll, doc) }
puts "coll.count: #{coll.count}"
coll.remove