2012-08-16 19:43:01 -04:00
|
|
|
require 'sidekiq/extensions/generic_proxy'
|
|
|
|
|
|
|
|
module Sidekiq
|
|
|
|
module Extensions
|
|
|
|
##
|
2013-06-01 03:06:15 -04:00
|
|
|
# Adds 'delay', 'delay_for' and `delay_until` methods to all Classes to offload class method
|
2012-08-16 19:43:01 -04:00
|
|
|
# execution to Sidekiq. Examples:
|
|
|
|
#
|
|
|
|
# User.delay.delete_inactive
|
|
|
|
# Wikipedia.delay.download_changes_for(Date.today)
|
|
|
|
#
|
|
|
|
class DelayedClass
|
|
|
|
include Sidekiq::Worker
|
|
|
|
|
|
|
|
def perform(yml)
|
|
|
|
(target, method_name, args) = YAML.load(yml)
|
2014-02-14 09:43:34 -05:00
|
|
|
target.__send__(method_name, *args)
|
2012-08-16 19:43:01 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
module Klass
|
2013-12-15 17:16:31 -05:00
|
|
|
def sidekiq_delay(options={})
|
2012-10-17 13:17:19 -04:00
|
|
|
Proxy.new(DelayedClass, self, options)
|
2012-08-16 19:43:01 -04:00
|
|
|
end
|
2013-12-15 17:16:31 -05:00
|
|
|
def sidekiq_delay_for(interval, options={})
|
2012-10-17 13:17:19 -04:00
|
|
|
Proxy.new(DelayedClass, self, options.merge('at' => Time.now.to_f + interval.to_f))
|
2012-08-16 19:43:01 -04:00
|
|
|
end
|
2013-12-15 17:16:31 -05:00
|
|
|
def sidekiq_delay_until(timestamp, options={})
|
2012-11-08 12:05:11 -05:00
|
|
|
Proxy.new(DelayedClass, self, options.merge('at' => timestamp.to_f))
|
|
|
|
end
|
2013-12-15 17:16:31 -05:00
|
|
|
alias_method :delay, :sidekiq_delay
|
|
|
|
alias_method :delay_for, :sidekiq_delay_for
|
|
|
|
alias_method :delay_until, :sidekiq_delay_until
|
2012-08-16 19:43:01 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2014-06-28 18:19:42 -04:00
|
|
|
Module.__send__(:include, Sidekiq::Extensions::Klass) unless defined?(::Rails)
|