mastodon/lib/mastodon/worker_batch_middleware.rb

12 lines
225 B
Ruby

# frozen_string_literal: true
class Mastodon::WorkerBatchMiddleware
def call(_worker, msg, _queue, _redis_pool = nil)
if (batch = Thread.current[:batch])
batch.add_jobs([msg['jid']])
end
yield
end
end