dispatch_policy 0.2.0 → 0.4.0

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dispatch_policy
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.4.0
 platform: ruby
 authors:
 - José Galisteo
@@ -10,7 +10,7 @@ cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: activejob
+  name: rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -24,77 +24,119 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '7.1'
 - !ruby/object:Gem::Dependency
-  name: activerecord
+  name: pg
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.1'
+        version: '1.4'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.1'
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.20'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.20'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
 - !ruby/object:Gem::Dependency
-  name: railties
+  name: good_job
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.1'
-  type: :runtime
+        version: '4.0'
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.1'
+        version: '4.0'
 - !ruby/object:Gem::Dependency
-  name: pg
+  name: solid_queue
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
 - !ruby/object:Gem::Dependency
-  name: minitest
+  name: turbo-rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.5'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.5'
 - !ruby/object:Gem::Dependency
-  name: simplecov
+  name: puma
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '6.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '6.0'
 - !ruby/object:Gem::Dependency
-  name: sprockets-rails
+  name: foreman
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.87'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.87'
+- !ruby/object:Gem::Dependency
+  name: debug
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -107,10 +149,38 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: DispatchPolicy stages ActiveJob enqueues into a policy table and admits
-  them through declared gates. Supports per-partition throttle/concurrency, dedupe,
-  round-robin fairness, and ships a minimal Rails engine to inspect pending/admitted
-  state.
+- !ruby/object:Gem::Dependency
+  name: capybara
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.40'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.40'
+- !ruby/object:Gem::Dependency
+  name: selenium-webdriver
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.20'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.20'
+description: Stages perform_later into a dedicated table, runs a tick loop that admits
+  jobs through declared gates (throttle, concurrency), then forwards survivors to
+  the real ActiveJob adapter. Embedded as a periodic job. Compatible with good_job
+  and solid_queue.
 email:
 - ceritium@gmail.com
 executables: []
@@ -120,47 +190,70 @@ files:
 - CHANGELOG.md
 - MIT-LICENSE
 - README.md
+- app/assets/images/dispatch_policy/logo-large.svg
+- app/assets/images/dispatch_policy/logo-small.svg
+- app/assets/javascripts/dispatch_policy/turbo.es2017-umd.min.js
+- app/assets/stylesheets/dispatch_policy/application.css
 - app/controllers/dispatch_policy/application_controller.rb
+- app/controllers/dispatch_policy/assets_controller.rb
+- app/controllers/dispatch_policy/dashboard_controller.rb
+- app/controllers/dispatch_policy/partitions_controller.rb
 - app/controllers/dispatch_policy/policies_controller.rb
+- app/controllers/dispatch_policy/staged_jobs_controller.rb
 - app/models/dispatch_policy/adaptive_concurrency_stats.rb
 - app/models/dispatch_policy/application_record.rb
-- app/models/dispatch_policy/partition_inflight_count.rb
-- app/models/dispatch_policy/partition_observation.rb
+- app/models/dispatch_policy/inflight_job.rb
+- app/models/dispatch_policy/partition.rb
 - app/models/dispatch_policy/staged_job.rb
-- app/models/dispatch_policy/throttle_bucket.rb
+- app/models/dispatch_policy/tick_sample.rb
+- app/views/dispatch_policy/dashboard/index.html.erb
+- app/views/dispatch_policy/partitions/index.html.erb
+- app/views/dispatch_policy/partitions/show.html.erb
 - app/views/dispatch_policy/policies/index.html.erb
 - app/views/dispatch_policy/policies/show.html.erb
+- app/views/dispatch_policy/shared/_capacity.html.erb
+- app/views/dispatch_policy/shared/_hints.html.erb
+- app/views/dispatch_policy/shared/_partition_row.html.erb
+- app/views/dispatch_policy/staged_jobs/show.html.erb
 - app/views/layouts/dispatch_policy/application.html.erb
 - config/routes.rb
-- db/migrate/20260424000001_create_dispatch_policy_tables.rb
-- db/migrate/20260424000002_create_adaptive_concurrency_stats.rb
-- db/migrate/20260424000003_create_adaptive_concurrency_samples.rb
-- db/migrate/20260424000004_rename_samples_to_partition_observations.rb
-- db/migrate/20260425000001_add_duration_to_partition_observations.rb
+- db/migrate/20260501000001_create_dispatch_policy_tables.rb
 - lib/dispatch_policy.rb
