dynamoid 3.2.0 → 3.3.0

data/lib/dynamoid/persistence/import.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+require 'securerandom'
+
+module Dynamoid
+  module Persistence
+    class Import
+      def self.call(model_class, array_of_attributes)
+        new(model_class, array_of_attributes).call
+      end
+
+      def initialize(model_class, array_of_attributes)
+        @model_class = model_class
+        @array_of_attributes = array_of_attributes
+      end
+
+      def call
+        models = @array_of_attributes.map(&method(:build_model))
+
+        unless Dynamoid.config.backoff
+          import(models)
+        else
+          import_with_backoff(models)
+        end
+
+        models.each { |d| d.new_record = false }
+        models
+      end
+
+      private
+
+      def build_model(attributes)
+        attrs = attributes.symbolize_keys
+
+        if Dynamoid::Config.timestamps
+          time_now = DateTime.now.in_time_zone(Time.zone)
+          attrs[:created_at] ||= time_now
+          attrs[:updated_at] ||= time_now
+        end
+
+        @model_class.build(attrs).tap do |model|
+          model.hash_key = SecureRandom.uuid if model.hash_key.blank?
+        end
+      end
+
+      def import_with_backoff(models)
+        backoff = nil
+        table_name = @model_class.table_name
+        items = array_of_dumped_attributes(models)
+
+        Dynamoid.adapter.batch_write_item(table_name, items) do |has_unprocessed_items|
+          if has_unprocessed_items
+            backoff ||= Dynamoid.config.build_backoff
+            backoff.call
+          else
+            backoff = nil
+          end
+        end
+      end
+
+      def import(models)
+        Dynamoid.adapter.batch_write_item(@model_class.table_name, array_of_dumped_attributes(models))
+      end
+
+      def array_of_dumped_attributes(models)
+        models.map do |m|
+          Dumping.dump_attributes(m.attributes, @model_class.attributes)
+        end
+      end
+    end
+  end
+end

data/lib/dynamoid/persistence/save.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+module Dynamoid
+  module Persistence
+    class Save
+      def self.call(model)
+        new(model).call
+      end
+
+      def initialize(model)
+        @model = model
+      end
+
+      def call
+        @model.hash_key = SecureRandom.uuid if @model.hash_key.blank?
+
+        # Add an optimistic locking check if the lock_version column exists
+        if @model.class.attributes[:lock_version]
+          @model.lock_version = (@model.lock_version || 0) + 1
+        end
+
+        attributes_dumped = Dumping.dump_attributes(@model.attributes, @model.class.attributes)
+        Dynamoid.adapter.write(@model.class.table_name, attributes_dumped, conditions_for_write)
+
+        @model.new_record = false
+        true
+      rescue Dynamoid::Errors::ConditionalCheckFailedException => e
+        if @model.new_record?
+          raise Dynamoid::Errors::RecordNotUnique.new(e, @model)
+        else
+          raise Dynamoid::Errors::StaleObjectError.new(@model, 'persist')
+        end
+      end
+
+      private
+
+      # Should be called after incrementing `lock_version` attribute
+      def conditions_for_write
+        conditions = {}
+
+        # Add an 'exists' check to prevent overwriting existing records with new ones
+        if @model.new_record?
+          conditions[:unless_exists] = [@model.class.hash_key]
+          if @model.range_key
+            conditions[:unless_exists] << @model.range_key
+          end
+        end
+
+        # Add an optimistic locking check if the lock_version column exists
+        if @model.class.attributes[:lock_version]
+          # Uses the original lock_version value from Dirty API
+          # in case user changed 'lock_version' manually
+          if @model.changes[:lock_version][0]
+            conditions[:if] ||= {}
+            conditions[:if][:lock_version] = @model.changes[:lock_version][0]
+          end
+        end
+
+        conditions
+      end
+    end
+  end
+end

