mongo-ruby-driver/examples/benchmarks.rb
2008-12-04 17:08:57 -07:00

37 lines
824 B
Ruby

require "rubygems"
require "benchwarmer"
$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
require 'mongo'
include XGen::Mongo::Driver
host = ARGV[0] || 'localhost'
port = ARGV[1] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT
puts "Connecting to #{host}:#{port}"
db = Mongo.new(host, port).db('ruby-mongo-examples-complex')
coll = db.collection('test')
coll.clear
OBJS_COUNT = 100
TEST_COUNT = 100
puts "Generating benchmark data"
msgs = %w{hola hello aloha ciao}
arr = OBJS_COUNT.times.map {|x| { :number => x, :rndm => (rand(5)+1), :msg => msgs[rand(4)] }}
puts "Running benchmark"
Benchmark.warmer(TEST_COUNT) do
report "single object inserts" do
coll.clear
arr.each {|x| coll.insert(x)}
end
report "multiple object insert" do
coll.clear
coll.insert(arr)
end
end
coll.clear