dynamini 1.0.2 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 53c9e5cbb49d02855c69193b07ef3382bf94f632
-  data.tar.gz: 3788e242f66c6bd679e84dc2801e4334180d2f3f
+  metadata.gz: b6793462d0bd5515eddf8c4108022460d76c55db
+  data.tar.gz: 33867af1d14657a57e66bc526b8e726eac5001b8
 SHA512:
-  metadata.gz: d5e7b8a5c86e942f43abea007fe2f8d414c705180469407a6ed2bed4f6dfc9345921e763c47074e27ab1f1e181cfdb0ff8973135efd611702f9f6a43549fbe67
-  data.tar.gz: 84f1fb83aa4db6dd6f3ffa9e34f89b74938dfd9ad218bfceec5a49e4776174cbf207e67948c2f99a21d3777a0c3e2d1e4debec763c81ad1bc72d138d8cbf3e86
+  metadata.gz: 7f3d2a791f4249a428f942a8934af442f0612ea6b520057f0bb2022d81a3c205df93a05838c6840397333153ba2be87f1e1a27ce17007d3afe4be960d1d86b61
+  data.tar.gz: 0ac3f2a81f739b53cce9ca24e4a788827dfbfb0b19d2693a13ba6485a1321127104d9aa07da7ef480457ae331666d9e713e1aef3f45d1ee5af2d57cae0caaa72

data/lib/dynamini.rb

@@ -2,6 +2,7 @@ module Dynamini
   require 'active_model'
   require 'dynamini/base'
   require 'dynamini/configuration'
+  require 'dynamini/test_client'
 
   class << self
     attr_writer :configuration
@@ -14,12 +15,4 @@ module Dynamini
   def self.configure
     yield(configuration)
   end
-end
-
-
-
-# Dynamini.configure do |config|
-#   config.aws_region = 'eu-west-1'
-#   config.access_key_id =
-#   config.secret_access_key =
-# end
+end

data/lib/dynamini/base.rb

@@ -6,7 +6,7 @@ module Dynamini
     BATCH_SIZE = 25
 
     class << self
-      attr_writer :hash_key, :table_name, :batch_write_queue
+      attr_writer :hash_key, :table_name, :batch_write_queue, :in_memory
 
       def table_name
         @table_name || name.demodulize.downcase.pluralize
@@ -16,15 +16,23 @@ module Dynamini
         @hash_key || :id
       end
 
+      def in_memory
+        @in_memory || false
+      end
+
       def batch_write_queue
         @batch_write_queue ||= []
       end
 
       def client
-        @client ||= Aws::DynamoDB::Client.new(
-            region: Dynamini.configuration.region,
-            access_key_id: Dynamini.configuration.access_key_id,
-            secret_access_key: Dynamini.configuration.secret_access_key)
+        if in_memory
+          @client ||= Dynamini::TestClient.new(hash_key)
+        else
+          @client ||= Aws::DynamoDB::Client.new(
+              region: Dynamini.configuration.region,
+              access_key_id: Dynamini.configuration.access_key_id,
+              secret_access_key: Dynamini.configuration.secret_access_key)
+        end
       end
 
       def create(attributes, options={})
@@ -54,10 +62,14 @@ module Dynamini
 
       def batch_find(ids = [])
         return [] if ids.length < 1
+        objects = []
         raise StandardError, 'Batch find is limited to 100 items' if ids.length > 100
         key_structure = ids.map { |i| {hash_key => i} }
         response = self.dynamo_batch_get(key_structure)
-        response.responses[table_name]
+        response.responses[table_name].each do |item|
+          objects << self.new(item.symbolize_keys, false)
+        end
+        objects
       end
 
       def enqueue_for_save(attributes, options = {})
@@ -158,11 +170,11 @@ module Dynamini
     end
 
     def save_to_dynamo
-      Base.client.update_item(table_name: self.class.table_name, key: key, attribute_updates: attribute_updates)
+      self.class.client.update_item(table_name: self.class.table_name, key: key, attribute_updates: attribute_updates)
     end
 
     def touch_to_dynamo
-      Base.client.update_item(table_name: self.class.table_name, key: key, attribute_updates: {updated_at: {value: updated_at, action: 'PUT'}})
+      self.class.client.update_item(table_name: self.class.table_name, key: key, attribute_updates: {updated_at: {value: updated_at, action: 'PUT'}})
     end
 
     def self.dynamo_batch_get(key_structure)

data/lib/dynamini/test_client.rb

@@ -0,0 +1,64 @@
+module Dynamini
+  require 'ostruct'
+
+  class TestClient
+
+    attr_reader :hash_key
+
+    def initialize(hash_key)
+      @data = {}
+      @hash_key = hash_key
+    end
+
+    def update_item(args = {})
+      table = args[:table_name]
+      @data[table] ||= {}
+      @data[table][args[:key][hash_key]] = flatten_attribute_updates(args[:attribute_updates])
+      @data[table][args[:key][hash_key]][hash_key] = args[:key][hash_key]
+    end
+
+    def get_item(args = {})
+      attributes_hash = @data[args[:table_name]][args[:key][hash_key]]
+      item = attributes_hash.nil? ? nil : attributes_hash
+      OpenStruct.new(item: item)
+
+    end
+
+    def batch_get_item(args = {})
+      responses = {}
+
+      args[:request_items].each do |k, v|
+        responses[k] = []
+        v[:keys].each do |key_hash|
+          item = @data[k][key_hash.values.first]
+          responses[k] << item
+        end
+      end
+
+      OpenStruct.new(responses: responses)
+    end
+
+    def batch_write_item(request_options)
+      request_options[:request_items].each do |k, v|
+        @data[k] ||= {}
+        v.each do |request_hash|
+          item = request_hash[:put_request][:item]
+          key = item[hash_key]
+          @data[k][key] = item
+        end
+      end
+    end
+
+    private
+
+    def flatten_attribute_updates(attribute_updates)
+      attribute_hash = {}
+
+      attribute_updates.each do |k, v|
+        attribute_hash[k] = v[:value]
+      end
+      attribute_hash
+    end
+
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dynamini
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.1.0
 platform: ruby
 authors:
 - Greg Ward
@@ -55,6 +55,20 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: '3'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Lightweight DynamoDB interface designed as a drop-in replacement for
   ActiveRecord.
 email: dev@retailcommon.com
@@ -65,6 +79,7 @@ files:
 - lib/dynamini.rb
 - lib/dynamini/base.rb
 - lib/dynamini/configuration.rb
+- lib/dynamini/test_client.rb
 homepage: https://github.com/47colborne/dynamini
 licenses:
 - MIT