benchmark-memcached.rb

require 'rubygems'
require 'benchmark'

gem 'memcached', '=1.0.6'
require 'memcached'

TEST_DATA_NUM = 100
$memcached = Memcached::new('localhost:11211')

# ベンチマークを計測する
Benchmark.bm do |x|
  x.report('memcached - set') {
    a = ('a'..'z').to_a + ('A'..'Z').to_a + ('0'..'9').to_a
    TEST_DATA_NUM.times {|i|
      data = ''
      ((i + 1) * 10).times { data += a[rand(a.size)] }
      $memcached.set((i + 1).to_s, data)
    }
  }

  x.report('memcached - get_multi 1 key') {
    1.step(TEST_DATA_NUM, 1){|i|
      if $memcached.get(i.to_s) == nil
        raise 'Could not get data from memcached - memcached - 1'
      end
    }
  }

  x.report('memcached - get_multi 5 keys') {
    1.step(TEST_DATA_NUM, 5){|i|
      keys = []
      5.times {|ii| keys << (i + ii).to_s }
      if $memcached.get(keys) == nil
        raise 'Could not get data from memcached - memcached - 5'
      end
    }
  }

  x.report('memcached - get_multi 10 keys') {
    1.step(TEST_DATA_NUM, 10){|i|
      keys = []
      10.times {|ii| keys << (i + ii).to_s }
      if $memcached.get(keys) == nil
        raise 'Could not get data from memcached - memcached - 10'
      end
    }
  }

  x.report('memcached - get_multi 100 keys') {
    1.step(TEST_DATA_NUM, 100){|i|
      keys = []
      100.times {|ii| keys << (i + ii).to_s }
      if $memcached.get(keys) == nil
        raise 'Could not get data from memcached - memcached - 100'
      end
    }
  }
end