simplekiq 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 4caaf4c9bfc49507c627f025b8847b4cbfd36805897abc1c5d14117327cb0313
+  data.tar.gz: cdff30ac40b315e7e8dca636d013153631b710b8fb65c2bfffdf36632900b293
+SHA512:
+  metadata.gz: 43789a7c661f3981035069935eecdefa4c03f8bfc2bd0eaeba8840018f54813f103475a5cd149752cbbef48e30f7c7363c98af548d85277995e0699df58b00dd
+  data.tar.gz: 9ab17b4b9d419d2551fd8cfe6070443015e35c1dc9c6adf195302a912aff52caf67fc7a5edea2c9f7733c43e0e799b790e00be07a4fbc4b6d0586501372be20d

data/.circleci/config.yml

@@ -0,0 +1,125 @@
+version: 2.1
+
+orbs:
+  gem: doximity/gem-publisher@0
+
+executors:
+  ruby-2-6:
+    resource_class: small
+    docker:
+      - image: cimg/ruby:2.6
+        environment:
+          BUNDLE_VERSION: "~> 1.17"
+
+# yaml anchor filters
+master_only: &master_only
+  filters:
+    branches:
+      only: master
+    tags:
+      ignore: /.*/
+pr_only: &pr_only
+  filters:
+    branches:
+      ignore: master
+    tags:
+      ignore: /.*/
+version_tags_only: &version_tags_only
+  filters:
+    branches:
+      ignore: /.*/
+    tags:
+      only: /^v.*/
+
+jobs:
+  build:
+    executor: ruby-2-6
+    steps:
+      - checkout
+      - run:
+          name: Install Bundler specific version
+          command: |
+            gem install bundler --version "${BUNDLE_VERSION}" --force
+      - restore_cache:
+          keys:
+            - v1-bundle-{{ checksum "Gemfile.lock" }}-
+      - run:
+          name: Install Ruby Dependencies
+          command: |
+            bundle config set --local path 'vendor/bundle'
+            bundle config set --local frozen 'true'
+            bundle install --local --jobs=4 --retry=3
+      - save_cache:
+          key: v1-bundle-{{ checksum "Gemfile.lock" }}-
+          paths:
+            - vendor/bundle
+      - run:
+          name: Run Tests
+          command: bundle exec rake ci:specs
+      - store_test_results:
+          name: Store test results
+          path: tmp/test-results
+      - run:
+          name: Run StandardRB
+          command: bundle exec standardrb
+      - persist_to_workspace:
+          root: .
+          paths:
+            - vendor/bundle
+
+workflows:
+  version: 2
+
+  trunk:
+    jobs:
+      - build:
+          <<: *master_only
+      - gem/build:
+          <<: *master_only
+          executor: ruby-2-6
+          name: gem-build
+          requires:
+            - build
+
+  pull-requests:
+    jobs:
+      - build:
+          <<: *pr_only
+      - gem/build:
+          <<: *pr_only
+          executor: ruby-2-6
+          name: gem-build
+          requires:
+            - build
+      - pre-release-approval:
+          <<: *pr_only
+          type: approval
+          requires:
+            - gem-build
+      - gem/publish:
+          <<: *pr_only
+          name: gem-publish
+          to_rubygems: true
+          pre_release: true
+          requires:
+            - pre-release-approval
+          context: artifact_publishing
+
+  final-release:
+    jobs:
+      - build:
+          <<: *version_tags_only
+      - gem/build:
+          <<: *version_tags_only
+          executor: ruby-2-6
+          name: gem-build
+          requires:
+            - build
+      - gem/publish:
+          <<: *version_tags_only
+          name: gem-publish
+          to_rubygems: true
+          pre_release: false
+          requires:
+            - gem-build
+          context: artifact_publishing

data/.gitignore

