undertow 0.2.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bf230477188804dcf6c63e0c97c615defb3577c78f85e930449d7dba6d9d965e
-  data.tar.gz: 6653637c116fff79ba2a2806326882f7e7e939acdf1ca0f01a653e22530dd3c7
+  metadata.gz: 1aa09fe4d1c2e088f4b7ca874175db916486f8a6296d49af88544bec8ec254be
+  data.tar.gz: 30115bb681233cf045251670a9f2ed369e6290340414f5f031f292cd2e26fe23
 SHA512:
-  metadata.gz: 74aca2058a93a4474bd275de1ab3d077b7e83e5a6449cb0921938585821c29f39ab531240eabbfb89d28ba7ac4a73cd4f3c58183e001992847d1a3ebae6e89b2
-  data.tar.gz: 6c18ad42284eb95008c44bfd6240e79ae14c7d99aa50befa8c1ccb417d07286a55049c1467f6e4f66a66d8f93df8e7c1b613cc045217df6ba81c9334af3cf368
+  metadata.gz: a68ec79086258526eb24fbd2690b5df15701975706ee4ae09bbbd800d4e024d1b9c4546d7df03f465de7bbdcd4efc6252b73ac2244252b74d6f93707ea5dff1b
+  data.tar.gz: 9406d057b05d1a76fade09a0d3b3b698d3503a11bccfdd68591f4eead07bce128dfd3b431fbf4e6078eac092bc4f641a51d0dda036bdd200ab928dbec24155af

data/lib/undertow/drain_job.rb

@@ -5,7 +5,7 @@ module Undertow
   # each model's configured on_drain handler.
   #
   # Publishes two ActiveSupport::Notifications events:
-  #   drain.undertow , after a successful on_drain call ({ model:, ids:, deleted_ids: })
+  #   drain.undertow , after a successful on_drain call ({ model:, upserted_ids:, deleted_ids:, duration_ms: })
   #   error.undertow , when on_drain raises ({ model:, exception: })
   class DrainJob < ActiveJob::Base
     queue_as { Undertow.configuration.queue_name }
@@ -30,9 +30,9 @@ module Undertow
       # preventing the race where srem fires after a concurrent sadd.
       Buffer.deregister_model(model_name)
 
-      ids         = Buffer.pop_pending(model_name, max)
-      deleted_ids = Buffer.pop_deleted(model_name, max)
-      return if ids.empty? && deleted_ids.empty?
+      upserted_ids = Buffer.pop_pending(model_name, max)
+      deleted_ids  = Buffer.pop_deleted(model_name, max)
+      return if upserted_ids.empty? && deleted_ids.empty?
 
       # If the batch was capped, re-register so the next scheduler tick picks up.
       Buffer.reregister_model(model_name) if Buffer.remaining(model_name).positive?
@@ -41,19 +41,26 @@ module Undertow
       raise "No Undertow config registered for #{model_name}" unless config
       raise "#{model_name} is missing undertow_on_drain" unless config.on_drain
 
-      config.on_drain.call(model_name, ids, deleted_ids)
+      duration_ms = measure_ms { config.on_drain.call(model_name, upserted_ids, deleted_ids) }
 
       ActiveSupport::Notifications.instrument('drain.undertow', {
         model: model_name,
-        ids: ids,
-        deleted_ids: deleted_ids
+        upserted_ids: upserted_ids,
+        deleted_ids: deleted_ids,
+        duration_ms: duration_ms
       })
     rescue StandardError => e
-      Buffer.restore_pending(model_name, ids)         if ids&.any?
-      Buffer.restore_deleted(model_name, deleted_ids) if deleted_ids&.any?
+      Buffer.restore_pending(model_name, upserted_ids) if upserted_ids&.any?
+      Buffer.restore_deleted(model_name, deleted_ids)  if deleted_ids&.any?
       Buffer.reregister_model(model_name)
       ActiveSupport::Notifications.instrument('error.undertow', { model: model_name, exception: e })
       Rails.logger.error("[Undertow::DrainJob] #{model_name}: #{e.message}") if defined?(Rails)
     end
+
+    def measure_ms
+      start = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+      yield
+      ((Process.clock_gettime(Process::CLOCK_MONOTONIC) - start) * 1000).round(2)
+    end
   end
 end

data/lib/undertow/dsl.rb

@@ -5,14 +5,14 @@ module Undertow
   # that calls these methods automatically registers itself with Undertow and
   # gets Trackable behavior wired in at boot, no include needed.
   #
-  #   class Activity < ApplicationRecord
-  #     undertow_on_drain ->(model_name, ids, deleted_ids) { ActivityReindexJob.perform_later(ids, deleted_ids) }
-  #     undertow_skip     %w[lock_version searchkick_reindexing]
+  #   class Post < ApplicationRecord
+  #     undertow_on_drain ->(model_name, upserted_ids, deleted_ids) { PostSyncJob.perform_later(upserted_ids, deleted_ids) }
+  #     undertow_skip     %w[view_count updated_at]
   #
-  #     undertow_depends_on :provider, foreign_key: :provider_id, watched_columns: %w[approved mobile]
-  #     undertow_depends_on :location_series,
-  #                         resolver:        ->(ls) { Activity.where(series_id: ls.series_id) },
-  #                         watched_columns: %w[location_id hidden]
+  #     undertow_depends_on :author, foreign_key: :author_id, watched_columns: %w[name bio]
+  #     undertow_depends_on :tag,
+  #                         resolver:        ->(tag) { Post.joins(:post_tags).where(post_tags: { tag_id: tag.id }) },
+  #                         watched_columns: %w[name slug]
   #   end
   #
   module DSL

data/lib/undertow/trackable.rb

@@ -39,6 +39,23 @@ module Undertow
         Buffer.push_deleted(name, ids)
       end
 
+      # Enqueues records into the pending buffer without going through AR
+      # callbacks. Use this to replay a batch through the drain handler without
+      # triggering other model lifecycle callbacks, e.g. from a console or a
+      # data migration.
+      #
+      #   Post.undertow_requeue(Post.where(author_id: 123))
+      #   Post.undertow_requeue([136543, 136544])
+      #   Post.undertow_requeue  # all records
+      def undertow_requeue(scope_or_ids = :all)
+        ids = case scope_or_ids
+              when :all                   then pluck(:id)
+              when ActiveRecord::Relation then scope_or_ids.pluck(:id)
+              else                             Array(scope_or_ids)
+              end
+        _push_undertow_pending(ids)
+      end
+
       def _push_dep_pending(record, dep)
         ids = _dep_ids_for(record, dep)
         _push_undertow_pending(ids) if ids.any?

data/lib/undertow/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Undertow
-  VERSION = '0.2.1'
+  VERSION = '0.4.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: undertow
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.4.0
 platform: ruby
 authors:
 - Nathan Allen
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-06-13 00:00:00.000000000 Z
+date: 2026-07-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord