1
0
Fork 0
mirror of https://github.com/mperham/sidekiq.git synced 2022-11-09 13:52:34 -05:00

Add API to reschedule a job, fixes #645

This commit is contained in:
Mike Perham 2013-01-29 13:17:59 -08:00
parent c4154a8802
commit 1f431cb56d
2 changed files with 27 additions and 0 deletions

View file

@ -210,6 +210,11 @@ module Sidekiq
@parent.delete(score, jid) @parent.delete(score, jid)
end end
def reschedule(at)
@parent.delete(score, jid)
@parent.schedule(at, item)
end
def retry def retry
raise "Retry not available on jobs not in the Retry queue." unless item["failed_at"] raise "Retry not available on jobs not in the Retry queue." unless item["failed_at"]
Sidekiq.redis do |conn| Sidekiq.redis do |conn|
@ -235,6 +240,12 @@ module Sidekiq
Sidekiq.redis {|c| c.zcard(@zset) } Sidekiq.redis {|c| c.zcard(@zset) }
end end
def schedule(timestamp, message)
Sidekiq.redis do |conn|
conn.zadd(@zset, timestamp.to_s, Sidekiq.dump_json(message))
end
end
def each(&block) def each(&block)
# page thru the sorted set backwards so deleting entries doesn't screw up indexing # page thru the sorted set backwards so deleting entries doesn't screw up indexing
page = -1 page = -1

View file

@ -292,6 +292,22 @@ class TestApi < MiniTest::Unit::TestCase
end end
end end
it 'can reschedule jobs' do
add_retry('foo1')
add_retry('foo2')
retries = Sidekiq::RetrySet.new
assert_equal 2, retries.size
refute(retries.map { |r| r.score > (Time.now.to_f + 9) }.any?)
retries.each do |retri|
retri.reschedule(Time.now.to_f + 10) if retri.jid == 'foo2'
end
assert_equal 2, retries.size
assert(retries.map { |r| r.score > (Time.now.to_f + 9) }.any?)
end
def add_retry(jid = 'bob', at = Time.now.to_f) def add_retry(jid = 'bob', at = Time.now.to_f)
payload = Sidekiq.dump_json('class' => 'ApiWorker', 'args' => [1, 'mike'], 'queue' => 'default', 'jid' => jid, 'retry_count' => 2, 'failed_at' => Time.now.utc) payload = Sidekiq.dump_json('class' => 'ApiWorker', 'args' => [1, 'mike'], 'queue' => 'default', 'jid' => jid, 'retry_count' => 2, 'failed_at' => Time.now.utc)
Sidekiq.redis do |conn| Sidekiq.redis do |conn|