@@ -0,0 +1,10 @@
+.rspec_status
+/.ruby-version
+/*.gem
+/.bundle/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/vendor/bundle/

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/CHANGELOG.md

@@ -0,0 +1,16 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [0.0.3]
+* Misc minimal fixes to get the gem building and releasable
+  [#2](https://github.com/doximity/rake-ui/pull/3)
+* Copy over library code from prior sources with maintained history
+  [#1](https://github.com/doximity/simplekiq/pull/1)
+
+## [0.0.2]
+* Scrubbed version from rubygems, do not use
+## [0.0.1]
+* Scrubbed version from rubygems, do not use

data/CONTRIBUTING.md

@@ -0,0 +1,31 @@
+# Contributing
+
+We welcome contributions to this repository. Feel free to submit issues for bugs you encounter and pull requests for code and documentation contributions.
+
+In order to prevent licensing issues, we require all contributors to sign an individual contributor license agreement, which is reproduced below:
+
+## Individual Contributor License Agreement
+
+In order to clarify the intellectual property license granted with Contributions from any person or entity, Doximity Inc. ("Doximity") must have a Contributor License Agreement ("CLA") on file that has been signed by each Contributor, indicating agreement to the license terms below. This license is for your protection as a Contributor as well as the protection of Doximity; it does not change your rights to use your own Contributions for any other purpose.
+
+You accept and agree to the following terms and conditions for Your present and future Contributions submitted to Doximity. Except for the license granted herein to Doximity and recipients of software distributed by Doximity, You reserve all right, title, and interest in and to Your Contributions.
+
+### Definitions
+
+"You" (or "Your") shall mean the copyright owner or legal entity authorized by the copyright owner that is making this Agreement with Doximity. For legal entities, the entity making a Contribution and all other entities that control, are controlled by, or are under common control with that entity are considered to be a single Contributor. For the purposes of this definition, "control" means (i) the power, direct or indirect, to cause the direction or management of such entity, whether by contract or otherwise, or (ii) ownership of fifty percent (50%) or more of the outstanding shares, or (iii) beneficial ownership of such entity.
+
+1. "Contribution" shall mean any original work of authorship, including any modifications or additions to an existing work, that is intentionally submitted by You to Doximity for inclusion in, or documentation of, any of the products owned or managed by Doximity (the "Work"). For the purposes of this definition, "submitted" means any form of electronic, verbal, or written communication sent to Doximity or its representatives, including but not limited to communication on electronic mailing lists, source code control systems, and issue tracking systems that are managed by, or on behalf of, Doximity for the purpose of discussing and improving the Work, but excluding communication that is conspicuously marked or otherwise designated in writing by You as "Not a Contribution."
+
+2. Grant of Copyright License. Subject to the terms and conditions of this Agreement, You hereby grant to Doximity and to recipients of software distributed by Doximity a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable copyright license to reproduce, prepare derivative works of, publicly display, publicly perform, sublicense, and distribute Your Contributions and such derivative works.
+
+3. Grant of Patent License. Subject to the terms and conditions of this Agreement, You hereby grant to Doximity and to recipients of software distributed by Doximity a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable (except as stated in this section) patent license to make, have made, use, offer to sell, sell, import, and otherwise transfer the Work, where such license applies only to those patent claims licensable by You that are necessarily infringed by Your Contribution(s) alone or by combination of Your Contribution(s) with the Work to which such Contribution(s) was submitted. If any entity institutes patent litigation against You or any other entity (including a cross-claim or counterclaim in a lawsuit) alleging that your Contribution, or the Work to which you have contributed, constitutes direct or contributory patent infringement, then any patent licenses granted to that entity under this Agreement for that Contribution or Work shall terminate as of the date such litigation is filed.
+
+4. You represent that you are legally entitled to grant the above license. If your employer(s) has rights to intellectual property that you create that includes your Contributions, you represent that you have received permission to make Contributions on behalf of that employer, that your employer has waived such rights for your Contributions to Doximity, or that your employer has executed a separate Corporate CLA with Doximity.
+
+5. You represent that each of Your Contributions is Your original creation (see section 7 for submissions on behalf of others). You represent that Your Contribution submissions include complete details of any third-party license or other restriction (including, but not limited to, related patents and trademarks) of which you are personally aware and which are associated with any part of Your Contributions.
+
+6. You are not expected to provide support for Your Contributions, except to the extent You desire to provide support. You may provide support for free, for a fee, or not at all. Unless required by applicable law or agreed to in writing, You provide Your Contributions on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied, including, without limitation, any warranties or conditions of TITLE, NON- INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A PARTICULAR PURPOSE.
+
+7. Should You wish to submit work that is not Your original creation, You may submit it to Doximity separately from any Contribution, identifying the complete details of its source and of any license or other restriction (including, but not limited to, related patents, trademarks, and license agreements) of which you are personally aware, and conspicuously marking the work as "Submitted on behalf of a third-party: [[]named here]".
+
+8. You agree to notify Doximity of any facts or circumstances of which you become aware that would make these representations inaccurate in any respect.

data/CONTRIBUTORS.md

@@ -0,0 +1,22 @@
+## List of All Known Code Contributors to Simplekiq
+
+### Jack Noble
+* Collaborated on initial concept
+* Wrote the majority of the code as of initial release
+
+### John Wilkinson
+* Collaborated on initial concept
+* Conducted the gem extraction and release
+
+### Brian Dillard
+* Added additional comment documentation
+* Added support for `on_complete` batch callback support in `Simplekiq::BatchingJob`
+
+### Austin Madden
+* Fixed bug with batch statuses in callbacks for empty batches
+
+### Tiffany Troha
+* Added support for specifying `sidekiq_options` for the child job in `Simplekiq::BatchingJob`
+
+### [Daniel Pepper](https://github.com/dpep)
+* On request, graciously took down his unused `simplekiq` placeholder from rubygems so we could continue using the name :raised_hands:

data/Gemfile

@@ -0,0 +1,4 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in simplekiq.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,81 @@
+PATH
+  remote: .
+  specs:
+    simplekiq (0.0.3)
+      sidekiq (~> 5.2.9)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ast (2.4.2)
+    coderay (1.1.3)
+    connection_pool (2.2.5)
+    diff-lcs (1.5.0)
+    method_source (1.0.0)
+    parallel (1.22.1)
+    parser (3.1.1.0)
+      ast (~> 2.4.1)
+    pry (0.13.1)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    rack (2.2.3)
+    rack-protection (2.2.0)
+      rack
+    rainbow (3.1.1)
+    rake (12.3.3)
+    redis (4.5.1)
+    regexp_parser (2.2.1)
+    rexml (3.2.5)
+    rspec (3.10.0)
+      rspec-core (~> 3.10.0)
+      rspec-expectations (~> 3.10.0)
+      rspec-mocks (~> 3.10.0)
+    rspec-core (3.10.1)
+      rspec-support (~> 3.10.0)
+    rspec-expectations (3.10.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.10.0)
+    rspec-mocks (3.10.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.10.0)
+    rspec-support (3.10.3)
+    rspec_junit_formatter (0.5.1)
+      rspec-core (>= 2, < 4, != 2.12.0)
+    rubocop (1.26.1)
+      parallel (~> 1.10)
+      parser (>= 3.1.0.0)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8, < 3.0)
+      rexml
+      rubocop-ast (>= 1.16.0, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 1.4.0, < 3.0)
+    rubocop-ast (1.16.0)
+      parser (>= 3.1.1.0)
+    rubocop-performance (1.13.3)
+      rubocop (>= 1.7.0, < 2.0)
+      rubocop-ast (>= 0.4.0)
+    ruby-progressbar (1.11.0)
+    sidekiq (5.2.10)
+      connection_pool (~> 2.2, >= 2.2.2)
+      rack (~> 2.0)
+      rack-protection (>= 1.5.0)
+      redis (~> 4.5, < 4.6.0)
+    standard (1.9.1)
+      rubocop (= 1.26.1)
+      rubocop-performance (= 1.13.3)
+    unicode-display_width (2.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  pry
+  rake (~> 12.0)
+  rspec (~> 3.2)
+  rspec_junit_formatter
+  simplekiq!
+  standard
+
+BUNDLED WITH
+   2.1.4

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright 2022 Doximity, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

data/README.md

@@ -0,0 +1,131 @@
+# Simplekiq
+
+Any time that you find yourself needing to string together a long chain of jobs, particularly when there are multiple stages of Sidekiq-pro batches and callbacks involved, come home instead to the simple flavor of orchestrated job flow with Simplekiq.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem "simplekiq"
+```
+
+Note that this gem requires you be a Sidekiq Pro paid subscriber to be able to use it, so after following the installation docs for getting the private gem configured with your system, ensure you have `sidekiq-pro` at version `~> 5.0.0` or higher and that it's being required:
+
+```ruby
+gem "sidekiq-pro", "~> 5.0.0"
+```
+
+And then execute:
+
+    $ bundle install
+
+Or install it yourself as:
+
+    $ gem install simplekiq
+
+## Usage
+
+There are currently two primary components of the system which were designed to work in harmony:
+
+* [Simplekiq::OrchestrationJob](./lib/simplekiq/orchestration_job.rb) - A mixin for a Sidekiq jobs to be able to orchestrate a flow of jobs in one place. It makes long complicated flows between jobs easier to understand, iterate on, and test. It eliminates the need to hop between dozens of files to determine when, where, and why a particular job gets called.
+* [Simplekiq::BatchingJob](./lib/simplekiq/batching_job.rb) - A mixin designed to make breaking a large job into a batched process dead simple and contained within a single class while still being trivially composable in orchestrations.
+
+## Tool Drilldown
+
+### Simplekiq::OrchestrationJob
+
+Mixing in the [Simplekiq::Orchestration](./lib/simplekiq/orchestration_job.rb) module lets you define a human-readable workflow of jobs in a single file with almost* no special requirements or restrictions on how the child jobs are designed. In most cases, Sidekiq jobs not designed for use in orchestrations should be compatible for use in orchestrations. A job implementing `OrchestrationJob` might look like:
+
+```ruby
+class SomeOrchestrationJob < BaseJob
+  include Sidekiq::Worker
+  include Simplekiq::OrchestrationJob
+
+  def perform_orchestration(some_id)
+    @some_model = SomeModel.find(some_id) # 1.
+
+    run SomeInitialSetupJob, some_model.id # 2.
+
+    in_parallel do
+      some_related_models.each do |related_model|
+        run SomeParallelizableJob, related_model.id # 3.
+      end
+    end
+
+    run SomeFinalizationJob, some_model.id # 4.
+  end
+
+  private
+
+  attr_reader :some_model
+
+  def some_related_models
+    @some_related_models ||= some_model.some_relation
+  end
+end
+```
+
+Let's use the above example to describe some specifics of how the flow works.
+
+1. `SomeOrchestrationJob` pulls up some instance of parent model `SomeModel`.
+2. It does some initial work in `SomeInitialSetupJob`, which blocks the rest of the workflow until it completes successfully.
+3. Then it will run a `SomeParallelizableJob` for each of some number of associated models `some_related_models`. These jobs will all run parallel to each other independently.
+4. Finally, after all of the parallel jobs from #3 complete successfully, `SomeFinalizationJob` will run and then after it finishes the orchestration will be complete.
+
+**Note** - it's fine to add utility methods and `attr_accessor`s to keep the code tidy and maintainable.
+
+When `SomeOrchestrationJob` itself gets called though, the first thing it does it turn these directives into a big serialized structure indicating which job will be called under what conditions (eg, serial or in parallel) and with what arguments, and then keeps passing that between the simplekiq-internal jobs that actually conduct the flow.
+
+This means when you want to deploy a change to this flow all previous in-flight workflows will continue undisturbed because the workflow is frozen in sidekiq job arguments and will remain frozen until the workflow completes. This is generally a boon, but note that if you remove a job from a workflow you'll need to remember to either keep the job itself (eg, the `SomeFinalizationJob` class file from our above example) in the codebase or replace it with a stub so that any in-flight workflows won't crash due to not being able to pull up the prior-specified workflow.
+
+"almost* no special requirements or restrictions on how the child jobs are designed" - The one thing you'll want to keep in mind when feeding arbitrary jobs into orchestrations is that if the job creates any new sidekiq batches then those new sidekiq batches should be added as child sidekiq batches of the parent sidekiq batch of the job. The parent sidekiq batch of the job is the sidekiq batch that drives the orchestration from step to step, so if you don't do this it will move onto the next step in the orchestration once your job finishes even if the new sidekiq batches it started didn't finish. This sounds more complicated than it is, you can see an example of code that does this in [`BatchingJob#perform`](./lib/simplekiq/batching_job.rb):
+
+```ruby
+if batch # is there a parent batch?
+  batch.jobs do # open the parent batch back up
+    create_a_new_batch_and_add_jobs_to_it_to_run # make our new batch as a child batch of the parent batch
+  end # close the parent batch again
+else # there's no parent batches, this job was run directly outside of an orchestration
+  create_a_new_batch_and_add_jobs_to_it_to_run # make our new batch without a parent batch
+end
+```
+
+### Simplekiq::BatchingJob
+
+See the [Simplekiq::BatchingJob](./lib/simplekiq/batching_job.rb) module itself for a description and example usage in the header comments. Nutshell is that you should use this if you're planning on making a batched asynchronous process as it shaves off a lot of ceremony and unexpressive structure. eg - Instead of having `BeerBottlerJob` which queues some number of `BeerBottlerBatchJob`s to handle the broken down sub-tasks you can just have `BeerBottlerJob` with a method for batching, executing individual batches, and a callback that gets run after all batches have completed successfully.
+
+## History
+
+Simplekiq was initially released for private use within Doximity applications in Oct 2020 where it continued to be iterated on towards stability and general use until Jan 2022 when it was deemed settled enough for public release.
+
+The primary driving factor that inspired this work was a series of over a dozen differently defined and structured jobs part of a single workflow of which the logical flow was extraordinarily difficult to cognitively trace. This led to exteme difficulty in debugging and following problematic instances of the workflow in production as well as needlessly high cost to refactoring and iterative adjustments.
+
+The crux of the problem was that each job was highly coupled to its position in the overall flow as well as the absence of any central mechanism to indicate what the overall flow was. After building Simplekiq and implementing it into the flow, significant changes to the flow became quick adjustments requiring only a couple lines of code to change and folks unfamiliar with the system could quickly get up to speed by reading through the orchestration job.
+
+## Versioning
+
+This project follows semantic versioning. At time of writing it is sitting at 0.0.1 until its integration with the application it was extracted from is confirmed to be stable. Once confirmed it will be started off at 1.0.0 as it has otherwise been used in a production system already for some time.
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Note that this depends on `sidekiq-pro` which requires a [commercial license](https://sidekiq.org/products/pro.html) to install and use.
+
+Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+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 tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+TODO: Update this section with more specific/appropriate instructions once this is a public repository.
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request
+6. Sign the CLA if you haven't yet. See CONTRIBUTING.md
+
+## License
+
+The gem is licensed under an Apache 2 license. Contributors are required to sign an contributor license agreement. See LICENSE.txt and CONTRIBUTING.md for more information.

data/Rakefile

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+require "rspec/core/rake_task"
+
+FileList["tasks/*.rake"].each { |task| load task }
+
+RSpec::Core::RakeTask.new(:spec)
+
+task default: :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "simplekiq"
+
+# 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/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/simplekiq/batching_job.rb

@@ -0,0 +1,163 @@
+# frozen_string_literal: true
+
+# This module enables you to break up your work into batches and run those
+# batches as background jobs while keeping all your code in the same file.
+# Including this module *implements perform* you should not override it.
+# It expects you to implement two methods: #perform_batching and
+# #perform_batch.
+#
+# Optionally you may also implement any combination of Sidekiq::Batch
+# callbacks.
+#   - #on_complete
+#   - #on_success
+#   - #on_death
+#
+# #perform_batching should contain your code for breaking up your work into
+# smaller jobs. It handles all the Sidekiq::Batch boilerplate for you. Where
+# you would normally call ExampleBatchJob.perform_async you should use
+# #queue_batch. If you'd like to custommize the sidekiq batch object, you can
+# access it in perform_batching through the `sidekiq_batch` method.
+#
+# #perform_batch should contain the code that would be in your batch job. Under
+# the hood, #queue_batch queues a job which will run #perform_batch.
+#
+# [Sidekiq::Batch documentation](https://github.com/mperham/sidekiq/wiki/Batches)
+# explains batches, their lifecycle, callbacks, etc.
+#
+# class ExampleJob
+#   include Simplekiq::BatchingJob
+#
+#   def perform_batching(some_id)
+#     sidekiq_batch.description = "My custom batch description" # optional
+#
+#     Record.find(some_id).other_records.in_batches do |other_records|
+#       queue_batch(other_records.ids)
+#     end
+#   end
+#
+#   def perform_batch(other_record_ids)
+#     OtherRecord.where(id: other_record_ids).do_work
+#   end
+#
+#   def on_death(_status, options)
+#     same_id_as_before = options["args"].first
+#     Record.find(same_id_as_before).death!
+#   end
+
+#   def on_complete(_status, options)
+#     same_id_as_before = options["args"].first
+#     Record.find(same_id_as_before).complete!
+#   end
+
+#   def on_success(_status, options)
+#     same_id_as_before = options["args"].first
+#     Record.find(same_id_as_before).success!
+#   end
+# end
+#
+# ExampleJob.perform_async(some_id)
+#
+# Come home to the impossible flavor of batch creation
+
+module Simplekiq
+  module BatchingJob
+    include Sidekiq::Worker
+
+    BATCH_CLASS_NAME = "SimplekiqBatch"
+
+    class << self
+      def included(klass)
+        batch_job_class = Class.new(BaseBatch)
+        klass.const_set(BATCH_CLASS_NAME, batch_job_class)
+
+        klass.extend ClassMethods
+      end
+    end
+
+    module ClassMethods
+      def batch_sidekiq_options(options)
+        batch_class = const_get(BATCH_CLASS_NAME)
+        batch_class.instance_eval do
+          sidekiq_options(options)
+        end
+      end
+    end
+
+    def perform(*args)
+      self.batches = []
+
+      perform_batching(*args)
+
+      # If we're part of an existing sidekiq batch make this a child batch
+      # This is necessary for it work with orchestration; we could add an option
+      # to toggle the behavior on and off.
+      if batch
+        batch.jobs do
+          handle_batches(args)
+        end
+      else
+        handle_batches(args)
+      end
+    end
+
+    protected # TODO: should this be private?
+
+    attr_accessor :batches
+
+    def handle_batches(args)
+      if !batches.empty?
+        flush_batches(args)
+      else
+        # Empty batches with no jobs will never invoke callbacks, so handle
+        # that case by immediately manually invoking :complete & :success.
+        on_complete(nil, {"args" => args}) if respond_to?(:on_complete)
+        on_success(nil, {"args" => args}) if respond_to?(:on_success)
+      end
+    end
+
+    def flush_batches(args)
+      batch_job_class = self.class.const_get(BATCH_CLASS_NAME)
+      sidekiq_batch.description ||= "Simplekiq Batch Jobs for #{self.class.name}, args: #{args}"
+
+      sidekiq_batch.on("death", self.class, "args" => args) if respond_to?(:on_death)
+      sidekiq_batch.on("complete", self.class, "args" => args) if respond_to?(:on_complete)
+      sidekiq_batch.on("success", self.class, "args" => args) if respond_to?(:on_success)
+
+      sidekiq_batch.jobs do
+        batches.each do |job_args|
+          batch_job_class.perform_async(*job_args)
+        end
+      end
+    end
+
+    def queue_batch(*args)
+      batches << args
+    end
+
+    def batch_description=(description)
+      sidekiq_batch.description = description
+    end
+
+    private
+
+    def sidekiq_batch
+      @sidekiq_batch ||= Sidekiq::Batch.new
+    end
+  end
+
+  class BaseBatch
+    include Sidekiq::Worker
+
+    def perform(*args)
+      module_parent_of_class.new.perform_batch(*args)
+    end
+
+    private
+
+    def module_parent_of_class
+      # Borrowed from https://apidock.com/rails/Module/module_parent_name
+      parent_name = self.class.name =~ /::[^:]+\Z/ ? $`.freeze : nil
+      parent_name ? Object.const_get(parent_name) : Object
+    end
+  end
+end

data/lib/simplekiq/orchestration.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module Simplekiq
+  class Orchestration
+    attr_accessor :serial_workflow, :parallel_workflow
+    def initialize
+      @serial_workflow = []
+    end
+
+    def run(*step)
+      workflow = parallel_workflow || serial_workflow
+      workflow << step
+    end
+
+    def in_parallel
+      @parallel_workflow = []
+      yield
+      serial_workflow << @parallel_workflow if @parallel_workflow.any?
+    ensure
+      @parallel_workflow = nil
+      serial_workflow
+    end
+
+    def execute(parent_batch)
+      OrchestrationExecutor.execute(workflow: serialized_workflow, parent_batch: parent_batch)
+    end
+
+    def serialized_workflow
+      @serialized_workflow ||= serial_workflow.map do |step|
+        case step[0]
+        when Array
+          step.map do |(job, *args)|
+            {"klass" => job.name, "args" => args}
+          end
+        when Class
+          job, *args = step
+          {"klass" => job.name, "args" => args}
+        end
+      end
+    end
+  end
+end

data/lib/simplekiq/orchestration_executor.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module Simplekiq
+  class OrchestrationExecutor
+    def self.execute(workflow:, parent_batch:)
+      new.run_step(parent_batch, workflow, 0)
+    end
+
+    def run_step(parent_batch, workflow, step)
+      return if workflow.empty?
+
+      nest_under(parent_batch) do
+        *jobs = workflow.at(step)
+        sidekiq_batch = Sidekiq::Batch.new
+        sidekiq_batch.on(
+          :success,
+          self.class,
+          "orchestration_workflow" => workflow, "step" => step + 1
+        )
+
+        sidekiq_batch.jobs do
+          jobs.each do |job|
+            Object.const_get(job["klass"]).perform_async(*job["args"])
+          end
+        end
+      end
+    end
+
+    def nest_under(parent_batch)
+      if parent_batch
+        parent_batch.jobs do
+          yield
+        end
+      else
+        yield
+      end
+    end
+
+    def on_success(status, options)
+      return if options["step"] == options["orchestration_workflow"].length
+
+      parent_batch = Sidekiq::Batch.new(status.parent_bid)
+      run_step(parent_batch, options["orchestration_workflow"], options["step"])
+    end
+  end
+end

data/lib/simplekiq/orchestration_job.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require "forwardable"
+
+module Simplekiq
+  module OrchestrationJob
+    include Sidekiq::Worker
+
+    extend Forwardable
+    def_delegators :orchestration, :run, :in_parallel
+
+    def perform(*args)
+      perform_orchestration(*args)
+      orchestration.execute(batch)
+    end
+
+    def workflow_plan(*args)
+      perform_orchestration(*args)
+      orchestration.serialized_workflow
+    end
+
+    private
+
+    def orchestration
+      @orchestration ||= Orchestration.new
+    end
+  end
+end

data/lib/simplekiq/version.rb

@@ -0,0 +1,3 @@
+module Simplekiq
+  VERSION = "0.0.3"
+end

data/lib/simplekiq.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require "sidekiq"
+
+# NB: You must explicitly require sidekiq-ent in your app!
+# require "sidekiq-ent"
+
+require "simplekiq/orchestration_executor"
+require "simplekiq/orchestration"
+require "simplekiq/orchestration_job"
+require "simplekiq/batching_job"
+
+module Simplekiq
+end

data/simplekiq.gemspec

@@ -0,0 +1,38 @@
+require_relative "lib/simplekiq/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "simplekiq"
+  spec.version = Simplekiq::VERSION
+  spec.authors = ["Jack Noble", "John Wilkinson"]
+  spec.email = ["jcwilkinson@doximity.com"]
+  spec.summary = "Sidekiq-based workflow orchestration library"
+  spec.description = "Provides tools for representing long chains of parallel and serial jobs in a flat, simple way."
+  spec.homepage = "https://github.com/doximity/simplekiq"
+  spec.license = "APACHE-2.0"
+  spec.required_ruby_version = Gem::Requirement.new(">= 2.5.0")
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = spec.homepage
+  # TODO: spec.metadata["changelog_uri"] = "TODO: Put your gem's CHANGELOG.md URL here."
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(File.expand_path("..", __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.executables = []
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "rspec", "~> 3.2"
+  spec.add_development_dependency "rake", "~> 12.0"
+  spec.add_development_dependency "rspec_junit_formatter"
+  spec.add_development_dependency "pry"
+  spec.add_development_dependency "standard"
+
+  spec.add_dependency "sidekiq", "~> 5.2.9"
+
+  # Can't define this explicitly because it would be inappropriate to vendor this
+  # pay-to-use library into the gem and it is not available on rubygems and
+  # this otherwise interferes with our publishing process.
+  # spec.add_dependency "sidekiq-pro", "~> 5.0.0"
+end

data/tasks/ci.rake

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+namespace :ci do
+  desc "Run specs"
+  task :specs do
+    reports = "tmp/test-results/rspec"
+    sh "mkdir -p #{reports}"
+    sh "bundle exec rspec ./spec " \
+          "--format progress "\
+          "--format RspecJunitFormatter " \
+          "-o #{reports}/results.xml"
+  end
+end

metadata

@@ -0,0 +1,180 @@
+--- !ruby/object:Gem::Specification
+name: simplekiq
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Jack Noble
+- John Wilkinson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2022-04-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: rspec_junit_formatter
+  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'
+- !ruby/object:Gem::Dependency
+  name: pry
+  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'
+- !ruby/object:Gem::Dependency
+  name: standard
+  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'
+- !ruby/object:Gem::Dependency
+  name: sidekiq
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.2.9
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.2.9
+description: Provides tools for representing long chains of parallel and serial jobs
+  in a flat, simple way.
+email:
+- jcwilkinson@doximity.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".circleci/config.yml"
+- ".gitignore"
+- ".rspec"
+- CHANGELOG.md
+- CONTRIBUTING.md
+- CONTRIBUTORS.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/simplekiq.rb
+- lib/simplekiq/batching_job.rb
+- lib/simplekiq/orchestration.rb
+- lib/simplekiq/orchestration_executor.rb
+- lib/simplekiq/orchestration_job.rb
+- lib/simplekiq/version.rb
+- simplekiq.gemspec
+- tasks/ci.rake
+- vendor/cache/ast-2.4.2.gem
+- vendor/cache/coderay-1.1.3.gem
+- vendor/cache/connection_pool-2.2.5.gem
+- vendor/cache/diff-lcs-1.5.0.gem
+- vendor/cache/method_source-1.0.0.gem
+- vendor/cache/parallel-1.22.1.gem
+- vendor/cache/parser-3.1.1.0.gem
+- vendor/cache/pry-0.13.1.gem
+- vendor/cache/rack-2.2.3.gem
+- vendor/cache/rack-protection-2.2.0.gem
+- vendor/cache/rainbow-3.1.1.gem
+- vendor/cache/rake-12.3.3.gem
+- vendor/cache/redis-4.5.1.gem
+- vendor/cache/regexp_parser-2.2.1.gem
+- vendor/cache/rexml-3.2.5.gem
+- vendor/cache/rspec-3.10.0.gem
+- vendor/cache/rspec-core-3.10.1.gem
+- vendor/cache/rspec-expectations-3.10.1.gem
+- vendor/cache/rspec-mocks-3.10.2.gem
+- vendor/cache/rspec-support-3.10.3.gem
+- vendor/cache/rspec_junit_formatter-0.5.1.gem
+- vendor/cache/rubocop-1.26.1.gem
+- vendor/cache/rubocop-ast-1.16.0.gem
+- vendor/cache/rubocop-performance-1.13.3.gem
+- vendor/cache/ruby-progressbar-1.11.0.gem
+- vendor/cache/sidekiq-5.2.10.gem
+- vendor/cache/standard-1.9.1.gem
+- vendor/cache/unicode-display_width-2.1.0.gem
+homepage: https://github.com/doximity/simplekiq
+licenses:
+- APACHE-2.0
+metadata:
+  homepage_uri: https://github.com/doximity/simplekiq
+  source_code_uri: https://github.com/doximity/simplekiq
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.5.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.11
+signing_key: 
+specification_version: 4
+summary: Sidekiq-based workflow orchestration library
+test_files: []