-- lib/dispatch_policy/active_job_perform_all_later_patch.rb
-- lib/dispatch_policy/dispatch_context.rb
-- lib/dispatch_policy/dispatchable.rb
+- lib/dispatch_policy/assets.rb
+- lib/dispatch_policy/bypass.rb
+- lib/dispatch_policy/config.rb
+- lib/dispatch_policy/context.rb
+- lib/dispatch_policy/cursor_pagination.rb
+- lib/dispatch_policy/decision.rb
 - lib/dispatch_policy/engine.rb
+- lib/dispatch_policy/forwarder.rb
 - lib/dispatch_policy/gate.rb
 - lib/dispatch_policy/gates/adaptive_concurrency.rb
 - lib/dispatch_policy/gates/concurrency.rb
-- lib/dispatch_policy/gates/fair_interleave.rb
-- lib/dispatch_policy/gates/global_cap.rb
 - lib/dispatch_policy/gates/throttle.rb
+- lib/dispatch_policy/inflight_tracker.rb
+- lib/dispatch_policy/job_extension.rb
+- lib/dispatch_policy/operator_hints.rb
+- lib/dispatch_policy/pipeline.rb
 - lib/dispatch_policy/policy.rb
+- lib/dispatch_policy/policy_dsl.rb
+- lib/dispatch_policy/railtie.rb
+- lib/dispatch_policy/registry.rb
+- lib/dispatch_policy/repository.rb
+- lib/dispatch_policy/serializer.rb
 - lib/dispatch_policy/tick.rb
 - lib/dispatch_policy/tick_loop.rb
 - lib/dispatch_policy/version.rb
+- lib/generators/dispatch_policy/install/install_generator.rb
+- lib/generators/dispatch_policy/install/templates/create_dispatch_policy_tables.rb.tt
+- lib/generators/dispatch_policy/install/templates/dispatch_tick_loop_job.rb.tt
+- lib/generators/dispatch_policy/install/templates/initializer.rb.tt
 homepage: https://github.com/ceritium/dispatch_policy
 licenses:
 - MIT
 metadata:
   homepage_uri: https://github.com/ceritium/dispatch_policy
   source_code_uri: https://github.com/ceritium/dispatch_policy
-  bug_tracker_uri: https://github.com/ceritium/dispatch_policy/issues
-  changelog_uri: https://github.com/ceritium/dispatch_policy/blob/master/CHANGELOG.md
-  rubygems_mfa_required: 'true'
 rdoc_options: []
 require_paths:
 - lib
@@ -168,7 +261,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.1'
+      version: 3.1.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -177,6 +270,5 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 4.0.3
 specification_version: 4
-summary: Per-partition admission control (throttle, concurrency, dedupe, fairness)
-  for ActiveJob.
+summary: Per-partition admission control for ActiveJob (Postgres).
 test_files: []

data/app/models/dispatch_policy/partition_inflight_count.rb

@@ -1,42 +0,0 @@
-# frozen_string_literal: true
-
-module DispatchPolicy
-  class PartitionInflightCount < ApplicationRecord
-    self.table_name = "dispatch_policy_partition_counts"
-
-    def self.fetch_many(policy_name:, gate_name:, partition_keys:)
-      return {} if partition_keys.empty?
-
-      where(policy_name: policy_name, gate_name: gate_name.to_s, partition_key: partition_keys)
-        .pluck(:partition_key, :in_flight).to_h
-        .tap { |h| partition_keys.each { |k| h[k] ||= 0 } }
-    end
-
-    def self.total_for(policy_name:, gate_name:)
-      where(policy_name: policy_name, gate_name: gate_name.to_s).sum(:in_flight)
-    end
-
-    def self.increment(policy_name:, gate_name:, partition_key:, by: 1)
-      now = Time.current
-      sql = <<~SQL.squish
-        INSERT INTO #{quoted_table_name}
-          (policy_name, gate_name, partition_key, in_flight, created_at, updated_at)
-        VALUES (?, ?, ?, ?, ?, ?)
-        ON CONFLICT (policy_name, gate_name, partition_key)
-        DO UPDATE SET
-          in_flight  = #{quoted_table_name}.in_flight + EXCLUDED.in_flight,
-          updated_at = EXCLUDED.updated_at
-      SQL
-      connection.exec_update(
-        sanitize_sql_array([ sql, policy_name, gate_name.to_s, partition_key.to_s, by, now, now ])
-      )
-    end
-
-    def self.decrement(policy_name:, gate_name:, partition_key:, by: 1)
-      where(policy_name: policy_name, gate_name: gate_name.to_s, partition_key: partition_key.to_s)
-        .update_all([
-          "in_flight = GREATEST(in_flight - ?, 0), updated_at = ?", by, Time.current
-        ])
-    end
-  end
-end

