lambda_punch 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b9dd35b9fd9bf01ed3c9d3538d650181cd33ef3d78113e6e743f16a48e4cd84d
-  data.tar.gz: a72d215a6474275f2815bc34feb7dd2970d5a05d76a2396e04cb9744194e7433
+  metadata.gz: 415dae29a1a4d52820fbd86aa6dc1a03dc4eb3129098851a2db9ed94385c257f
+  data.tar.gz: 06a861cc3d854392ad2fcb049b4e2acce49536cbb6ec41ba4ea3021b4e43c994
 SHA512:
-  metadata.gz: 922db1f4eeb16924b5293ece9e70568a7f2035d87dbfc40c9c42742c8dbae6a013578a345b9913e9cf6a87cb4d865a938a7324b79462960a0ecb9a543850849e
-  data.tar.gz: 198b757f6a2d4f2692af2659095cd9d9e5ce427596094d43eebfcdfd290686f4eb371662a94a67781ea5a2576fb874b690b21979b34473ce7c9272157df63cb4
+  metadata.gz: e3fc61d1620bf1676c96c6fb37221bdc9138797feeebecae40c20300c001a62b560231ccc69b936514d5846220066175a757af24b3ba42aaccbf054377c3971c
+  data.tar.gz: 92ef0f6674b3d42c5f35fc2912653db1b8cb15506e8dd9085fe34b84be305b742bc12decb575ab7f61befc412e5903c9116ea57a9a9838482aacba475a640c17

data/.github/workflows/test.yml

@@ -0,0 +1,17 @@
+name: Test
+on: [push]
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+      - name: Bootstrap
+        run: |
+          ./bin/bootstrap
+      - name: Setup
+        run: |
+          ./bin/setup
+      - name: Test
+        run: |
+          ./bin/test

data/.gitignore

