mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
233 lines
8.1 KiB
Ruby
233 lines
8.1 KiB
Ruby
require 'securerandom'
|
|
require 'sidekiq/middleware/chain'
|
|
|
|
module Sidekiq
|
|
class Client
|
|
|
|
##
|
|
# Define client-side middleware:
|
|
#
|
|
# client = Sidekiq::Client.new
|
|
# client.middleware do |chain|
|
|
# chain.use MyClientMiddleware
|
|
# end
|
|
# client.push('class' => 'SomeWorker', 'args' => [1,2,3])
|
|
#
|
|
# All client instances default to the globally-defined
|
|
# Sidekiq.client_middleware but you can change as necessary.
|
|
#
|
|
def middleware(&block)
|
|
@chain ||= Sidekiq.client_middleware
|
|
if block_given?
|
|
@chain = @chain.dup
|
|
yield @chain
|
|
end
|
|
@chain
|
|
end
|
|
|
|
attr_accessor :redis_pool
|
|
|
|
# Sidekiq::Client normally uses the default Redis pool but you may
|
|
# pass a custom ConnectionPool if you want to shard your
|
|
# Sidekiq jobs across several Redis instances (for scalability
|
|
# reasons, e.g.)
|
|
#
|
|
# Sidekiq::Client.new(ConnectionPool.new { Redis.new })
|
|
#
|
|
# Generally this is only needed for very large Sidekiq installs processing
|
|
# more than thousands jobs per second. I do not recommend sharding unless
|
|
# you truly cannot scale any other way (e.g. splitting your app into smaller apps).
|
|
# Some features, like the API, do not support sharding: they are designed to work
|
|
# against a single Redis instance only.
|
|
def initialize(redis_pool=nil)
|
|
@redis_pool = redis_pool || Thread.current[:sidekiq_via_pool] || Sidekiq.redis_pool
|
|
end
|
|
|
|
##
|
|
# The main method used to push a job to Redis. Accepts a number of options:
|
|
#
|
|
# queue - the named queue to use, default 'default'
|
|
# class - the worker class to call, required
|
|
# args - an array of simple arguments to the perform method, must be JSON-serializable
|
|
# retry - whether to retry this job if it fails, true or false, default true
|
|
# backtrace - whether to save any error backtrace, default false
|
|
#
|
|
# All options must be strings, not symbols. NB: because we are serializing to JSON, all
|
|
# symbols in 'args' will be converted to strings.
|
|
#
|
|
# Returns a unique Job ID. If middleware stops the job, nil will be returned instead.
|
|
#
|
|
# Example:
|
|
# push('queue' => 'my_queue', 'class' => MyWorker, 'args' => ['foo', 1, :bat => 'bar'])
|
|
#
|
|
def push(item)
|
|
normed = normalize_item(item)
|
|
payload = process_single(item['class'], normed)
|
|
|
|
if payload
|
|
raw_push([payload])
|
|
payload['jid']
|
|
end
|
|
end
|
|
|
|
##
|
|
# Push a large number of jobs to Redis. In practice this method is only
|
|
# useful if you are pushing tens of thousands of jobs or more, or if you need
|
|
# to ensure that a batch doesn't complete prematurely. This method
|
|
# basically cuts down on the redis round trip latency.
|
|
#
|
|
# Takes the same arguments as #push except that args is expected to be
|
|
# an Array of Arrays. All other keys are duplicated for each job. Each job
|
|
# is run through the client middleware pipeline and each job gets its own Job ID
|
|
# as normal.
|
|
#
|
|
# Returns an array of the of pushed jobs' jids. The number of jobs pushed can be less
|
|
# than the number given if the middleware stopped processing for one or more jobs.
|
|
def push_bulk(items)
|
|
normed = normalize_item(items)
|
|
payloads = items['args'].map do |args|
|
|
raise ArgumentError, "Bulk arguments must be an Array of Arrays: [[1], [2]]" if !args.is_a?(Array)
|
|
process_single(items['class'], normed.merge('args' => args, 'jid' => SecureRandom.hex(12), 'enqueued_at' => Time.now.to_f))
|
|
end.compact
|
|
|
|
raw_push(payloads) if !payloads.empty?
|
|
payloads.collect { |payload| payload['jid'] }
|
|
end
|
|
|
|
# Allows sharding of jobs across any number of Redis instances. All jobs
|
|
# defined within the block will use the given Redis connection pool.
|
|
#
|
|
# pool = ConnectionPool.new { Redis.new }
|
|
# Sidekiq::Client.via(pool) do
|
|
# SomeWorker.perform_async(1,2,3)
|
|
# SomeOtherWorker.perform_async(1,2,3)
|
|
# end
|
|
#
|
|
# Generally this is only needed for very large Sidekiq installs processing
|
|
# more than thousands jobs per second. I do not recommend sharding unless
|
|
# you truly cannot scale any other way (e.g. splitting your app into smaller apps).
|
|
# Some features, like the API, do not support sharding: they are designed to work
|
|
# against a single Redis instance.
|
|
def self.via(pool)
|
|
raise ArgumentError, "No pool given" if pool.nil?
|
|
raise RuntimeError, "Sidekiq::Client.via is not re-entrant" if x = Thread.current[:sidekiq_via_pool] && x != pool
|
|
Thread.current[:sidekiq_via_pool] = pool
|
|
yield
|
|
ensure
|
|
Thread.current[:sidekiq_via_pool] = nil
|
|
end
|
|
|
|
class << self
|
|
|
|
def default
|
|
@default ||= new
|
|
end
|
|
|
|
def push(item)
|
|
default.push(item)
|
|
end
|
|
|
|
def push_bulk(items)
|
|
default.push_bulk(items)
|
|
end
|
|
|
|
# Resque compatibility helpers. Note all helpers
|
|
# should go through Worker#client_push.
|
|
#
|
|
# Example usage:
|
|
# Sidekiq::Client.enqueue(MyWorker, 'foo', 1, :bat => 'bar')
|
|
#
|
|
# Messages are enqueued to the 'default' queue.
|
|
#
|
|
def enqueue(klass, *args)
|
|
klass.client_push('class' => klass, 'args' => args)
|
|
end
|
|
|
|
# Example usage:
|
|
# Sidekiq::Client.enqueue_to(:queue_name, MyWorker, 'foo', 1, :bat => 'bar')
|
|
#
|
|
def enqueue_to(queue, klass, *args)
|
|
klass.client_push('queue' => queue, 'class' => klass, 'args' => args)
|
|
end
|
|
|
|
# Example usage:
|
|
# Sidekiq::Client.enqueue_to_in(:queue_name, 3.minutes, MyWorker, 'foo', 1, :bat => 'bar')
|
|
#
|
|
def enqueue_to_in(queue, interval, klass, *args)
|
|
int = interval.to_f
|
|
now = Time.now.to_f
|
|
ts = (int < 1_000_000_000 ? now + int : int)
|
|
|
|
item = { 'class' => klass, 'args' => args, 'at' => ts, 'queue' => queue }
|
|
item.delete('at'.freeze) if ts <= now
|
|
|
|
klass.client_push(item)
|
|
end
|
|
|
|
# Example usage:
|
|
# Sidekiq::Client.enqueue_in(3.minutes, MyWorker, 'foo', 1, :bat => 'bar')
|
|
#
|
|
def enqueue_in(interval, klass, *args)
|
|
klass.perform_in(interval, *args)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def raw_push(payloads)
|
|
@redis_pool.with do |conn|
|
|
conn.multi do
|
|
atomic_push(conn, payloads)
|
|
end
|
|
end
|
|
true
|
|
end
|
|
|
|
def atomic_push(conn, payloads)
|
|
if payloads.first['at']
|
|
conn.zadd('schedule'.freeze, payloads.map do |hash|
|
|
at = hash.delete('at'.freeze).to_s
|
|
[at, Sidekiq.dump_json(hash)]
|
|
end)
|
|
else
|
|
q = payloads.first['queue']
|
|
to_push = payloads.map { |entry| Sidekiq.dump_json(entry) }
|
|
conn.sadd('queues'.freeze, q)
|
|
conn.lpush("queue:#{q}", to_push)
|
|
end
|
|
end
|
|
|
|
def process_single(worker_class, item)
|
|
queue = item['queue']
|
|
|
|
middleware.invoke(worker_class, item, queue, @redis_pool) do
|
|
item
|
|
end
|
|
end
|
|
|
|
def normalize_item(item)
|
|
raise(ArgumentError, "Message must be a Hash of the form: { 'class' => SomeWorker, 'args' => ['bob', 1, :foo => 'bar'] }") unless item.is_a?(Hash)
|
|
raise(ArgumentError, "Message must include a class and set of arguments: #{item.inspect}") if !item['class'] || !item['args']
|
|
raise(ArgumentError, "Message args must be an Array") unless item['args'].is_a?(Array)
|
|
raise(ArgumentError, "Message class must be either a Class or String representation of the class name") unless item['class'].is_a?(Class) || item['class'].is_a?(String)
|
|
|
|
normalized_hash(item['class'.freeze])
|
|
.each{ |key, value| item[key] = value if item[key].nil? }
|
|
|
|
item['class'.freeze] = item['class'.freeze].to_s
|
|
item['queue'.freeze] = item['queue'.freeze].to_s
|
|
item['jid'.freeze] ||= SecureRandom.hex(12)
|
|
item['enqueued_at'.freeze] ||= Time.now.to_f
|
|
item
|
|
end
|
|
|
|
def normalized_hash(item_class)
|
|
if item_class.is_a?(Class)
|
|
raise(ArgumentError, "Message must include a Sidekiq::Worker class, not class name: #{item_class.ancestors.inspect}") if !item_class.respond_to?('get_sidekiq_options'.freeze)
|
|
item_class.get_sidekiq_options
|
|
else
|
|
Sidekiq.default_worker_options
|
|
end
|
|
end
|
|
end
|
|
end
|