dynamini 1.4.7 → 1.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a5266d3ff025b2e2a6ff3a943121c121ade2c8b1
-  data.tar.gz: e49ab48e044498293a291a0f6cc9761825211459
+  metadata.gz: 661e45e76eaaaa1e48e12e3a8dfdd79013f1092d
+  data.tar.gz: eabe1c032f334bbb0a6b5249f525a2b18436d5b6
 SHA512:
-  metadata.gz: c091a21c83e7ee291ee967401691cb4dc642805ad053a6b783a4273548ec24c3e40441575179c8b6461e8208fac09b7add4abf79de4d7227a0c57858415db277
-  data.tar.gz: 481216095b8f4fe682be21196070178e2b729a24185f68b7a9a2f336cfd1e1c21148c3998371595d830cd8a2b82721a6b29ca3d28a02ba502786cf19acf650f1
+  metadata.gz: b773405d0e13858b136a44a8f19135a6b00a5dbf9823d7569cfcc0f034a75287940cc59ea087ce4de2e698f981ba401afd130be3548ed66265d927cb42150ab3
+  data.tar.gz: c9e5b0ceeba1269af53ddd7e596a1a0c5b29684b9e85b9a6d92f1cdd14cbcfd5e8bbad2d2b75c99887174a1cf3a7a0910fbdb1e167293d74b59326368a7f732f

data/lib/dynamini/base.rb

@@ -182,6 +182,19 @@ module Dynamini
       end
     end
 
+    def increment!(attribute, amount=1)
+      if amount.is_a?(Integer) || amount.is_a?(Float)
+        current_value = self.send(attribute)
+        if current_value.nil? || current_value.is_a?(Integer) || current_value.is_a?(Float)
+          increment_to_dynamo(attribute, amount)
+        else
+          raise StandardError, 'You cannot increment! a non-numeric non-nil value. If your current value is a numeric string, use :handle to autocast it as a number.'
+        end
+      else
+        raise StandardError, 'You cannot increment an attribute by a non-numeric value.'
+      end
+    end
+
     def delete
       delete_from_dynamo
       self
@@ -236,6 +249,10 @@ module Dynamini
       self.class.client.delete_item(table_name: self.class.table_name, key: key)
     end
 
+    def increment_to_dynamo(attribute, amount)
+      self.class.client.update_item(table_name: self.class.table_name, key: key, attribute_updates: {attribute => {value: amount, action: 'ADD'}, updated_at: {value: attributes[:updated_at], action: 'PUT'}})
+    end
+
     def self.dynamo_batch_get(key_structure)
       client.batch_get_item(request_items: {table_name => {keys: key_structure}})
     end

data/lib/dynamini/test_client.rb

@@ -12,7 +12,7 @@ module Dynamini
 
     def update_item(args = {})
       table = args[:table_name]
-      updates = flatten_attribute_updates(args[:attribute_updates]).merge({hash_key => args[:key][hash_key]})
+      updates = flatten_attribute_updates(args).merge({hash_key => args[:key][hash_key]})
       @data[table] ||= {}
       if @data[table][args[:key][hash_key]].present?
         @data[table][args[:key][hash_key]].merge!(updates)
@@ -24,7 +24,7 @@ module Dynamini
     def get_item(args = {})
       table = args[:table_name]
       @data[table] ||= {}
-      attributes_hash = @data[table ][args[:key][hash_key]]
+      attributes_hash = @data[table][args[:key][hash_key]]
       item = attributes_hash.nil? ? nil : attributes_hash
       OpenStruct.new(item: item)
     end
@@ -64,11 +64,15 @@ module Dynamini
 
     private
 
-    def flatten_attribute_updates(attribute_updates)
+    def flatten_attribute_updates(args = {})
       attribute_hash = {}
 
-      attribute_updates.each do |k, v|
-        attribute_hash[k] = v[:value]
+      args[:attribute_updates].each do |k, v|
+        if v[:action] == 'ADD'
+          attribute_hash[k] = v[:value] + @data[args[:table_name]][args[:key][hash_key]][k].to_f
+        else
+          attribute_hash[k] = v[:value]
+        end
       end
       attribute_hash
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dynamini
 version: !ruby/object:Gem::Version
-  version: 1.4.7
+  version: 1.5.0
 platform: ruby
 authors:
 - Greg Ward