lambdakiq 0.0.1 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 94edc48f3f08f3e16711bb2ece50cc7cc5fcbda5bdbff4ca83de387b1c991fed
-  data.tar.gz: 3ad808f67b8b497b4fde72f0377bb24efc8cf2d812cd4e7786a379b418af4f93
+  metadata.gz: 5c2216c121ae2ba5e013c8852bc71d8e3e84342cd67053e4d9933a88562c543a
+  data.tar.gz: f1c48212d2f14307fbb7a09ac10748d108b0d705270df8e3a5b54047bdcee758
 SHA512:
-  metadata.gz: 126c569c9e85bf812dda786e2f14a1eba6206de6072f80d235d60077fec67a9a1bc2195055e1dbf84f58cdf1a632f6559fe81908d2906801d87bae430191d9e6
-  data.tar.gz: 93d33b2b486c0b92a905fd1c3b1c5c4872652e2e42d52c804df42e2ec0209e5fb3cc8744bec480796119c49479d0b43524e33326f01bafd91d8dc9c36992e94d
+  metadata.gz: 687c1792d5f9e15ce30db4226a0248b46a874e2c8b55e81327a871977ccbff375ec67eb3c8279a2ff565e7e1c9f05f5147f70a0bec1e6308e8a8fbc2863440f3
+  data.tar.gz: 0e88a280f225db1dfb8251b0622e99bf3d781dba6c817fe8f70b8345725613421578da31cc5bac37d8664fd0390431518a741f92d71e8da4087506c742719e10

data/Gemfile.lock

@@ -1,53 +1,98 @@
 PATH
   remote: .
   specs:
-    lambdakiq (1.0.0)
+    lambdakiq (1.0.1)
       activejob
       aws-sdk-sqs
+      concurrent-ruby
+      railties
 
 GEM
   remote: https://rubygems.org/
   specs:
-    activejob (6.0.3.4)
-      activesupport (= 6.0.3.4)
+    actionpack (6.1.1)
+      actionview (= 6.1.1)
+      activesupport (= 6.1.1)
+      rack (~> 2.0, >= 2.0.9)
+      rack-test (>= 0.6.3)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.0, >= 1.2.0)
+    actionview (6.1.1)
+      activesupport (= 6.1.1)
+      builder (~> 3.1)
+      erubi (~> 1.4)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.1, >= 1.2.0)
+    activejob (6.1.1)
+      activesupport (= 6.1.1)
       globalid (>= 0.3.6)
-    activesupport (6.0.3.4)
+    activesupport (6.1.1)
       concurrent-ruby (~> 1.0, >= 1.0.2)
-      i18n (>= 0.7, < 2)
-      minitest (~> 5.1)
-      tzinfo (~> 1.1)
-      zeitwerk (~> 2.2, >= 2.2.2)
+      i18n (>= 1.6, < 2)
+      minitest (>= 5.1)
+      tzinfo (~> 2.0)
+      zeitwerk (~> 2.3)
     aws-eventstream (1.1.0)
-    aws-partitions (1.399.0)
-    aws-sdk-core (3.109.3)
+    aws-partitions (1.422.0)
+    aws-sdk-core (3.111.2)
       aws-eventstream (~> 1, >= 1.0.2)
       aws-partitions (~> 1, >= 1.239.0)
       aws-sigv4 (~> 1.1)
       jmespath (~> 1.0)
-    aws-sdk-sqs (1.34.0)
+    aws-sdk-sqs (1.35.0)
       aws-sdk-core (~> 3, >= 3.109.0)
       aws-sigv4 (~> 1.1)
     aws-sigv4 (1.2.2)
       aws-eventstream (~> 1, >= 1.0.2)
+    builder (3.2.4)
     coderay (1.1.3)
-    concurrent-ruby (1.1.7)
+    concurrent-ruby (1.1.8)
+    crass (1.0.6)
+    erubi (1.10.0)
     globalid (0.4.2)
       activesupport (>= 4.2.0)
-    i18n (1.8.5)
+    i18n (1.8.7)
       concurrent-ruby (~> 1.0)
     jmespath (1.4.0)
+    loofah (2.9.0)
+      crass (~> 1.0.2)
+      nokogiri (>= 1.5.9)
+    macaddr (1.7.2)
+      systemu (~> 2.6.5)
     method_source (1.0.0)
+    mini_portile2 (2.5.0)
     minitest (5.14.2)
     minitest-focus (1.2.1)
       minitest (>= 4, < 6)
     mocha (1.11.2)
+    nokogiri (1.11.1)
+      mini_portile2 (~> 2.5.0)
+      racc (~> 1.4)
     pry (0.13.1)
       coderay (~> 1.1)
       method_source (~> 1.0)
+    racc (1.5.2)
+    rack (2.2.3)
+    rack-test (1.1.0)
+      rack (>= 1.0, < 3)
+    rails-dom-testing (2.0.3)
+      activesupport (>= 4.2.0)
+      nokogiri (>= 1.6)
+    rails-html-sanitizer (1.3.0)
+      loofah (~> 2.3)
+    railties (6.1.1)
+      actionpack (= 6.1.1)
+      activesupport (= 6.1.1)
+      method_source
+      rake (>= 0.8.7)
+      thor (~> 1.0)
     rake (13.0.1)
