forked from tobi/delayed_job
-
Notifications
You must be signed in to change notification settings - Fork 955
/
message_sending.rb
52 lines (46 loc) · 1.72 KB
/
message_sending.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
require 'active_support/core_ext/module/aliasing'
module Delayed
class DelayProxy < Delayed::Compatibility.proxy_object_class
def initialize(payload_class, target, options)
@payload_class = payload_class
@target = target
@options = options
end
def method_missing(method, *args)
Job.enqueue({:payload_object => @payload_class.new(@target, method.to_sym, args)}.merge(@options))
end
end
module MessageSending
def delay(options = {})
DelayProxy.new(PerformableMethod, self, options)
end
alias_method :__delay__, :delay
def send_later(method, *args)
warn '[DEPRECATION] `object.send_later(:method)` is deprecated. Use `object.delay.method'
__delay__.__send__(method, *args)
end
def send_at(time, method, *args)
warn '[DEPRECATION] `object.send_at(time, :method)` is deprecated. Use `object.delay(:run_at => time).method'
__delay__(:run_at => time).__send__(method, *args)
end
module ClassMethods
def handle_asynchronously(method, opts = {})
aliased_method, punctuation = method.to_s.sub(/([?!=])$/, ''), $1 # rubocop:disable PerlBackrefs
with_method, without_method = "#{aliased_method}_with_delay#{punctuation}", "#{aliased_method}_without_delay#{punctuation}"
define_method(with_method) do |*args|
curr_opts = opts.clone
curr_opts.each_key do |key|
next unless (val = curr_opts[key]).is_a?(Proc)
curr_opts[key] = if val.arity == 1
val.call(self)
else
val.call
end
end
delay(curr_opts).__send__(without_method, *args)
end
alias_method_chain method, :delay
end
end
end
end