/
trigger.rb
59 lines (49 loc) · 1.46 KB
/
trigger.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
56
57
58
59
module Bluepill
class Trigger
@implementations = {}
def self.inherited(klass)
@implementations[klass.name.split('::').last.underscore.to_sym] = klass
end
def self.[](name)
@implementations[name]
end
attr_accessor :process, :logger, :mutex, :scheduled_events
def initialize(process, options = {})
self.process = process
self.logger = options[:logger]
self.mutex = Mutex.new
self.scheduled_events = []
end
def reset!
self.cancel_all_events
end
def notify(transition)
raise "Implement in subclass"
end
def dispatch!(event)
self.process.dispatch!(event, self.class.name.split("::").last)
end
def schedule_event(event, delay)
# TODO: maybe wrap this in a ScheduledEvent class with methods like cancel
thread = Thread.new(self) do |trigger|
begin
sleep delay.to_f
trigger.dispatch!(event)
trigger.mutex.synchronize do
trigger.scheduled_events.delete_if { |_, thread| thread == Thread.current }
end
rescue StandardError => e
trigger.logger.err(e)
trigger.logger.err(e.backtrace.join("\n"))
end
end
self.scheduled_events.push([event, thread])
end
def cancel_all_events
self.logger.info "Canceling all scheduled events"
self.mutex.synchronize do
self.scheduled_events.each {|_, thread| thread.kill}
end
end
end
end