lame ruby kv store

rubykv-data.rb

require 'msgpack'
module RubyKV
  class Data

    attr_reader :data_dir, :name

    def initialize(name, data_dir)
      @name = name
      @data_dir = data_dir
      @store = {}
    end

    def get(key)
      @store[key]
    end

    def set(key, value)
      if @store.has_key?(key) 
        puts "Overwritting existing key: #{key}"
      end
      @store[key] = value
    end

    def delete(key)
      @store.delete key
    end

    def keys
      @store.keys
    end

    def save
      begin
        File.open("#{@data_dir}/#{@name}.mpk", 'w'){|f| f.write(@store.to_msgpack) }
        puts "Database saved!"
      rescue Exception => e
        puts "Error saving database: #{e}"
      end
    end

    def load
      begin
        data = File.open("#{@data_dir}/#{@name}.mpk", 'r')
        @store = MessagePack.unpack data.read
        data.close
        puts "Database loaded"
      rescue Exception => e
        puts "Error loading database: #{e}"
      end
    end
  end
end


# @foo = RubyKV::Data.new "mystore", "/tmp/"
# 1000.times do |n|
#   @foo.set "key_#{n}", "data_#{n}"
# end
# @foo.save

# @foo = RubyKV::Data.new "mystore", "/tmp/"
# @foo
# @foo.load
# @foo