-    thread_safe (0.3.6)
-    tzinfo (1.2.8)
-      thread_safe (~> 0.1)
+    systemu (2.6.5)
+    thor (1.1.0)
+    tzinfo (2.0.4)
+      concurrent-ruby (~> 1.0)
+    uuid (2.3.9)
+      macaddr (~> 1.0)
     zeitwerk (2.4.2)
 
 PLATFORMS
@@ -61,6 +106,7 @@ DEPENDENCIES
   mocha
   pry
   rake
+  uuid
 
 BUNDLED WITH
    2.1.4

data/README.md

@@ -1,12 +1,248 @@
+![Test](https://github.com/customink/lambdakiq/workflows/Test/badge.svg)
+
+![Lambdakiq: ActiveJob on SQS & Lambda](images/Lambdakiq.png)
 
 # Lambdakiq
 
-TODO ...
+<a href="https://lamby.custominktech.com"><img src="https://user-images.githubusercontent.com/2381/59363668-89edeb80-8d03-11e9-9985-2ce14361b7e3.png" alt="Lamby: Simple Rails & AWS Lambda Integration using Rack." align="right" width="300" /></a>A drop-in replacement for [Sidekiq](https://github.com/mperham/sidekiq) when running Rails in AWS Lambda using the [Lamby](https://lamby.custominktech.com) gem.
+
+Lambdakiq allows you to leverage AWS' managed infrastructure to the fullest extent. Gone are the days of managing pods and long polling processes. Instead AWS delivers messages directly to your Rails' job functions and scales it up and down as needed. Observability is built in using AWS CloudWatch Metrics, Dashboards, and Alarms. Learn more about [Using AWS Lambda with Amazon SQS](https://docs.aws.amazon.com/lambda/latest/dg/with-sqs.html) or get started now.
+
+## Key Features
+
+* Distinct web & jobs Lambda functions.
+* AWS fully managed polling. Event-driven.
+* Maximum 12 retries. Per job configurable.
+* Mirror Sidekiq's retry [backoff](https://github.com/mperham/sidekiq/wiki/Error-Handling#automatic-job-retry) timing.
+* Last retry is at 11 hours 30 minutes.
+* Supports ActiveJob's wait/delay. Up to 15 minutes.
+* Dead messages are stored for up to 14 days.
+
+## Project Setup
+
+This gem assumes your Rails application is on AWS Lambda, ideally with our [Lamby](https://lamby.custominktech.com) gem. It could be using Lambda's traditional zip package type or the newer [container](https://dev.to/aws-heroes/lambda-containers-with-rails-a-perfect-match-4lgb) format. If Rails on Lambda is new to you, consider following our [quick start](https://lamby.custominktech.com/docs/quick_start) guide to get your first application up and running. From there, to use Lambdakiq, here are steps to setup your project
+
+
+### Bundle & Config
+
+Add the Lambdakiq gem to your `Gemfile`.
+
+```ruby
+gem 'lambdakiq'
+```
+
+Open `config/initializers/production.rb` and set Lambdakiq as your ActiveJob queue adapter.
+
+```ruby
+config.active_job.queue_adapter = :lambdakiq
+```
+
+Open `app/jobs/application_job.rb` and add our worker module. The queue name will be set by an environment using CloudFormation further down.
+
+```ruby
+class ApplicationJob < ActiveJob::Base
+  include Lambdakiq::Worker
+  queue_as ENV['JOBS_QUEUE_NAME']
+end
+```
+
+### SQS Resources
+
+Open up your project's SAM [`template.yaml`](https://lamby.custominktech.com/docs/anatomy#file-template-yaml) file and make the following additions and changes. First, we need to create your [SQS queues](https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-sqs-queues.html) under the `Resources` section.
+
+```yaml
+JobsQueue:
+  Type: AWS::SQS::Queue
+  Properties:
+    ReceiveMessageWaitTimeSeconds: 10
+    RedrivePolicy:
+      deadLetterTargetArn: !GetAtt JobsDLQueue.Arn
+      maxReceiveCount: 13
+    VisibilityTimeout: 301
+
+JobsDLQueue:
+  Type: AWS::SQS::Queue
+  Properties:
+    MessageRetentionPeriod: 1209600
+```
+
+In this example above we are also creating a queue to automatically handle our redrives and storage for any dead messages. We use [long polling](https://docs.aws.amazon.com/AWSSimpleQueueService/latest/SQSDeveloperGuide/sqs-short-and-long-polling.html#sqs-long-polling) to receive messages for lower costs. In most cases your message is consumed almost immediately. Sidekiq polling is around 10s too.
+
+The max receive count is 13 which means you get 12 retries. This is done so we can mimic Sidekiq's [automatic retry and backoff](https://github.com/mperham/sidekiq/wiki/Error-Handling#automatic-job-retry). The dead letter queue retains messages for the maximum of 14 days. This can be changed as needed. We also make no assumptions on how you want to handle dead jobs.
+
+### Queue Name Environment Variable
+
+We need to pass the newly created queue's name as an environment variable to your soon to be created jobs function. Since it is common for your Rails web and jobs functions to share these, we can leverage [SAM's Globals](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-specification-template-anatomy-globals.html) section.
+
+```yaml
+Globals:
+  Function:
+    Environment:
+      Variables:
+        RAILS_ENV: !Ref RailsEnv
+        JOBS_QUEUE_NAME: !GetAtt JobsQueue.QueueName
+```
+
+We can remove the `Environment` section from our web function and all functions in this stack will now use the globals. Here we are using an [intrinsic function](https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/intrinsic-function-reference-getatt.html) to pass the queue's name as the `JOBS_QUEUE_NAME` environment variable.
+
+### IAM Permissions
+
+Both functions will need capabilities to access the SQS jobs queue. We can add or extend the [SAM Policies](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-resource-function.html#sam-function-policies) section of our `RailsLambda` web function so it (and our soon to be created jobs function) have full capabilities to this new queue.
+
+```yaml
+Policies:
+  - Version: '2012-10-17'
+    Statement:
+      - Effect: Allow
+        Action:
+          - sqs:*
+        Resource:
+          - !Sub arn:aws:sqs:${AWS::Region}:${AWS::AccountId}:${JobsQueue.QueueName}
+```
+
+Now we can duplicate our `RailsLambda` resource YAML (except for the `Events` property) to a new `JobsLambda` one. This gives us a distinct Lambda function to process jobs whose events, memory, timeout, and more can be independently tuned. However, both the `web` and `jobs` functions will use the same ECR container image!
+
+```yaml
+JobsLambda:
+  Type: AWS::Serverless::Function
+  Metadata:
+    DockerContext: ./.lamby/RailsLambda
+    Dockerfile: Dockerfile
+    DockerTag: jobs
+  Properties:
+    Events:
+      SQSJobs:
+        Type: SQS
+        Properties:
+          Queue: !GetAtt JobsQueue.Arn
+          BatchSize: 1
+    MemorySize: 1792
+    PackageType: Image
+    Policies:
+      - Version: '2012-10-17'
+        Statement:
+          - Effect: Allow
+            Action:
+              - sqs:*
+            Resource:
+              - !Sub arn:aws:sqs:${AWS::Region}:${AWS::AccountId}:${JobsQueue.QueueName}
+    Timeout: 300
+```
+
+Here are some key aspects of our `JobsLambda` resource above:
+
+* The `Events` property uses the [SQS Type](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-property-function-sqs.html).
+* Our [BatchSize](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-property-function-sqs.html#sam-function-sqs-batchsize) is set to one so we can handle retrys more easily without worrying about idempotency in larger batches.
+* The `Metadata`'s Docker properties must be the same as our web function except for the `DockerTag`. This is needed for the image to be shared. This works around a known [SAM issue](https://github.com/aws/aws-sam-cli/issues/2466) vs using the `ImageConfig` property.
+* The jobs function `Timeout` must be lower than the `JobsQueue`'s `VisibilityTimeout` property. When the batch size is one, the queue's visibility is generally one second more.
+
+🎉 Deploy your application and have fun with ActiveJob on SQS & Lambda.
+
+## Configuration
+
+Most general Lambdakiq configuration options are exposed via the Rails standard configuration method.
+
+### Rails Configs
+
+```ruby
+config.lambdakiq
+```
+
+* `max_retries=` - Retries for all jobs. Default is the Lambdakiq maximum of `12`.
+* `metrics_namespace=` - The CloudWatch Embedded Metrics namespace. Default is `Lambdakiq`.
+* `metrics_logger=` - Set to the Rails logger which is STDOUT via Lamby/Lambda.
+
+### ActiveJob Configs
+
+You can also set configuration options on a per job basis using the `lambdakiq_options` method.
+
+```ruby
+class OrderProcessorJob < ApplicationJob
+  lambdakiq_options retry: 2
+end
+```
+
+* `retry` - Overrides the default Lambdakiq `max_retries` for this one job.
+
+## Concurrency & Limits
+
+AWS SQS is highly scalable with [few limits](https://docs.aws.amazon.com/AWSSimpleQueueService/latest/SQSDeveloperGuide/sqs-quotas.html). As your jobs in SQS increases so should your concurrent functions to process that work. However, as this article, ["Why isn't my Lambda function with an Amazon SQS event source scaling optimally?"](https://aws.amazon.com/premiumsupport/knowledge-center/lambda-sqs-scaling/) describes it is possible that errors will effect your concurrency.
+
+To help keep your queue and workers scalable, reduce the errors raised by your jobs. You an also reduce the retry count.
+
+## Observability with CloudWatch
+
+Get ready to gain way more insights into your ActiveJobs using AWS' [CloudWatch](https://aws.amazon.com/cloudwatch/) service. Every AWS service, including SQS & Lambda, publishes detailed [CloudWatch Metrics](https://docs.aws.amazon.com/AmazonCloudWatch/latest/monitoring/working_with_metrics.html). This gem leverages [CloudWatch Embedded Metrics](https://docs.aws.amazon.com/AmazonCloudWatch/latest/monitoring/CloudWatch_Embedded_Metric_Format.html) to add detailed ActiveJob metrics to that system. You can mix and match these data points to build your own [CloudWatch Dashboards](https://docs.aws.amazon.com/AmazonCloudWatch/latest/monitoring/CloudWatch_Dashboards.html). If needed, any combination can be used to trigger [CloudWatch Alarms](https://docs.aws.amazon.com/AmazonCloudWatch/latest/monitoring/AlarmThatSendsEmail.html). Much like Sumo Logic, you can search & query for data using [CloudWatch Logs Insights](https://docs.aws.amazon.com/AmazonCloudWatch/latest/logs/AnalyzingLogData.html).
+
+![CloudWatch Dashboard](https://user-images.githubusercontent.com/2381/106465990-be7a6200-6468-11eb-8461-93db0046cda5.png)
+
+Metrics are published under the `Lambdakiq` namespace. This is configurable using `config.lambdakiq.metrics_namespace` but should not be needed since all metrics are published using these three dimensions which allow you to easily segment metrics/dashboards to a specific application.
+
+### Metric Dimensions
+
+* `AppName` - This is the name of your Rails application. Ex: `MyApp`
+* `JobEvent` - Name of the ActiveSupport Notification. Ex: `*.active_job`.
+* `JobName` - The class name of the ActiveSupport job. Ex: `NotificationJob`
+
+### ActiveJob Event Names
+For reference, here are the `JobEvent` names published by ActiveSupport. A few of these are instrumented by Lambdakiq since we use custom retry logic like Sidekiq. These event/metrics are found in the Rails application CloudWatch logs because they publish/enqueue jobs.
+
+* `enqueue.active_job`
+* `enqueue_at.active_job`
+
+While these event/metrics can be found in the jobs function's log.
+
+* `perform_start.active_job`
+* `perform.active_job`
+* `enqueue_retry.active_job`
+* `retry_stopped.active_job`
+
+### Metric Properties
+
+These are the properties published with each metric. Remember, properties can not be used as metric data in charts but can be searched using [CloudWatch Logs Insights](https://docs.aws.amazon.com/AmazonCloudWatch/latest/logs/AnalyzingLogData.html).
+
+* `JobId` - ActiveJob Unique ID. Ex: `9f3b6977-6afc-4769-aed6-bab1ad9a0df5`
+* `QueueName` - SQS Queue Name. Ex: `myapp-JobsQueue-14F18LG6XFUW5.fifo`
+* `MessageId` - SQS Message ID. Ex: `5653246d-dc5e-4c95-9583-b6b83ec78602`
+* `ExceptionName` - Class name of error raised. Present in perform and retry events.
+* `EnqueuedAt` - When ActiveJob enqueued the message. Ex: `2021-01-14T01:43:38Z`
+* `Executions` - The number of current executions. Counts from `1` and up.
+* `JobArg#{n}` - Enumerated serialized arguments.
+
+### Metric Data
+
+And finally, here are the metrics which each dimension can chart using [CloudWatch Metrics & Dashboards](https://docs.aws.amazon.com/AmazonCloudWatch/latest/monitoring/CloudWatch_Dashboards.html).
+
+* `Duration` - Of the job event in milliseconds.
+* `Count` - Of the event.
+* `ExceptionCount` - Of the event. Useful with `ExceptionName`.
+
+### CloudWatch Dashboard Examples
+
+Please share how you are using CloudWatch to monitor and/or alert on your ActiveJobs with Lambdakiq!
+
+💬 https://github.com/customink/lambdakiq/discussions/3
+
+
+## Common Questions
+
+**Are Scheduled Jobs Supported?** - No. If you need a scheduled job please use the [SAM Schedule](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-property-function-schedule.html) event source which invokes your function with an [Eventbridege AWS::Events::Rule](https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-resource-events-rule.html).
+
+**Are FIFO Queues Supported?** - Yes. When you create your [AWS::SQS::Queue](https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-sqs-queues.html) resources you can set the [FifoQueue](https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-sqs-queues.html#aws-sqs-queue-fifoqueue) property to `true`. Remember that both your jobs queue and the redrive queue must be the same. When using FIFO we:
+
+* Simulate `delay_seconds` for ActiveJob's wait by using visibility timeouts under the hood. We still cap it to non-FIFO's 15 minutes.
+* Set both the messages `message_group_id` and `message_deduplication_id` to the unique job id provided by ActiveJob.
+
+**Can I Use Multiple Queues?** - Yes. Nothing is stopping you from creating any number of queues and/or functions to process them. Your subclasses can use ActiveJob's `queue_as` method as needed. This is an easy way to handle job priorities too.
 
 ```ruby
-# TODO ...
+class SomeLowPriorityJob < ApplicationJob
+  queue_as ENV['BULK_QUEUE_NAME']
+end
 ```
 
+**What Is The Max Message Size?** - 256KB. ActiveJob messages should be small however since Rails uses the [GlobalID](https://github.com/rails/globalid) gem to avoid marshaling large data structures to jobs.
+
 ## Contributing
 
 After checking out the repo, run:

data/Rakefile

@@ -5,7 +5,7 @@ require "rake/testtask"
 Rake::TestTask.new(:test) do |t|
   t.libs << "test"
   t.libs << "lib"
-  t.test_files = FileList["test/**/*_test.rb"]
+  t.test_files = FileList["test/cases/**/*_test.rb"]
   t.verbose = false
   t.warning = false
 end

data/bin/_console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "lambdakiq"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/console

@@ -1,14 +1,6 @@
-#!/usr/bin/env ruby
+#!/bin/bash
+set -e
 
-require "bundler/setup"
-require "lambdakiq"
-
-# You can add fixtures and/or initialization code here to make experimenting
-# with your gem easier. You can also use a different console, if you like.
-
-# (If you use this, don't forget to add pry to your Gemfile!)
-# require "pry"
-# Pry.start
-
-require "irb"
-IRB.start(__FILE__)
+docker-compose run \
+  lambdakiqgem \
+  ./bin/_console

data/{lamby.gemspec → lambdakiq.gemspec}

@@ -12,17 +12,20 @@ Gem::Specification.new do |spec|
   spec.homepage      = "https://github.com/customink/lambdakiq"
   spec.license       = "MIT"
   spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
-    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|images)/}) }
   end
   spec.bindir        = "exe"
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
   spec.add_dependency 'activejob'
   spec.add_dependency 'aws-sdk-sqs'
+  spec.add_dependency 'concurrent-ruby'
+  spec.add_dependency 'railties'
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'rake'
   spec.add_development_dependency 'minitest'
   spec.add_development_dependency 'minitest-focus'
   spec.add_development_dependency 'mocha'
   spec.add_development_dependency 'pry'
+  spec.add_development_dependency 'uuid'
 end

data/lib/lambdakiq.rb

@@ -1,16 +1,41 @@
+require 'json'
+require 'digest'
 require 'active_job'
-require 'aws-sdk-sqs'
+require 'active_job/queue_adapters'
+require 'active_support/all'
 require 'lambdakiq/version'
-
-# if defined?(Rails)
-#   require 'rails/railtie'
-#   require 'lambdakiq/railtie'
-# end
+require 'lambdakiq/error'
+require 'lambdakiq/adapter'
+require 'lambdakiq/client'
+require 'lambdakiq/queue'
+require 'lambdakiq/message'
+require 'lambdakiq/event'
+require 'lambdakiq/job'
+require 'lambdakiq/record'
+require 'lambdakiq/backoff'
+require 'lambdakiq/metrics'
+require 'lambdakiq/worker'
+require 'rails/railtie'
+require 'lambdakiq/railtie'
 
 module Lambdakiq
 
-  extend self
+  def handler(event)
+    Job.handler(event)
+  end
+
+  def jobs?(event)
+    Event.jobs?(event)
+  end
 
-  # autoload :Xyz, 'lambdakiq/xyz'
+  def client
+    @client ||= Client.new
+  end
+
+  def config
+    Lambdakiq::Railtie.config.lambdakiq
+  end
+
+  extend self
 
 end

data/lib/lambdakiq/adapter.rb

@@ -0,0 +1,38 @@
+module ActiveJob
+  module QueueAdapters
+    class LambdakiqAdapter
+
+      def enqueue(job, options = {})
+        job.lambdakiq_async? ? _enqueue_async(job, options) : _enqueue(job, options)
+      end
+
+      def enqueue_at(job, timestamp)
+        enqueue job, delay_seconds: delay_seconds(timestamp)
+      end
+
+      private
+
+      def delay_seconds(timestamp)
+        ds = (timestamp - Time.current.to_i).to_i
+        [ds, 900].min
+      end
+
+      def _enqueue(job, options = {})
+        queue = Lambdakiq.client.queues[job.queue_name]
+        queue.send_message job, options
+      end
+
+      def _enqueue_async(job, options = {})
+        Concurrent::Promise
+          .execute { _enqueue(job, options) }
+          .on_error { |e| async_enqueue_error(e) }
+      end
+
+      def async_enqueue_error(e)
+        msg = "[Lambdakiq] Failed to queue job #{job}. Reason: #{e}"
+        Rails.logger.error(msg)
+      end
+
+    end
+  end
+end

data/lib/lambdakiq/backoff.rb

@@ -0,0 +1,40 @@
+module Lambdakiq
+  class Backoff
+
+    MAX_VISIBILITY_TIMEOUT = 43200 # 12 Hours
+
+    attr_reader :count
+
+    class << self
+
+      def backoff(count)
+        new(count).backoff
+      end
+
+    end
+
+    def initialize(count)
+      @count = count
+    end
+
+    # From Sidekiq: https://git.io/fhi5O
+    #
+    def backoff
+      case count
+      when 1 then 30
+      when 2 then 46
+      when 3 then 76
+      when 4 then 156
+      when 5 then 346
+      when 6 then 730
+      when 7 then 1416
+      when 8 then 2536
+      when 9 then 4246
+      when 10 then 6726
+      when 11 then 10180
+      when 12 then 14836
+      end
+    end
+
+  end
+end

data/lib/lambdakiq/client.rb

@@ -0,0 +1,37 @@
+module Lambdakiq
+  class Client
+
+    class_attribute :default_options,
+                    instance_writer: false,
+                    instance_predicate: false,
+                    default: Hash.new
+
+    attr_reader :queues
+
+    def initialize
+      @queues = Hash.new do |h, name|
+        h[name] = Queue.new(name)
+      end
+    end
+
+    def sqs
+      @sqs ||= begin
+        require 'aws-sdk-sqs'
+        Aws::SQS::Client.new(options)
+      end
+    end
+
+    private
+
+    def options
+      default_options.tap do |opts|
+        opts[:region] ||= region if region
+      end
+    end
+
+    def region
+      ENV['AWS_REGION']
+    end
+
+  end
+end

data/lib/lambdakiq/error.rb

@@ -0,0 +1,21 @@
+module Lambdakiq
+  class Error < StandardError
+  end
+
+  class JobError < Error
+    attr_reader :original_exception, :job
+
+    def initialize(error)
+      @original_exception = error
+      super(error.message)
+      set_backtrace Rails.backtrace_cleaner.clean(error.backtrace)
+    end
+  end
+
+  class FifoDelayError < Error
+    def initialize(error)
+      super
+      set_backtrace([])
+    end
+  end
+end

data/lib/lambdakiq/event.rb

@@ -0,0 +1,19 @@
+module Lambdakiq
+  module Event
+
+    def jobs?(event)
+      records(event).any? { |r| job?(r) }
+    end
+
+    def job?(record)
+      record.dig('messageAttributes', 'lambdakiq', 'stringValue') == '1'
+    end
+
+    def records(event)
+      event['Records'] || []
+    end
+
+    extend self
+
+  end
+end

data/lib/lambdakiq/job.rb

@@ -0,0 +1,120 @@
+module Lambdakiq
+  class Job
+
+    attr_reader :record, :error
+
+    class << self
+
+      def handler(event)
+        records = Event.records(event)
+        jobs = records.map { |record| new(record) }
+        jobs.each(&:perform)
+        jwerror = jobs.detect{ |j| j.error }
+        return unless jwerror
+        raise JobError.new(jwerror.error)
+      end
+
+    end
+
+    def initialize(record)
+      @record = Record.new(record)
+      @error = false
+    end
+
+    def job_data
+      @job_data ||= JSON.parse(record.body).tap do |data|
+        data['provider_job_id'] = record.message_id
+        data['executions'] = record.receive_count - 1
+      end
+    end
+
+    def active_job
+      @active_job ||= ActiveJob::Base.deserialize(job_data)
+    end
+
+    def queue
+      Lambdakiq.client.queues[active_job.queue_name]
+    end
+
+    def executions
+      active_job.executions
+    end
+
+    def perform
+      if fifo_delay?
+        fifo_delay
+        raise FifoDelayError, active_job.job_id
+      end
+      execute
+    end
+
+    def execute
+      ActiveJob::Base.execute(job_data)
+      delete_message
+    rescue Exception => e
+      increment_executions
+      perform_error(e)
+    end
+
+    private
+
+    def client_params
+      { queue_url: queue.queue_url, receipt_handle: record.receipt_handle }
+    end
+
+    def perform_error(e)
+      if change_message_visibility
+        instrument :enqueue_retry, error: e, wait: record.next_visibility_timeout
+        @error = e
+      else
+        instrument :retry_stopped, error: e
+        delete_message
+      end
+    end
+
+    def delete_message
+      client.delete_message(client_params)
+    rescue Exception => e
+      true
+    end
+
+    def change_message_visibility
+      return false if max_receive_count?
+      params = client_params.merge visibility_timeout: record.next_visibility_timeout
+      client.change_message_visibility(params)
+      true
+    end
+
+    def client
+      Lambdakiq.client.sqs
+    end
+
+    def max_receive_count?
+      executions > retry_limit
+    end
+
+    def retry_limit
+      config_retry = [Lambdakiq.config.max_retries, 12].min
+      [ (active_job.lambdakiq_retry || config_retry),
+        (queue.max_receive_count - 1) ].min
+    end
+
+    def fifo_delay?
+      queue.fifo? && record.fifo_delay_seconds?
+    end
+
+    def fifo_delay
+      params = client_params.merge visibility_timeout: record.fifo_delay_visibility_timeout
+      client.change_message_visibility(params)
+    end
+
+    def increment_executions
+      active_job.executions = active_job.executions + 1
+    end
+
+    def instrument(name, error: nil, wait: nil)
+      active_job.send :instrument, name, error: error, wait: wait
+    end
+
+  end
+end

data/lib/lambdakiq/message.rb

@@ -0,0 +1,67 @@
+module Lambdakiq
+  class Message
+    LAMBDAKIQ_ATTRIBUTE = { 'lambdakiq' => { string_value: '1', data_type: 'String' } }.freeze
+
+    attr_reader :queue, :job, :options
+
+    def initialize(queue, job, options = {})
+      @queue = queue
+      @job = job
+      @options = options
+    end
+
+    def params
+      message_params.merge(message_options)
+    end
+
+    private
+
+    def message_params
+      { message_body: message_body,
+        message_attributes: message_attributes }
+        .merge(message_params_fifo)
+    end
+
+    def message_options
+      if queue.fifo?
+        options.except(:delay_seconds)
+      else
+        options
+      end
+    end
+
+    def message_body
+      JSON.dump(job.serialize)
+    end
+
+    def message_params_fifo
+      if queue.fifo?
+        { message_group_id: job.job_id,
+          message_deduplication_id: job.job_id }
+      else
+        {}
+      end
+    end
+
+    def message_attributes
+      LAMBDAKIQ_ATTRIBUTE.merge(delay_seconds_attribute)
+    end
+
+    def delay_seconds
+      options[:delay_seconds] || 0
+    end
+
+    def delay_seconds?
+      !delay_seconds.zero?
+    end
+
+    def delay_seconds_attribute
+      if queue.fifo? && delay_seconds?
+        { 'delay_seconds' => { string_value: delay_seconds.to_s, data_type: 'String' } }
+      else
+        {}
+      end
+    end
+
+  end
+end

data/lib/lambdakiq/metrics.rb

@@ -0,0 +1,110 @@
+module Lambdakiq
+  class Metrics
+    attr_reader :event
+
+    class << self
+      def log(event)
+        new(event).log
+      end
+    end
+
+    def initialize(event)
+      @event = event
+      @metrics = []
+      @properties = {}
+      instrument!
+    end
+
+    def log
+      logger.info JSON.dump(message)
+    end
+
+    private
+
+    def job
+      event.payload[:job]
+    end
+
+    def job_name
+      job.class.name
+    end
+
+    def logger
+      Lambdakiq.config.metrics_logger
+    end
+
+    def namespace
+      Lambdakiq.config.metrics_namespace
+    end
+
+    def exception_name
+      event.payload[:exception].try(:first) ||
+        event.payload[:error]&.class&.name
+    end
+
+    def dimensions
+      [
+        { AppName: rails_app_name },
+        { JobEvent: event.name },
+        { JobName: job_name }
+      ]
+    end
+
+    def instrument!
+      put_metric 'Duration', event.duration.to_i, 'Milliseconds'
+      put_metric 'Count', 1, 'Count'
+      put_metric 'ExceptionCount', 1, 'Count' if exception_name
+      set_property 'JobId', job.job_id
+      set_property 'JobName', job_name
+      set_property 'QueueName', job.queue_name
+      set_property 'MessageId', job.provider_job_id if job.provider_job_id
+      set_property 'ExceptionName', exception_name if exception_name
+      set_property 'EnqueuedAt', job.enqueued_at if job.enqueued_at
+      set_property 'Executions', job.executions if job.executions
+      job.arguments.each_with_index do |argument, index|
+        set_property "JobArg#{index+1}", argument
+      end
+    end
+
+    def put_metric(name, value, unit = nil)
+      @metrics << { 'Name': name }.tap do |m|
+        m['Unit'] = unit if unit
+      end
+      set_property name, value
+    end
+
+    def set_property(name, value)
+      @properties[name] = value
+      self
+    end
+
+    def message
+      {
+        '_aws': {
+          'Timestamp': timestamp,
+          'CloudWatchMetrics': [
+            {
+              'Namespace': namespace,
+              'Dimensions': [dimensions.map(&:keys).flatten],
+              'Metrics': @metrics
+            }
+          ]
+        }
+      }.tap do |m|
+        dimensions.each { |d| m.merge!(d) }
+        m.merge!(@properties)
+      end
+    end
+
+    def timestamp
+      Time.current.strftime('%s%3N').to_i
+    end
+
+    def rails_app_name
+      Lambdakiq.config.metrics_app_name ||
+        Rails.application.class.name.split('::').first
+    end
+
+  end
+end
+

data/lib/lambdakiq/queue.rb

@@ -0,0 +1,55 @@
+module Lambdakiq
+  class Queue
+
+    attr_reader :queue_name,
+                :queue_url
+
+    def initialize(queue_name)
+      @queue_name = queue_name
+      @queue_url = get_queue_url
+      attributes
+    end
+
+    def send_message(job, options = {})
+      client.send_message send_message_params(job, options)
+    end
+
+    def attributes
+      @attributes ||= client.get_queue_attributes({
+        queue_url: queue_url,
+        attribute_names: ['All']
+      }).attributes
+    end
+
+    def redrive_policy
+      @redrive_policy ||= JSON.parse(attributes['RedrivePolicy'])
+    end
+
+    def max_receive_count
+      redrive_policy['maxReceiveCount'].to_i
+    end
+
+    def fifo?
+      queue_name.ends_with?('.fifo')
+    end
+
+    private
+
+    def client
+      Lambdakiq.client.sqs
+    end
+
+    def get_queue_url
+      client.get_queue_url(queue_name: queue_name).queue_url
+    end
+
+    def send_message_params(job, options)
+      { queue_url: queue_url }.merge(message_params(job, options))
+    end
+
+    def message_params(job, options)
+      Message.new(self, job, options).params
+    end
+
+  end
+end

data/lib/lambdakiq/railtie.rb

@@ -1,5 +1,19 @@
 module Lambdakiq
   class Railtie < ::Rails::Railtie
     config.lambdakiq = ActiveSupport::OrderedOptions.new
+    config.lambdakiq.max_retries = 12
+    config.lambdakiq.metrics_namespace = 'Lambdakiq'
+
+    config.after_initialize do
+      config.active_job.logger = Rails.logger
+      config.lambdakiq.metrics_logger = Rails.logger
+    end
+
+    initializer "lambdakiq.metrics" do |app|
+      ActiveSupport::Notifications.subscribe(/active_job/) do |*args|
+        event = ActiveSupport::Notifications::Event.new *args
+        Lambdakiq::Metrics.log(event)
+      end
+    end
   end
 end

data/lib/lambdakiq/record.rb

@@ -0,0 +1,58 @@
+module Lambdakiq
+  class Record
+
+    attr_reader :data
+
+    def initialize(data)
+      @data = data
+    end
+
+    def body
+      data['body']
+    end
+
+    def message_id
+      data['messageId']
+    end
+
+    def receipt_handle
+      data['receiptHandle']
+    end
+
+    def queue_name
+      @queue_name ||= data['eventSourceARN'].split(':').last
+    end
+
+    def attributes
+      data['attributes']
+    end
+
+    def fifo_delay_visibility_timeout
+      fifo_delay_seconds - (Time.current - sent_at).to_i
+    end
+
+    def fifo_delay_seconds
+      data.dig('messageAttributes', 'delay_seconds', 'stringValue').try(:to_i)
+    end
+
+    def fifo_delay_seconds?
+      fifo_delay_seconds && (sent_at + fifo_delay_seconds).future?
+    end
+
+    def sent_at
+       @sent_at ||= begin
+        ts = attributes['SentTimestamp'].to_i / 1000
+        Time.zone ? Time.zone.at(ts) : Time.at(ts)
+      end
+    end
+
+    def receive_count
+      @receive_count ||= attributes['ApproximateReceiveCount'].to_i
+    end
+
+    def next_visibility_timeout
+      @next_visibility_timeout ||= Backoff.backoff(receive_count)
+    end
+
+  end
+end

data/lib/lambdakiq/version.rb

@@ -1,3 +1,3 @@
 module Lambdakiq
-  VERSION = '0.0.1'
+  VERSION = '1.0.1'
 end

data/lib/lambdakiq/worker.rb

@@ -0,0 +1,28 @@
+module Lambdakiq
+  module Worker
+    extend ActiveSupport::Concern
+
+    included do
+      class_attribute :lambdakiq_options_hash,
+                      instance_predicate: false,
+                      default: Hash.new
+    end
+
+    class_methods do
+
+      def lambdakiq_options(options = {})
+        self.lambdakiq_options_hash = options.symbolize_keys
+      end
+
+    end
+
+    def lambdakiq_retry
+      lambdakiq_options_hash[:retry]
+    end
+
+    def lambdakiq_async?
+      !!lambdakiq_options_hash[:async]
+    end
+
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lambdakiq
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 1.0.1
 platform: ruby
 authors:
 - Ken Collins
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-11-29 00:00:00.000000000 Z
+date: 2021-02-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activejob
@@ -38,6 +38,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -122,6 +150,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: uuid
+  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: Scalable Rails Background Processing with AWS Lambda & SQS.
 email:
 - kcollins@customink.com
@@ -139,6 +181,7 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- bin/_console
 - bin/_setup
 - bin/_test
 - bin/bootstrap
@@ -147,10 +190,21 @@ files:
 - bin/test
 - bin/update
 - docker-compose.yml
-- lamby.gemspec
+- lambdakiq.gemspec
 - lib/lambdakiq.rb
+- lib/lambdakiq/adapter.rb
+- lib/lambdakiq/backoff.rb
+- lib/lambdakiq/client.rb
+- lib/lambdakiq/error.rb
+- lib/lambdakiq/event.rb
+- lib/lambdakiq/job.rb
+- lib/lambdakiq/message.rb
+- lib/lambdakiq/metrics.rb
+- lib/lambdakiq/queue.rb
 - lib/lambdakiq/railtie.rb
+- lib/lambdakiq/record.rb
 - lib/lambdakiq/version.rb
+- lib/lambdakiq/worker.rb
 - vendor/.keep
 homepage: https://github.com/customink/lambdakiq
 licenses: