benchmark-memcache-client.rb

require 'rubygems'
require 'benchmark'

gem 'memcache-client', '=1.8.5'
require 'memcache'

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


# ベンチマークを計測する
Benchmark.bm do |x|
  x.report('memcache-client - 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)] }
      if !$memcached.set((i + 1).to_s, data)
        raise 'Could not set data to memcached server.'
      end
    }
  }

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

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

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

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