aws-sdk-rails 3.9.1 → 3.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1491284a363942f9a0d4fe8712449caf2989869377fdb283632bec9c11b71b0b
-  data.tar.gz: c1f65e37b61266b80a20c06c21a39f84f7283ade2b9a590e5b80d3963ada71c7
+  metadata.gz: 34ec24104d9eb230956480f01ab20a9eefc670510727104c4b7ee930f31b281c
+  data.tar.gz: 8a60649a724d96aac5c927ad8090b3d942d860f3d6b755a3815cdd65777d46fa
 SHA512:
-  metadata.gz: 456ead7237daf5d83f55f4858f11d40a3d213a3bd8f8ded27c56be10e94e7aba6708a2adad8739b324991f28d22b97e32b8272d073f8f0bc082d390abfff8267
-  data.tar.gz: 97da8e92ec42c8404af945049feae80044aa2d6d1fed276eaf40bd50baa2b07826cfd201ed7eda00a92a9d1636749aaec0cc1adbc159a300c1d5b9690c86fc96
+  metadata.gz: 268c6e207ee1736ba3dad6450c7ccf0461ae0bd5dc55eb1bbae755a5ac5853f3edbef91fb140447c1a62327ca73a91bf534e39f908e51a8e771eab859a9fa389
+  data.tar.gz: 3033033fa8121807e2306c57368e29095a746741df12eec4f61b0daa7e388cafb1bf93eb35be67664205c3d0040cdb26151d729bb36b11fed2143547b2205780

data/VERSION

@@ -1 +1 @@
-3.9.1
+3.10.0

data/lib/active_job/queue_adapters/amazon_sqs_adapter/params.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+module ActiveJob
+  module QueueAdapters
+    class AmazonSqsAdapter
+      # == build request parameter of Aws::SQS::Client
+      class Params
+        class << self
+          def assured_delay_seconds(timestamp)
+            delay = (timestamp - Time.now.to_f).floor
+            delay = 0 if delay.negative?
+            raise ArgumentError, 'Unable to queue a job with a delay great than 15 minutes' if delay > 15.minutes
+
+            delay
+          end
+        end
+
+        def initialize(job, body)
+          @job = job
+          @body = body || job.serialize
+        end
+
+        def queue_url
+          @queue_url ||= Aws::Rails::SqsActiveJob.config.queue_url_for(@job.queue_name)
+        end
+
+        def entry
+          if Aws::Rails::SqsActiveJob.fifo?(queue_url)
+            default_entry.merge(options_for_fifo)
+          else
+            default_entry
+          end
+        end
+
+        private
+
+        def default_entry
+          {
+            message_body: Aws::Json.dump(@body),
+            message_attributes: message_attributes
+          }
+        end
+
+        def message_attributes
+          {
+            'aws_sqs_active_job_class' => {
+              string_value: @job.class.to_s,
+              data_type: 'String'
+            },
+            'aws_sqs_active_job_version' => {
+              string_value: Aws::Rails::VERSION,
+              data_type: 'String'
+            }
+          }
+        end
+
+        def options_for_fifo
+          options = {}
+          options[:message_deduplication_id] =
+            Digest::SHA256.hexdigest(Aws::Json.dump(deduplication_body))
+
+          message_group_id = @job.message_group_id if @job.respond_to?(:message_group_id)
+          message_group_id ||= Aws::Rails::SqsActiveJob.config.message_group_id
+
+          options[:message_group_id] = message_group_id
+          options
+        end
+
+        def deduplication_body
+          ex_dedup_keys = @job.excluded_deduplication_keys if @job.respond_to?(:excluded_deduplication_keys)
+          ex_dedup_keys ||= Aws::Rails::SqsActiveJob.config.excluded_deduplication_keys
+
+          @body.except(*ex_dedup_keys)
+        end
+      end
+    end
+  end
+end

data/lib/active_job/queue_adapters/amazon_sqs_adapter.rb

@@ -10,55 +10,44 @@ module ActiveJob
       end
 
       def enqueue_at(job, timestamp)
-        delay = (timestamp - Time.now.to_f).floor
-
-        delay = 0 if delay.negative?
-        raise ArgumentError, 'Unable to queue a job with a delay great than 15 minutes' if delay > 15.minutes
-
+        delay = Params.assured_delay_seconds(timestamp)
         _enqueue(job, nil, delay_seconds: delay)
       end
 