@@ -1,4 +1,5 @@
-/.bundle/
+/.bundle/*
+/vendor/bunlde/*
 /.yardoc
 /_yardoc/
 /coverage/

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [0.0.2] - 2021-06-28
+
+- Alpha release.
+
 ## [0.0.1] - 2021-06-23
 
-- Initial release
+- Project name hold.

data/Dockerfile

@@ -0,0 +1,2 @@
+FROM public.ecr.aws/sam/build-ruby2.7
+WORKDIR /var/task

data/Gemfile

@@ -1,10 +1,5 @@
-# frozen_string_literal: true
-
 source "https://rubygems.org"
-
-# Specify your gem's dependencies in lambda_punch.gemspec
 gemspec
 
-gem "rake", "~> 13.0"
-
-gem "minitest", "~> 5.0"
+gem "rake"
+gem "minitest"

data/Gemfile.lock

@@ -0,0 +1,30 @@
+PATH
+  remote: .
+  specs:
+    lambda_punch (0.0.2)
+      concurrent-ruby
+      rb-inotify
+      timeout
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    concurrent-ruby (1.1.9)
+    ffi (1.15.3)
+    minitest (5.14.4)
+    rake (13.0.3)
+    rb-inotify (0.10.1)
+      ffi (~> 1.0)
+    timeout (0.1.1)
+
+PLATFORMS
+  x86_64-darwin-19
+  x86_64-linux
+
+DEPENDENCIES
+  lambda_punch!
+  minitest
+  rake
+
+BUNDLED WITH
+   2.2.21

data/README.md

@@ -1,43 +1,126 @@
-# LambdaPunch
+![LambdaPunch](https://user-images.githubusercontent.com/2381/123561512-c23fb580-d776-11eb-9780-71d606cd8f2c.png)
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/lambda_punch`. To experiment with that code, run `bin/console` for an interactive prompt.
+![Test](https://github.com/customink/lambda_punch/workflows/Test/badge.svg)
 
-TODO: Delete this and the text above, and describe your gem
+# 👊 LambdaPunch
 
-## Installation
+<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>Asynchronous background job processing for AWS Lambda with Ruby using [Lambda Extensions](https://docs.aws.amazon.com/lambda/latest/dg/runtimes-extensions-api.html). Inspired by the [SuckerPunch](https://github.com/brandonhilkert/sucker_punch) gem but specifically tooled to work with Lambda's invoke model.
 
-Add this line to your application's Gemfile:
+**For a more robust background job solution, please consider using AWS SQS with the [Lambdakiq](https://github.com/customink/lambdakiq) gem. 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.**
+
+## 🏗 Architecture
+
+Because AWS Lambda [freezes the execution environment](https://docs.aws.amazon.com/lambda/latest/dg/runtimes-context.html) after each invoke, there is no "process" that continues to run after the handler's response. However, thanks to Lambda Extensions along with its "early return", we can do two important things. First, we leverage the [rb-inotify](https://github.com/guard/rb-inotify) gem to send the extension process a simulated `POST-INVOKE` event. We then use [Distributed Ruby](https://ruby-doc.org/stdlib-3.0.1/libdoc/drb/rdoc/DRb.html) (DRb) from the extension to signal your application to work jobs off a queue. Both of these are synchronous calls. Once complete the LambdaPunch extensions signals it is done and your function is ready for the next request.
+
+<img src="https://user-images.githubusercontent.com/2381/123647632-408f6c80-d7f6-11eb-8e39-fb4ee92b1ffa.png" width="100%" alt="AWS Lambda Extensions with LambdaPunch async job queue processing." >
+
+The LambdaPunch extension process is very small and lean. It only requires a few Ruby libraries and needed gems from your application's bundle. Its only job is to send signals back to your application on the runtime. It does this within a few milliseconds and adds no noticeable overhead to your function.
+
+## 🎁 Installation
+
+Add this line to your project's `Gemfile` and then make sure to `bundle install` afterward.
 
 ```ruby
 gem 'lambda_punch'
 ```
 
-And then execute:
+Now, within your application's handler file, make sure to start the LambdaPunch DRb server outside of your handler method. Within the handler method, add an `ensure` section that lets the extension process know the request is done.
 
-    $ bundle install
+```ruby
+LambdaPunch.start_server!
 
-Or install it yourself as:
+def handler(event:, context:)
+  # ...
+ensure
+  LambdaPunch.handled!(context)
+end
+```
 
-    $ gem install lambda_punch
+Within your project or [Rails application's](https://lamby.custominktech.com/docs/anatomy) `Dockerfile`, after you copy your code, add this `RUN` command to install the extension within your container's `/opt/extensions` directory.
+
+```dockerfile
+RUN bundle exec rake lambda_punch:install
+```
+
+## 🧰 Usage
+
+Anywhere in your application's code, use the `LambdaPunch.push` method to add blocks of code to your jobs queue.
+
+```ruby
+LambdaPunch.push do
+  # ...
+end
+```
+
+For example, if you are using Rails with AWS Lambda via the [Lamby](https://lamby.custominktech.com/) gem along with [New Relic APM](https://dev.to/aws-heroes/using-new-relic-apm-with-rails-on-aws-lambda-51gi) here is how your handler function might appear to ensure their metrics are flushed after each request.
+
+```ruby
+def handler(event:, context:)
+  Lamby.handler $app, event, context
+ensure
+  LambdaPunch.push { NewRelic::Agent.agent.flush_pipe_data }
+  LambdaPunch.handled!(context)
+end
+```
 
-## Usage
+### ActiveJob
 
-TODO: Write usage instructions here
+🚧 COMING SOON 🚧 - A simple ActiveJob adapter...
 
-## Development
+### Timeouts
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+Your function's timeout is the max amount to handle the request and process all extension's invoke events. If your function times out, it is possible that queued jobs will not be processed until the next invoke.
+
+If your application integrates with API Gateway (which has a 30 second timeout) then it is possible your function can be set with a higher timeout to perform background work. Since work is done after each invoke, the LambdaPunch queue should be empty when your function receives the `SHUTDOWN` event. If jobs are in the queue when this happens they will have two seconds max to work them down before being lost.
+
+**For a more robust background job solution, please consider using AWS SQS with the [Lambdakiq](https://github.com/customink/lambdakiq) gem. 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.**
+
+### Logging
+
+The default log level is `fatal`, so you will not see any LambdaPunch lines in your logs. However, if you want some low level debugging information on how LambdaPunch is working, you can use this environment variable to change the log level.
+
+```yaml
+Environment:
+  Variables:
+    LAMBDA_PUNCH_LOG_LEVEL: debug
+```
+
+## 📊 CloudWatch Metrics
+
+When using Extensions, your function's CloudWatch `Duration` metrics will be the sum of your response time combined with your extension's execution time. For example, if your request takes `200ms` to respond but your background task takes `1000ms` your duration will be a combined `1200ms`. For more details see the _"Performance impact and extension overhead"_ section of the [Lambda Extensions API
+](https://docs.aws.amazon.com/lambda/latest/dg/runtimes-extensions-api.html)
+
+Thankfully, when using Lambda Extensions, CloudWatch will create a `PostRuntimeExtensionsDuration` metric that you can use to isolate your true response times `Duration` [using some metric math](https://docs.aws.amazon.com/AmazonCloudWatch/latest/monitoring/using-metric-math.html). Here is an example where the metric math above is used in the first "Duration (response)" widget.
+
+![metric-math](https://user-images.githubusercontent.com/2381/123561591-4eea7380-d777-11eb-8682-c20b9460f112.png)
+
+![durations](https://user-images.githubusercontent.com/2381/123561590-4e51dd00-d777-11eb-96b2-d886c91aedb0.png)
+
+## 👷🏽‍♀️ Development
+
+After checking out the repo, run the following commands to build a Docker image and install dependencies.
+
+```shell
+$ ./bin/bootstrap
+$ ./bin/setup
+```
+
+Then, to run the tests use the following command.
+
+```shell
+$ ./bin/test
+```
 
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+You can also run the `./bin/console` command for an interactive prompt within the development Docker container. Likewise you can use `./bin/run ...` followed by any command which would be executed within the same container.
 
-## Contributing
+## 💖 Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/metaskills/lambda_punch. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/metaskills/lambda_punch/blob/main/CODE_OF_CONDUCT.md).
+Bug reports and pull requests are welcome on GitHub at https://github.com/customink/lambda_punch. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/customink/lambda_punch/blob/main/CODE_OF_CONDUCT.md).
 
-## License
+## 👩‍⚖️ License
 
 The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
 
-## Code of Conduct
+## 🤝 Code of Conduct
 
-Everyone interacting in the LambdaPunch project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/metaskills/lambda_punch/blob/main/CODE_OF_CONDUCT.md).
+Everyone interacting in the LambdaPunch project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/customink/lambda_punch/blob/main/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 require "bundler/gem_tasks"
 require "rake/testtask"
 

data/bin/_setup

@@ -0,0 +1,6 @@
+#!/bin/sh
+set -e
+
+echo '== Installing dependencies =='
+bundle config set --local path 'vendor/bundle'
+bundle install

data/bin/_test

@@ -0,0 +1,4 @@
+#!/bin/sh
+set -e
+
+bundle exec rake test

data/bin/bootstrap

@@ -0,0 +1,6 @@
+#!/bin/sh
+set -e
+
+echo '== Building containers =='
+docker pull public.ecr.aws/sam/build-ruby2.7
+docker-compose build 

data/bin/console

@@ -1,15 +1,6 @@
-#!/usr/bin/env ruby
-# frozen_string_literal: true
+#!/bin/sh
+set -e
 
-require "bundler/setup"
-require "lambda_punch"
-
-# 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 \
+  lambdapunch \
+  bundle console

data/bin/run

@@ -0,0 +1,6 @@
+#!/bin/sh
+set -e
+
+docker-compose run \
+  lambdapunch \
+  $@

data/bin/setup

@@ -1,8 +1,6 @@
-#!/usr/bin/env bash
-set -euo pipefail
-IFS=$'\n\t'
-set -vx
+#!/bin/sh
+set -e
 
-bundle install
-
-# Do any other automated setup that you need to do here
+docker-compose run \
+  lambdapunch \
+  ./bin/_setup

data/bin/test

@@ -0,0 +1,6 @@
+#!/bin/sh
+set -e
+
+docker-compose run \
+  lambdapunch \
+  ./bin/_test

data/docker-compose.yml

@@ -0,0 +1,10 @@
+version: '3.7'
+services:
+  lambdapunch:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    environment:
+      - CI=${CI}
+    volumes:
+      - ${PWD}:/var/task:delegated

data/lambda_punch.gemspec

@@ -18,5 +18,7 @@ Gem::Specification.new do |spec|
   spec.bindir        = "exe"
   spec.executables   = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
-  # spec.add_dependency "example-gem", "~> 1.0"
+  spec.add_dependency "concurrent-ruby"
+  spec.add_dependency "rb-inotify"
+  spec.add_dependency "timeout"
 end

data/lib/lambda_punch.rb

@@ -1,8 +1,50 @@
-# frozen_string_literal: true
-
-require_relative "lambda_punch/version"
+require 'uri'
+require 'drb'
+require 'json'
+require 'tmpdir'
+require 'logger'
+require 'net/http'
+require 'singleton'
+require 'lambda_punch/api'
+require 'lambda_punch/error'
+require 'lambda_punch/logger'
+require 'lambda_punch/queue'
+require 'lambda_punch/server'
+require 'lambda_punch/worker'
+require 'lambda_punch/version'
+require 'lambda_punch/notifier'
+require 'lambda_punch/railtie' if defined?(Rails)
 
 module LambdaPunch
-  class Error < StandardError; end
-  # Your code goes here...
+  
+  def push(&block)
+    Queue.push(block)
+  end
+
+  def register!
+    Api.register!
+  end
+
+  def loop
+    Api.loop
+  end
+
+  def start_server!
+    Server.start!
+  end
+
+  def start_worker!
+    Worker.start!
+  end
+
+  def logger
+    @logger ||= Logger.new.logger
+  end
+
+  def handled!(context)
+    Notifier.handled!(context)
+  end
+
+  extend self
+
 end

data/lib/lambda_punch/api.rb

@@ -0,0 +1,93 @@
+module LambdaPunch
+  # Interface to Lambda's Extensions API using simple `Net::HTTP` calls.
+  # 
+  #   Lambda Extensions API
+  #   https://docs.aws.amazon.com/lambda/latest/dg/runtimes-extensions-api.html
+  # 
+  class Api
+
+    EXTENSION_NAME = 'lambdapunch'
+
+    include Singleton
+
+    class << self
+
+      def register!
+        instance.register!
+      end
+
+      def loop
+        instance.loop
+      end
+
+    end
+
+    def register!
+      return if @registered
+      uri = URI.parse "#{base_uri}/register"
+      http = Net::HTTP.new uri.host, uri.port
+      request = Net::HTTP::Post.new uri.request_uri
+      request['Content-Type'] = 'application/vnd.aws.lambda.extension+json'
+      request['Lambda-Extension-Name'] = EXTENSION_NAME
+      request.body = %q|{"events":["INVOKE","SHUTDOWN"]}|
+      http.request(request).tap do |r|
+        logger.debug "Api#register! => #{r.class.name.inspect}, body: #{r.body}"
+        @registered = true
+        @extension_id = r.each_header.to_h['lambda-extension-identifier']
+        logger.debug "Api::ExtensionId => #{@extension_id}"
+      end
+    end
+
+    def loop
+      resp = event_next
+      event_payload = JSON.parse(resp.body)
+      case event_payload['eventType']
+      when 'INVOKE'   then invoke(event_payload)
+      when 'SHUTDOWN' then shutdown
+      else
+        event_type_error(event_payload)
+      end
+    end
+
+    private
+
+    def event_next
+      uri = URI.parse "#{base_uri}/event/next"
+      http = Net::HTTP.new uri.host, uri.port
+      request = Net::HTTP::Get.new uri.request_uri
+      request['Content-Type'] = 'application/vnd.aws.lambda.extension+json'
+      request['Lambda-Extension-Identifier'] = @extension_id
+      http.request(request).tap do |r|
+        logger.debug "Api#event_next => #{r.class.name.inspect}, body: #{r.body}"
+      end
+    end
+
+    def invoke(event_payload)
+      logger.debug "Api#invoke => #{JSON.dump(event_payload)}" if logger.debug?
+      Worker.call(event_payload)
+    end
+
+    def shutdown
+      logger.info 'Api#shutdown...'
+      DRb.stop_service rescue true
+      exit
+    end
+
+    private
+
+    def base_uri
+      "http://#{ENV['AWS_LAMBDA_RUNTIME_API']}/2020-01-01/extension"
+    end
+
+    def logger
+      LambdaPunch.logger
+    end
+
+    def event_type_error(event_payload)
+      message = "Unknown event type: #{event_payload['eventType'].inspect}"
+      logger.fatal(message)
+      raise EventTypeError.new(message)
+    end
+
+  end
+end

data/lib/lambda_punch/error.rb

@@ -0,0 +1,7 @@
+module LambdaPunch
+  class Error < StandardError
+  end
+
+  class EventTypeError < Error
+  end
+end

data/lib/lambda_punch/extensions/lambdapunch

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+ENV['BUNDLE_GEMFILE'] = "#{ENV['LAMBDA_TASK_ROOT']}/Gemfile"
+require 'bundler/setup'
+require 'lambda_punch'
+
+LambdaPunch.register!
+LambdaPunch.start_worker!
+
+while true do
+  LambdaPunch.loop
+end

data/lib/lambda_punch/logger.rb

@@ -0,0 +1,24 @@
+module LambdaPunch
+  class Logger
+
+    def logger
+      @logger ||= ::Logger.new(STDOUT).tap do |l| 
+        l.level = level
+        l.formatter = proc { |_s, _d, _p, m| "[LambdaPunch] #{m}\n" }
+      end
+    end
+
+    def level=(value)
+      @level = value.to_s
+      @logger = nil
+    end
+
+    private
+
+    def level
+      l = (@level || ENV['LAMBDA_PUNCH_LOG_LEVEL'] || 'fatal').upcase.to_sym
+      ::Logger.const_defined?(l) ? ::Logger.const_get(l) : ::Logger::FATAL
+    end
+
+  end
+end

data/lib/lambda_punch/notifier.rb

@@ -0,0 +1,49 @@
+module LambdaPunch
+  class Notifier
+
+    FILE = "#{Dir.tmpdir}/lambdapunch-handled"
+    File.open(FILE, 'w') { |f| f.write('') }
+
+    class << self
+
+      def handled!(context)
+        File.open(FILE, 'w') do |f|
+          f.write context.aws_request_id
+        end
+      end
+
+      def request_id
+        File.read(FILE)
+      end
+
+    end
+
+    def initialize
+      @notifier = INotify::Notifier.new
+    end
+
+    def watch
+      @notifier.watch(FILE, :modify, :oneshot) { yield(request_id) }
+    end
+
+    def process
+      @notifier.process
+    end
+
+    def close
+      logger.debug "Notifier#close"
+      @notifier.close rescue true
+    end
+
+    def request_id
+      self.class.request_id
+    end
+
+    private
+
+    def logger
+      LambdaPunch.logger
+    end
+
+  end
+end

data/lib/lambda_punch/queue.rb

@@ -0,0 +1,41 @@
+module LambdaPunch
+  class Queue
+
+    class << self
+
+      def push(block)
+        jobs << block
+      end
+
+      def jobs
+        @jobs ||= Concurrent::Array.new
+      end
+
+    end
+
+    def call
+      jobs.each do |job| 
+        begin
+          job.call
+        rescue => e
+          logger.error "Queue#call::error => #{e.message}"
+          # ...
+        end
+      end
+      true
+    ensure
+      jobs.clear
+    end
+
+    private
+
+    def jobs
+      self.class.jobs
+    end
+
+    def logger
+      LambdaPunch.logger
+    end
+    
+  end
+end

data/lib/lambda_punch/railtie.rb

@@ -0,0 +1,12 @@
+require 'rails'
+require 'rails/engine'
+
+module LambdaPunch
+  class Railtie < Rails::Railtie
+    railtie_name :lambda_punch
+
+    rake_tasks do
+      load "lambda_punch/tasks/install.rake"
+    end
+  end
+end

data/lib/lambda_punch/server.rb

@@ -0,0 +1,26 @@
+module LambdaPunch
+  class Server
+
+    include Singleton
+
+    class << self
+
+      def uri
+        'druby://127.0.0.1:9030'
+      end
+
+      def start!
+        require 'concurrent'
+        LambdaPunch.logger.info "Server.start!..."
+        instance
+      end
+
+    end
+
+    def initialize
+      @queue = Queue.new
+      DRb.start_service self.class.uri, @queue
+    end
+
+  end
+end

data/lib/lambda_punch/tasks/install.rake

@@ -0,0 +1,11 @@
+namespace :lambda_punch do
+
+  desc "Install the LambdaPunch Lambda Extension."
+  task :install do
+    require 'fileutils'
+    FileUtils.mkdir_p '/opt/extensions'
+    extension = File.expand_path "#{__dir__}/../extensions/lambdapunch"
+    FileUtils.cp extension '/opt/extensions/'
+  end
+
+end

data/lib/lambda_punch/version.rb

@@ -1,5 +1,3 @@
-# frozen_string_literal: true
-
 module LambdaPunch
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/lambda_punch/worker.rb

@@ -0,0 +1,112 @@
+module LambdaPunch
+  # This `LambdaPunch::Worker` has a few responsibilities:
+  # 
+  #   1. Maintain a class level DRb reference to your function's `LambdaPunch::Queue` object.
+  #   2. Process extension `INVOKE` events by waiting for your function to complete.
+  #   3. Triggering your application to perform work after each request.
+  # 
+  class Worker
+
+    class << self
+      
+      # Method to lazily require rb-inotify and start the DRb service.
+      # 
+      def start!
+        LambdaPunch.logger.info "Worker.start!..."
+        require 'timeout'
+        require 'rb-inotify'
+        DRb.start_service
+        @queue = DRbObject.new_with_uri(Server.uri)
+      end
+
+      # Creates a new instance of this object with the event payload from the `LambdaPunch::Api#invoke` 
+      # method and immediately performs the `call` method which waits for the function's handler to complete.
+      #
+      def call(event_payload)
+        new(event_payload).call
+      end
+
+      # The `@queue` object is the local process' reference to the application `LambdaPunch::Queue`
+      # instance which does all the work in the applciation's scope.
+      # 
+      def queue
+        @queue
+      end
+
+    end
+
+    def initialize(event_payload)
+      @invoked = false
+      @event_payload = event_payload
+      @notifier = Notifier.new
+      @notifier.watch { |request_id| notified(request_id) }
+      @request_id_notifier = nil
+    end
+
+    # Here we wait for the application's handler to signal it is done via the `LambdaPunch::Notifier` or if the 
+    # function has timed out. In either event there may be work to perform in the `LambdaPunch::Queue`. This method
+    # also ensures any clean up is done. For example, closing file notifications.
+    #
+    def call
+      Timeout.timeout(timeout) { @notifier.process }
+    rescue Timeout::Error
+      logger.debug "Worker#call => Function timeout reached."
+    ensure
+      @notifier.close
+      self.class.queue.call
+    end
+
+    private
+
+    # The Notifier's watch handler would set this instance variable to `true`. We also return `true`
+    # if the extension's invoke palyload event has a `requestId` matching what the handler has written
+    # to the `LambdaPunch::Notifier` file location. See also `request_ids_match?` method.
+    #
+    def invoked?
+      @invoked || request_ids_match?
+    end
+
+    # The unique AWS reqeust id that both the extension and handler receive for each invoke. This one
+    # represents the extension's side.
+    # 
+    def request_id_payload
+      @event_payload['requestId']
+    end
+
+    # Set via the `LambdaPunch::Notifier` watch event from the your function's handler.
+    # 
+    def request_id_notifier
+      @request_id_notifier
+    end
+
+    # Check if notified via inotify or in some rare case the function's handler has already completed 
+    # and written the matching request id via the context object to the `LambdaPunch::Notifier` file.
+    #
+    def request_ids_match?
+      request_id_payload == (request_id_notifier || Notifier.request_id)
+    end
+
+    # The function's timeout in seconds using the `INVOKE` event payload's `deadlineMs` value.
+    # 
+    def timeout
+      deadline_milliseconds = @event_payload['deadlineMs']
+      deadline = Time.at(deadline_milliseconds / 1000.0)
+      deadline_timeout = deadline - Time.now
+      deadline_timeout > 0 ? deadline_timeout : 0
+    end
+
+    # Our `LambdaPunch::Notifier` instance callback.
+    # 
+    def notified(request_id)
+      @invoked = true
+      @request_id_notifier = request_id
+    end
+
+    def logger
+      LambdaPunch.logger
+    end
+
+    def noop ; end
+
+  end
+end

metadata

@@ -1,15 +1,57 @@
 --- !ruby/object:Gem::Specification
 name: lambda_punch
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Ken Collins
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-06-23 00:00:00.000000000 Z
-dependencies: []
+date: 2021-06-28 00:00:00.000000000 Z
+dependencies:
+- !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: rb-inotify
+  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: timeout
+  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'
 description: 'LambdaPunch: Async Processing using Lambda Extensions'
 email:
 - ken@metaskills.net
@@ -17,19 +59,37 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".github/workflows/main.yml"
+- ".github/workflows/test.yml"
 - ".gitignore"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
+- Dockerfile
 - Gemfile
+- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
+- bin/_setup
+- bin/_test
+- bin/bootstrap
 - bin/console
+- bin/run
 - bin/setup
+- bin/test
+- docker-compose.yml
 - lambda_punch.gemspec
 - lib/lambda_punch.rb
+- lib/lambda_punch/api.rb
+- lib/lambda_punch/error.rb
+- lib/lambda_punch/extensions/lambdapunch
+- lib/lambda_punch/logger.rb
+- lib/lambda_punch/notifier.rb
+- lib/lambda_punch/queue.rb
+- lib/lambda_punch/railtie.rb
+- lib/lambda_punch/server.rb
+- lib/lambda_punch/tasks/install.rake
 - lib/lambda_punch/version.rb
+- lib/lambda_punch/worker.rb
 homepage: https://github.com/customink/lambda_punch
 licenses:
 - MIT

data/.github/workflows/main.yml

@@ -1,18 +0,0 @@
-name: Ruby
-
-on: [push,pull_request]
-
-jobs:
-  build:
-    runs-on: ubuntu-latest
-    steps:
-    - uses: actions/checkout@v2
-    - name: Set up Ruby
-      uses: ruby/setup-ruby@v1
-      with:
-        ruby-version: 2.7.0
-    - name: Run the default task
-      run: |
-        gem install bundler -v 2.2.15
-        bundle install
-        bundle exec rake