-
Notifications
You must be signed in to change notification settings - Fork 129
/
periodic_timer.rb
55 lines (43 loc) · 1.13 KB
/
periodic_timer.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
53
54
55
module Cramp
module PeriodicTimer
extend ActiveSupport::Concern
included do
class_inheritable_accessor :periodic_timers, :instance_reader => false
self.periodic_timers ||= []
end
module ClassMethods
def periodic_timer(method, options = {})
self.periodic_timers << [method, options]
end
end
def initialize(*)
super
@timers = []
end
protected
def continue
super
start_periodic_timers
end
def init_async_body
super
if self.class.periodic_timers.any?
@body.callback { stop_periodic_timers }
@body.errback { stop_periodic_timers }
end
end
def start_periodic_timers
self.class.periodic_timers.each do |method, options|
@timers << EventMachine::PeriodicTimer.new(options[:every] || 1) { timer_method_wrapper(method) unless @finished }
end
end
def stop_periodic_timers
@timers.each {|t| t.cancel }
end
def timer_method_wrapper(method)
send(method)
rescue StandardError, LoadError, SyntaxError => exception
handle_exception(exception)
end
end
end