data/lib/dynamoid/persistence/update_fields.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module Dynamoid
+  module Persistence
+    class UpdateFields
+      def self.call(*args)
+        new(*args).call
+      end
+
+      def initialize(model_class, partition_key:, sort_key:, attributes:, conditions:)
+        @model_class = model_class
+        @partition_key = partition_key
+        @sort_key = sort_key
+        @attributes = attributes.symbolize_keys
+        @conditions = conditions
+      end
+
+      def call
+        if Dynamoid::Config.timestamps
+          @attributes[:updated_at] ||= DateTime.now.in_time_zone(Time.zone)
+        end
+
+        raw_attributes = update_item
+        @model_class.new(undump_attributes(raw_attributes))
+      rescue Dynamoid::Errors::ConditionalCheckFailedException
+      end
+
+      private
+
+      def update_item
+        Dynamoid.adapter.update_item(@model_class.table_name, @partition_key, options_to_update_item) do |t|
+          @attributes.each do |k, v|
+            value_casted = TypeCasting.cast_field(v, @model_class.attributes[k])
+            value_dumped = Dumping.dump_field(value_casted, @model_class.attributes[k])
+            t.set(k => value_dumped)
+          end
+        end
+      end
+
+      def undump_attributes(attributes)
+        Undumping.undump_attributes(attributes, @model_class.attributes)
+      end
+
+      def options_to_update_item
+        options = {}
+
+        if @model_class.range_key
+          value_casted = TypeCasting.cast_field(@sort_key, @model_class.attributes[@model_class.range_key])
+          value_dumped = Dumping.dump_field(value_casted, @model_class.attributes[@model_class.range_key])
+          options[:range_key] = value_dumped
+        end
+
+        conditions = @conditions.deep_dup
+        conditions[:if_exists] ||= {}
+        conditions[:if_exists][@model_class.hash_key] = @partition_key
+        options[:conditions] = conditions
+
+        options
+      end
+    end
+  end
+end

data/lib/dynamoid/persistence/upsert.rb

@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+
+module Dynamoid
+  module Persistence
+    class Upsert
+      def self.call(*args)
+        new(*args).call
+      end
+
+      def initialize(model_class, partition_key:, sort_key:, attributes:, conditions:)
+        @model_class = model_class
+        @partition_key = partition_key
+        @sort_key = sort_key
+        @attributes = attributes.symbolize_keys
+        @conditions = conditions
+      end
+
+      def call
+        if Dynamoid::Config.timestamps
+          @attributes[:updated_at] ||= DateTime.now.in_time_zone(Time.zone)
+        end
+
+        raw_attributes = update_item
+        @model_class.new(undump_attributes(raw_attributes))
+      rescue Dynamoid::Errors::ConditionalCheckFailedException
+      end
+
+      private
+
+      def update_item
+        Dynamoid.adapter.update_item(@model_class.table_name, @partition_key, options_to_update_item) do |t|
+          @attributes.each do |k, v|
+            value_casted = TypeCasting.cast_field(v, @model_class.attributes[k])
+            value_dumped = Dumping.dump_field(value_casted, @model_class.attributes[k])
+
+            t.set(k => value_dumped)
+          end
+        end
+      end
+
+      def options_to_update_item
+        options = {}
+
+        if @model_class.range_key
+          value_casted = TypeCasting.cast_field(@sort_key, @model_class.attributes[@model_class.range_key])
+          value_dumped = Dumping.dump_field(value_casted, @model_class.attributes[@model_class.range_key])
+          options[:range_key] = value_dumped
+        end
+
+        options[:conditions] = @conditions
+        options
+      end
+
+      def undump_attributes(raw_attributes)
+        Undumping.undump_attributes(raw_attributes, @model_class.attributes)
+      end
+    end
+  end
+end
+

data/lib/dynamoid/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Dynamoid
-  VERSION = '3.2.0'
+  VERSION = '3.3.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dynamoid
 version: !ruby/object:Gem::Version
-  version: 3.2.0
+  version: 3.3.0
 platform: ruby
 authors:
 - Josh Symonds
@@ -21,7 +21,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-05-15 00:00:00.000000000 Z
+date: 2019-08-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel
@@ -237,9 +237,11 @@ files:
 - gemfiles/rails_5_0.gemfile
 - gemfiles/rails_5_1.gemfile
 - gemfiles/rails_5_2.gemfile
+- gemfiles/rails_6_0.gemfile
 - lib/dynamoid.rb
 - lib/dynamoid/adapter.rb
 - lib/dynamoid/adapter_plugin/aws_sdk_v3.rb
+- lib/dynamoid/adapter_plugin/aws_sdk_v3/batch_get_item.rb
 - lib/dynamoid/adapter_plugin/aws_sdk_v3/create_table.rb
 - lib/dynamoid/adapter_plugin/aws_sdk_v3/item_updater.rb
 - lib/dynamoid/adapter_plugin/aws_sdk_v3/middleware/backoff.rb
@@ -278,8 +280,13 @@ files:
 - lib/dynamoid/finders.rb
 - lib/dynamoid/identity_map.rb
 - lib/dynamoid/indexes.rb
+- lib/dynamoid/loadable.rb
 - lib/dynamoid/middleware/identity_map.rb
 - lib/dynamoid/persistence.rb
+- lib/dynamoid/persistence/import.rb
+- lib/dynamoid/persistence/save.rb
+- lib/dynamoid/persistence/update_fields.rb
+- lib/dynamoid/persistence/upsert.rb
 - lib/dynamoid/primary_key_type_mapping.rb
 - lib/dynamoid/railtie.rb
 - lib/dynamoid/tasks.rb