mingle_keyvalue_store 0.1.0

data/lib/mingle_keyvalue_store.rb

@@ -0,0 +1,115 @@
+require "rubygems"
+require "aws"
+require "pstore"
+require "fileutils"
+require "tempfile"
+
+module Mingle
+  module KeyvalueStore
+    class PStoreBased
+      def initialize(path, namespace, key_column, value_column)
+        @namespace = namespace
+        @store_file = store_file(path)
+        @key_column = key_column
+        @value_column = value_column
+        @pstore = PStore.new(@store_file)
+      end
+
+      def [](store_key)
+        @pstore.transaction { @pstore[store_key] }
+      end
+
+      def []=(store_key, value)
+        @pstore.transaction do
+          @pstore["all_names"] ||= []
+          @pstore["all_names"] = (@pstore["all_names"] + [store_key]).uniq
+          @pstore[store_key] = JSON.parse(value.to_json)
+        end
+      end
+
+      def delete(store_key)
+        @pstore.transaction do
+          @pstore["all_names"].delete_if {|name| name == store_key}
+          @pstore.delete(store_key)
+        end
+        nil
+      end
+
+      def clear
+        FileUtils.rm_f(@store_file)
+        @pstore = PStore.new(@store_file)
+        nil
+      end
+
+      def names
+        @pstore.transaction { @pstore["all_names"] || [] }
+      end
+
+      def all_items
+        @pstore.transaction do
+          return [] unless @pstore["all_names"]
+          @pstore["all_names"].map do |name|
+            {
+              @key_column.to_s => name,
+              @value_column.to_s => @pstore[name].to_json
+            }
+          end
+        end
+
+      end
+
+      private
+
+      def store_file(path)
+        FileUtils.mkdir_p(path)
+
+        file = Tempfile.new("#{@namespace}_multitenancy_configs.pstore", path).path
+        FileUtils.mkdir_p(File.dirname(file))
+        file
+      end
+    end
+
+    class DynamodbBased
+      def initialize(table_name, key_column, value_column)
+        @key_column = key_column
+        @value_column = value_column
+        @table_name = table_name
+      end
+
+      def [](store_key)
+        if attribute = table_items[store_key].attributes[@value_column]
+          JSON.parse(attribute)
+        end
+      end
+
+      def []=(store_key, value)
+        table_items.create(@key_column => store_key, @value_column => value.to_json)
+      end
+
+      def clear
+        table_items.each(&:delete)
+      end
+
+      def names
+        table_items.map(&:hash_value)
+      end
+
+      def all_items
+        table_items.map { |item| item.attributes.to_hash }
+      end
+
+      def delete(key)
+        table_items.where(@key_column => key).first.delete
+      end
+
+      private
+
+      def table_items
+        table = AWS::DynamoDB.new.tables[@table_name]
+        table.hash_key = [@key_column, :string]
+        table.items
+      end
+
+    end
+  end
+end

metadata

@@ -0,0 +1,98 @@
+--- !ruby/object:Gem::Specification 
+name: mingle_keyvalue_store
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Ian
+- sdqali
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-10-16 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: aws-sdk
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 61
+        segments: 
+        - 1
+        - 11
+        - 3
+        version: 1.11.3
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+description: A key value store implentation that uses DynamoDB or Pstore underneath.
+email: 
+- reginaldthedog@gmail.com
+- sadiqalikm@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README.md
+- lib/mingle_keyvalue_store.rb
+homepage: https://github.com/ThoughtWorksStudios/mingle_keyvalue_store
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: A key value store implentation that uses DynamoDB or Pstore underneath.
+test_files: []
+