data/app/models/dispatch_policy/partition_observation.rb

@@ -1,76 +0,0 @@
-# frozen_string_literal: true
-
-module DispatchPolicy
-  # Minute-bucketed observability per (policy, partition). Any gate with
-  # partition_by gets an observation row here — adaptive, throttle,
-  # concurrency, whatever — so the admin chart shows queue lag / throughput
-  # for all partitioned policies, not just the adaptive ones.
-  #
-  # One row per (policy, partition, minute): total_lag_ms accumulates the
-  # sum of queue_lag_ms observations in that minute, total_duration_ms
-  # accumulates perform durations (used by :time_budget and :fair_time_share),
-  # observation_count increments, max_lag_ms / max_duration_ms track worst
-  # spikes. Averages are derived on read as total / count.
-  class PartitionObservation < ApplicationRecord
-    self.table_name = "dispatch_policy_partition_observations"
-
-    OBSERVATION_TTL = 2 * 60 * 60  # 2 hours
-
-    def self.observe!(policy_name:, partition_key:, queue_lag_ms:, duration_ms: 0, current_max: nil)
-      return if partition_key.nil? || partition_key.to_s.empty?
-
-      now = Time.current
-      lag = queue_lag_ms.to_i
-      dur = duration_ms.to_i
-      sql = <<~SQL.squish
-        INSERT INTO #{quoted_table_name}
-          (policy_name, partition_key, minute_bucket,
-           total_lag_ms, total_duration_ms, observation_count,
-           max_lag_ms, max_duration_ms, current_max,
-           created_at, updated_at)
-        VALUES (?, ?, date_trunc('minute', ?::timestamp), ?, ?, 1, ?, ?, ?, ?, ?)
-        ON CONFLICT (policy_name, partition_key, minute_bucket)
-        DO UPDATE SET
-          total_lag_ms      = #{quoted_table_name}.total_lag_ms + EXCLUDED.total_lag_ms,
-          total_duration_ms = #{quoted_table_name}.total_duration_ms + EXCLUDED.total_duration_ms,
-          observation_count = #{quoted_table_name}.observation_count + 1,
-          max_lag_ms        = GREATEST(#{quoted_table_name}.max_lag_ms, EXCLUDED.max_lag_ms),
-          max_duration_ms   = GREATEST(#{quoted_table_name}.max_duration_ms, EXCLUDED.max_duration_ms),
-          current_max       = COALESCE(EXCLUDED.current_max, #{quoted_table_name}.current_max),
-          updated_at        = EXCLUDED.updated_at
-      SQL
-      connection.exec_update(
-        sanitize_sql_array([
-          sql, policy_name, partition_key.to_s, now,
-          lag, dur, lag, dur, current_max, now, now
-        ])
-      )
-    end
-
-    # Sum of perform durations per partition over the last `window` seconds.
-    # Used by :fair_time_share to bias admission ordering toward partitions
-    # that have consumed less compute time recently.
-    def self.consumed_ms_by_partition(policy_name:, partition_keys:, window:)
-      return {} if partition_keys.empty?
-
-      # minute_bucket is floored on insert (date_trunc('minute', now)).
-      # An observation written T seconds ago lives in a bucket up to 60s
-      # earlier than T. Add a one-bucket pad to the lower bound so the
-      # most recent bucket is always inside the window — without it, the
-      # previous-minute bucket is silently excluded as soon as the wall
-      # clock crosses a minute boundary.
-      since = Time.current - window - 60
-      rows = where(policy_name: policy_name, partition_key: partition_keys.map(&:to_s))
-        .where("minute_bucket >= ?", since)
-        .group(:partition_key)
-        .pluck(Arel.sql("partition_key, SUM(total_duration_ms), SUM(observation_count)"))
-      rows.each_with_object({}) do |(key, total, count), acc|
-        acc[key] = { consumed_ms: total.to_i, count: count.to_i }
-      end
-    end
-
-    def self.prune!
-      where("minute_bucket < ?", Time.current - OBSERVATION_TTL).delete_all
-    end
-  end
-end

data/app/models/dispatch_policy/throttle_bucket.rb

@@ -1,41 +0,0 @@
-# frozen_string_literal: true
-
-module DispatchPolicy
-  class ThrottleBucket < ApplicationRecord
-    self.table_name = "dispatch_policy_throttle_buckets"
-
-    def self.lock(policy_name:, gate_name:, partition_key:, burst:)
-      now = Time.current
-      seed_sql = <<~SQL.squish
-        INSERT INTO #{quoted_table_name}
-          (policy_name, gate_name, partition_key, tokens, refilled_at, created_at, updated_at)
-        VALUES (?, ?, ?, ?, ?, ?, ?)
-        ON CONFLICT (policy_name, gate_name, partition_key) DO NOTHING
-      SQL
-      connection.exec_update(
-        sanitize_sql_array([
-          seed_sql, policy_name, gate_name.to_s, partition_key.to_s,
-          burst.to_f, now, now, now
-        ])
-      )
-
-      where(policy_name: policy_name, gate_name: gate_name.to_s, partition_key: partition_key.to_s)
-        .lock("FOR UPDATE")
-        .first!
-    end
-
-    def refill!(rate:, per:, burst:)
-      now = Time.current
-      elapsed = (now - refilled_at).to_f
-      new_tokens = tokens + (rate * elapsed / per)
-      self.tokens = [ new_tokens, burst.to_f ].min
-      self.refilled_at = now
-    end
-
-    def consume(n = 1)
-      return false if tokens < n
-      self.tokens -= n
-      true
-    end
-  end
-end

data/db/migrate/20260424000001_create_dispatch_policy_tables.rb

@@ -1,80 +0,0 @@
-# frozen_string_literal: true
-
-class CreateDispatchPolicyTables < ActiveRecord::Migration[7.1]
-  def change
-    create_table :dispatch_policy_staged_jobs do |t|
-      t.string   :job_class,       null: false
-      t.string   :policy_name,     null: false
-      t.jsonb    :arguments,       null: false
-      t.jsonb    :snapshot,        null: false, default: {}
-      t.jsonb    :context,         null: false, default: {}
-      t.integer  :priority,        null: false, default: 100
-      t.datetime :not_before_at
-      t.datetime :staged_at,       null: false
-      t.datetime :admitted_at
-      t.datetime :completed_at
-      t.datetime :lease_expires_at
-      t.string   :active_job_id
-      t.string   :dedupe_key
-      t.string   :round_robin_key
-      t.jsonb    :partitions,      null: false, default: {}
-
-      t.timestamps
-    end
-
-    add_index :dispatch_policy_staged_jobs,
-      %i[policy_name priority staged_at],
-      where: "admitted_at IS NULL",
-      name: "idx_dp_staged_dispatch_order"
-
-    add_index :dispatch_policy_staged_jobs,
-      %i[policy_name dedupe_key],
-      unique: true,
-      where: "dedupe_key IS NOT NULL AND completed_at IS NULL",
-      name: "idx_dp_staged_dedupe_active"
-
-    add_index :dispatch_policy_staged_jobs,
-      %i[lease_expires_at],
-      where: "admitted_at IS NOT NULL",
-      name: "idx_dp_staged_lease_expires"
-
-    add_index :dispatch_policy_staged_jobs,
-      %i[completed_at],
-      where: "completed_at IS NOT NULL",
-      name: "idx_dp_staged_completed_at"
-
-    add_index :dispatch_policy_staged_jobs,
-      %i[policy_name round_robin_key priority staged_at],
-      where: "admitted_at IS NULL AND round_robin_key IS NOT NULL",
-      name: "idx_dp_staged_round_robin"
-
-    create_table :dispatch_policy_partition_counts do |t|
-      t.string  :policy_name,   null: false
-      t.string  :gate_name,     null: false
-      t.string  :partition_key, null: false, default: "default"
-      t.integer :in_flight,     null: false, default: 0
-
-      t.timestamps
-    end
-
-    add_index :dispatch_policy_partition_counts,
-      %i[policy_name gate_name partition_key],
-      unique: true,
-      name: "idx_dp_partition_counts_unique"
-
-    create_table :dispatch_policy_throttle_buckets do |t|
-      t.string   :policy_name,   null: false
-      t.string   :gate_name,     null: false
-      t.string   :partition_key, null: false, default: "default"
-      t.float    :tokens,        null: false
-      t.datetime :refilled_at,   null: false
-
-      t.timestamps
-    end
-
-    add_index :dispatch_policy_throttle_buckets,
-      %i[policy_name gate_name partition_key],
-      unique: true,
-      name: "idx_dp_throttle_buckets_unique"
-  end
-end

data/db/migrate/20260424000002_create_adaptive_concurrency_stats.rb

@@ -1,22 +0,0 @@
-# frozen_string_literal: true
-
-class CreateAdaptiveConcurrencyStats < ActiveRecord::Migration[7.1]
-  def change
-    create_table :dispatch_policy_adaptive_concurrency_stats do |t|
-      t.string   :policy_name,     null: false
-      t.string   :gate_name,       null: false
-      t.string   :partition_key,   null: false, default: "default"
-      t.integer  :current_max,     null: false
-      t.float    :ewma_latency_ms, null: false, default: 0
-      t.integer  :sample_count,    null: false, default: 0
-      t.datetime :last_observed_at
-
-      t.timestamps
-    end
-
-    add_index :dispatch_policy_adaptive_concurrency_stats,
-      %i[policy_name gate_name partition_key],
-      unique: true,
-      name: "idx_dp_adaptive_concurrency_stats_unique"
-  end
-end

data/db/migrate/20260424000003_create_adaptive_concurrency_samples.rb

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-class CreateAdaptiveConcurrencySamples < ActiveRecord::Migration[7.1]
-  def change
-    create_table :dispatch_policy_adaptive_concurrency_samples do |t|
-      t.string   :policy_name,     null: false
-      t.string   :gate_name,       null: false
-      t.string   :partition_key,   null: false
-      t.datetime :minute_bucket,   null: false
-      t.float    :ewma_latency_ms, null: false, default: 0
-      t.integer  :current_max,     null: false
-
-      t.timestamps
-    end
-
-    add_index :dispatch_policy_adaptive_concurrency_samples,
-      %i[policy_name gate_name partition_key minute_bucket],
-      unique: true,
-      name: "idx_dp_adaptive_concurrency_samples_unique"
-
-    add_index :dispatch_policy_adaptive_concurrency_samples,
-      :minute_bucket,
-      name: "idx_dp_adaptive_concurrency_samples_time"
-  end
-end

data/db/migrate/20260424000004_rename_samples_to_partition_observations.rb

@@ -1,32 +0,0 @@
-# frozen_string_literal: true
-
-class RenameSamplesToPartitionObservations < ActiveRecord::Migration[7.1]
-  def up
-    drop_table :dispatch_policy_adaptive_concurrency_samples, if_exists: true
-
-    create_table :dispatch_policy_partition_observations do |t|
-      t.string   :policy_name,       null: false
-      t.string   :partition_key,     null: false
-      t.datetime :minute_bucket,     null: false
-      t.bigint   :total_lag_ms,      null: false, default: 0
-      t.integer  :observation_count, null: false, default: 0
-      t.integer  :max_lag_ms,        null: false, default: 0
-      t.integer  :current_max
-
-      t.timestamps
-    end
-
-    add_index :dispatch_policy_partition_observations,
-      %i[policy_name partition_key minute_bucket],
-      unique: true,
-      name: "idx_dp_partition_observations_unique"
-
-    add_index :dispatch_policy_partition_observations,
-      :minute_bucket,
-      name: "idx_dp_partition_observations_time"
-  end
-
-  def down
-    drop_table :dispatch_policy_partition_observations
-  end
-end

data/db/migrate/20260425000001_add_duration_to_partition_observations.rb

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-class AddDurationToPartitionObservations < ActiveRecord::Migration[7.1]
-  def change
-    add_column :dispatch_policy_partition_observations, :total_duration_ms, :bigint,  null: false, default: 0
-    add_column :dispatch_policy_partition_observations, :max_duration_ms,   :integer, null: false, default: 0
-  end
-end

data/lib/dispatch_policy/active_job_perform_all_later_patch.rb

@@ -1,32 +0,0 @@
-# frozen_string_literal: true
-
-module DispatchPolicy
-  # Rails 7.1's ActiveJob.perform_all_later(*jobs) bypasses ActiveJob::Base#enqueue
-  # and calls queue_adapter.enqueue_all directly. Dispatchable hooks on #enqueue,
-  # so without this patch the batch path would skip staging.
-  module ActiveJobPerformAllLaterPatch
-    def perform_all_later(*jobs)
-      jobs.flatten!
-
-      staged, remaining = jobs.partition do |job|
-        klass = job.class
-        klass.respond_to?(:dispatch_policy?) &&
-          klass.dispatch_policy? &&
-          DispatchPolicy.enabled?
-      end
-
-      staged_count = 0
-      staged.group_by(&:class).each do |klass, group|
-        staged_count += DispatchPolicy::StagedJob.stage_many!(
-          policy: klass.resolved_dispatch_policy,
-          jobs:   group
-        )
-      end
-
-      remaining_count = remaining.empty? ? 0 : super(*remaining)
-      staged_count + remaining_count.to_i
-    end
-  end
-end
-
-ActiveJob.singleton_class.prepend(DispatchPolicy::ActiveJobPerformAllLaterPatch)