50 lines
1.2 KiB
Ruby
Executable File
50 lines
1.2 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
|
#
|
|
# Note: Ruby 1.9 is faster than 1.8, as expected.
|
|
|
|
$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
|
|
require 'mongo'
|
|
|
|
include XGen::Mongo::Driver
|
|
|
|
N = 30_000
|
|
|
|
def report(str, t0, t1, n)
|
|
dt = t1.to_f - t0.to_f
|
|
printf("%16s: %03.8f\n", str, dt)
|
|
end
|
|
|
|
def benchmark(str, n, db, after_proc=nil)
|
|
coll = db.collection('benchmark')
|
|
t0 = Time.new
|
|
n.times { |i| yield coll, i }
|
|
after_proc.call if after_proc
|
|
report(str, t0, Time.new, n)
|
|
end
|
|
|
|
host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
|
|
port = ENV['MONGO_RUBY_DRIVER_PORT'] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT
|
|
|
|
db = Mongo.new(host, port).db('ruby-benchmark')
|
|
db.drop_collection('benchmark')
|
|
coll = db.collection('benchmark')
|
|
|
|
coll.create_index('foo', 'i')
|
|
|
|
# Call to db.error forces inserts to finish
|
|
benchmark('insert', N, db, Proc.new{db.error}) { |coll, i|
|
|
coll.insert('i' => i)
|
|
}
|
|
benchmark('find_first', N, db) { |coll, i|
|
|
coll.find_first
|
|
}
|
|
benchmark('find', N, db) { |coll, i|
|
|
coll.find('i' => 3).each { }
|
|
coll.find('i' => 234).each { }
|
|
coll.find('i' => 9876).each { }
|
|
}
|
|
benchmark('find gt/lt', N, db) { |coll, i|
|
|
h = {'i' => {'$gt' => 200, '$lt' => 200}}
|
|
coll.find(h).each {}
|
|
}
|