Key-Value Store for Codebrawl

README.md

A restful KV store

It relies on Sinatra to make everything pretty. Persistence is achieved with Marshal.

Methods

GET

curl "http://kvserver.dev/key1"
#=> value1

SET

curl -d "value3" "http://kvserver.dev/key3"
#=> value3

DELETE

curl -X delete "http://kvserver.dev/key3"
#=> value3

KEYS

curl "http://kvserver.dev"
#=> key1
    key2
    ...

store.rb

require 'sinatra/base'

class KVStore < Sinatra::Base
  set :file, "store.msh"
  set :logging, true

  configure do
    Store = Marshal.load(File.read settings.file) rescue {}
  end
  
  helpers do
    def save_to_disk
      open(settings.file, "w") do |f|
        f.write Marshal.dump(Store)
      end
    end
  end

  before('/*') do
    @key = params[:splat].first
  end

  after '/*' do
    # Write to disk only on actual changes
    save_to_disk if env["REQUEST_METHOD"] =~ /post|delete/i and not body.empty?
  end

  get('/')     { body Store.keys.join("\n") }              # KEYS
  get('/*')    { body Store[@key] }                        # GET
  post('/*')   { body Store[@key] = params.flatten.first } # SET
  delete('/*') { body Store.delete(@key) }                 # DELETE
end

KVStore.run! if __FILE__ == $0

test.rb

require './store'
require 'rack/test'

describe KVStore do
  include Rack::Test::Methods
  
  def app; KVStore; end
  
  let(:pairs){ {"test/key" => "test/value", "test/key2" => "test/value2"} }
  
  before(:each){ KVStore::Store.clear; KVStore::Store.merge! pairs }
  
  it 'should get a value' do
    pairs.each_pair do |k, v|
      get("/#{k}"){ |res| res.body.should == v }
    end
  end
  
  it 'should set a value' do
    post "/new/test", "5"
    last_response.body.should == "5"
  end
  
  it 'should remove a key' do
    pairs.each_pair do |k, v|
      delete("/#{k}"){ |res| res.body.should == v }
    end
  end
  
  it 'should get all the keys' do
    get("/"){ |res| res.body.should == pairs.keys.join("\n") }
  end
end