Riak basic benchmark script

gistfile1.txt

#!/usr/bin/env ruby

require 'bundler'

Bundler.setup( :utilities )
Bundler.require( :utilities )

class Value
	include Ripple::Document
	property :name, String
	property :value, Integer
	timestamps!
end

client = Riak::Client.new
bucket = client.bucket('values')
bucket.props = { 'r' => 1, 'w' => 3, 'rw' => 1, 'dw' => 3 }

MAX = 100

count = 0
(1..MAX).each do |i|
	v = Value.find(i)
	if v
		v.destroy
		count += 1
	end
end
puts "Deleted #{count}"

sum = 0
(1..MAX).each do |i|
	v = Value.new
	v.key = i
	v.name = "Value#{i}"
	v.value = rand(9999)
	v.save!
	sum += v.value
end

puts "Wrote #{MAX} #{sum}"

running = true

trap("INT") {
	running = false
}

while running
	count = 0
	sum = 0
	(1..MAX).each do |i|
		
		v = Value.find(i)
		if v
			count += 1
			sum += v.value
		end
	end
	puts "#{Time.now.strftime('%s')}: #{count} #{sum}"
end

count = 0
(1..MAX).each do |i|
	v = Value.find(i)
	if v
		v.destroy
		count += 1
	end
end
puts "Deleted #{count}"