require 'riak'

class RiakCache

  def initialize
    @client = Riak::Client.new(:pb_port => 8081, :protocol => :pbc)
    @bucket = @client.bucket("riakcache")
  end

  def read(key)
    @object = @bucket.get(key)
    @object.data
  end

  def write(key, value)
    @object = @bucket.get_or_new(key)
    @object.content_type = 'application/x-ruby-marshal'
    @object.data = value
    @object.store
  end

end