+      def enqueue_all(jobs)
+        enqueued_count = 0
+        jobs.group_by(&:queue_name).each do |queue_name, same_queue_jobs|
+          queue_url = Aws::Rails::SqsActiveJob.config.queue_url_for(queue_name)
+          base_send_message_opts = { queue_url: queue_url }
+
+          same_queue_jobs.each_slice(10) do |chunk|
+            entries = chunk.map do |job|
+              entry = Params.new(job, nil).entry
+              entry[:id] = job.job_id
+              entry[:delay_seconds] = Params.assured_delay_seconds(job.scheduled_at) if job.scheduled_at
+              entry
+            end
+
+            send_message_opts = base_send_message_opts.deep_dup
+            send_message_opts[:entries] = entries
+
+            send_message_batch_result = Aws::Rails::SqsActiveJob.config.client.send_message_batch(send_message_opts)
+            enqueued_count += send_message_batch_result.successful.count
+          end
+        end
+        enqueued_count
+      end
+
       private
 
       def _enqueue(job, body = nil, send_message_opts = {})
         body ||= job.serialize
-        queue_url = Aws::Rails::SqsActiveJob.config.queue_url_for(job.queue_name)
-        send_message_opts[:queue_url] = queue_url
-        send_message_opts[:message_body] = Aws::Json.dump(body)
-        send_message_opts[:message_attributes] = message_attributes(job)
-
-        if Aws::Rails::SqsActiveJob.fifo?(queue_url)
-          send_message_opts[:message_deduplication_id] =
-            Digest::SHA256.hexdigest(Aws::Json.dump(deduplication_body(job, body)))
-
-          message_group_id = job.message_group_id if job.respond_to?(:message_group_id)
-          message_group_id ||= Aws::Rails::SqsActiveJob.config.message_group_id
-
-          send_message_opts[:message_group_id] = message_group_id
-        end
+        params = Params.new(job, body)
+        send_message_opts = send_message_opts.merge(params.entry)
+        send_message_opts[:queue_url] = params.queue_url
 
         Aws::Rails::SqsActiveJob.config.client.send_message(send_message_opts)
       end
-
-      def message_attributes(job)
-        {
-          'aws_sqs_active_job_class' => {
-            string_value: job.class.to_s,
-            data_type: 'String'
-          },
-          'aws_sqs_active_job_version' => {
-            string_value: Aws::Rails::VERSION,
-            data_type: 'String'
-          }
-        }
-      end
-
-      def deduplication_body(job, body)
-        ex_dedup_keys = job.excluded_deduplication_keys if job.respond_to?(:excluded_deduplication_keys)
-        ex_dedup_keys ||= Aws::Rails::SqsActiveJob.config.excluded_deduplication_keys
-
-        body.except(*ex_dedup_keys)
-      end
     end
 
     # create an alias to allow `:amazon` to be used as the adapter name

data/lib/aws/rails/sqs_active_job/configuration.rb

@@ -126,7 +126,7 @@ module Aws
         def to_h
           h = {}
           instance_variables.each do |v|
-            v_sym = v.to_s.gsub('@', '').to_sym
+            v_sym = v.to_s.delete('@').to_sym
             val = instance_variable_get(v)
             h[v_sym] = val
           end

data/lib/aws-sdk-rails.rb

@@ -13,6 +13,7 @@ require_relative 'aws/rails/middleware/ebs_sqs_active_job_middleware'
 
 require_relative 'action_dispatch/session/dynamodb_store'
 require_relative 'active_job/queue_adapters/amazon_sqs_adapter'
+require_relative 'active_job/queue_adapters/amazon_sqs_adapter/params'
 require_relative 'active_job/queue_adapters/amazon_sqs_async_adapter'
 
 require_relative 'generators/aws_record/base'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-rails
 version: !ruby/object:Gem::Version
-  version: 3.9.1
+  version: 3.10.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-12-19 00:00:00.000000000 Z
+date: 2024-01-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-record
@@ -152,6 +152,7 @@ files:
 - bin/aws_sqs_active_job
 - lib/action_dispatch/session/dynamodb_store.rb
 - lib/active_job/queue_adapters/amazon_sqs_adapter.rb
+- lib/active_job/queue_adapters/amazon_sqs_adapter/params.rb
 - lib/active_job/queue_adapters/amazon_sqs_async_adapter.rb
 - lib/aws-sdk-rails.rb
 - lib/aws/rails/middleware/ebs_sqs_active_job_middleware.rb