$:.unshift('~/rufus/rufus-tokyo/lib')
 
require 'benchmark'
require 'rubygems'
require 'faker'
 
#
# the data
#
 
N = 1000
 
DATA = (0..N - 1).collect { |i|
  {
    'name' => Faker::Name.name,
    'sex' => (i % 2) ? 'male' : 'female',
    'birthday' => DateTime.new(1972, 10, 14),
    'divisions' => (i % 2) ? 'brd' : 'dev'
  }
}
 
DATA1 = DATA.collect { |e|
  h = e.dup
  h['birthday'] = h['birthday'].to_s
  h
}
  # Tokyo Cabinet tables only do strings
 
 
#
# AR ==========================================================================
#
 
#require_gem 'activerecord'
gem 'activerecord'; require 'active_record'
 
class TheMigration < ActiveRecord::Migration
 
  def self.up
    create_table :people do |t|
      t.column :name, :string
      t.column :sex, :string
      t.column :birthday, :timestamp
      t.column :divisions, :string
    end
    add_index :people, :name
    add_index :people, :sex
    add_index :people, :birthday
    add_index :people, :divisions
  end
 
  def self.down
    drop_table :people
  end
end
 
class Person < ActiveRecord::Base; end
 
ActiveRecord::Migration.verbose = false
 
 
def do_ar_benchmark
 
  begin
    TheMigration.down
  rescue Exception => e
  end
  TheMigration.up
 
  2.times { puts }
  puts "AR (#{ActiveRecord::Base.connection.adapter_name})"
 
  Benchmark.benchmark(' ' * 20 + Benchmark::Tms::CAPTION, 20) do |b|
 
    b.report('inserting data') do
      DATA.each { |h| Person.create(h) }
    end
    b.report('finding all') do
      Person.find(:all)
    end
    b.report('finding last') do
      Person.find(DATA.size)
    end
    b.report('find Alphonse') do
      Person.find_by_name('Alphonse Armalite')
    end
  end
end
 
 
## mysql
 
ActiveRecord::Base.establish_connection(
  :adapter => 'mysql',
  :database => 'test',
  :encoding => 'utf8',
  :pool => 30) # 2.2 ...
 
do_ar_benchmark
 
 
## sqlite3
 
FileUtils.rm_f('sqlite_test.db')
 
ActiveRecord::Base.establish_connection(
  :adapter => 'sqlite3',
  :database => 'sqlite_test.db')
 
do_ar_benchmark
 
puts
puts 'db weight : ' + `ls -lh ./sqlite_test.db | awk '{ print $5 }'`
 
 
#
# Tokyo Cabinet table =========================================================
#
 
require 'rufus/tokyo/cabinet/table'
 
FileUtils.rm_f('test.tdb')
 
table = Rufus::Tokyo::Table.new('test.tdb', :create, :write)
table.clear
 
2.times { puts }
puts 'TC table'
 
Benchmark.benchmark(' ' * 20 + Benchmark::Tms::CAPTION, 20) do |b|
 
  b.report('inserting data') do
    DATA1.each_with_index { |e, i| table[i.to_s] = e }
  end
  b.report('finding all') do
    table.query { |q| }
  end
  b.report('find last') do
    table[DATA.size.to_s]
  end
  b.report('find Alphonse') do
    table.query { |q| q.add('name', :equals, 'Alphonse Armalite') }
  end
end
 
puts
puts 'db weight : ' + `ls -lh ./test.tdb | awk '{ print $5 }'`
puts