jwilkins-god 0.7.9
Sign up to get free protection for your applications and to get access to all the features.
- data/History.txt +261 -0
- data/Manifest.txt +107 -0
- data/README.txt +59 -0
- data/Rakefile +35 -0
- data/bin/god +127 -0
- data/examples/events.god +84 -0
- data/examples/gravatar.god +54 -0
- data/examples/single.god +66 -0
- data/ext/god/extconf.rb +55 -0
- data/ext/god/kqueue_handler.c +123 -0
- data/ext/god/netlink_handler.c +167 -0
- data/init/god +42 -0
- data/lib/god.rb +649 -0
- data/lib/god/behavior.rb +52 -0
- data/lib/god/behaviors/clean_pid_file.rb +21 -0
- data/lib/god/behaviors/clean_unix_socket.rb +21 -0
- data/lib/god/behaviors/notify_when_flapping.rb +51 -0
- data/lib/god/cli/command.rb +206 -0
- data/lib/god/cli/run.rb +177 -0
- data/lib/god/cli/version.rb +23 -0
- data/lib/god/condition.rb +96 -0
- data/lib/god/conditions/always.rb +23 -0
- data/lib/god/conditions/complex.rb +86 -0
- data/lib/god/conditions/cpu_usage.rb +80 -0
- data/lib/god/conditions/degrading_lambda.rb +52 -0
- data/lib/god/conditions/disk_usage.rb +27 -0
- data/lib/god/conditions/flapping.rb +128 -0
- data/lib/god/conditions/http_response_code.rb +168 -0
- data/lib/god/conditions/lambda.rb +25 -0
- data/lib/god/conditions/memory_usage.rb +82 -0
- data/lib/god/conditions/process_exits.rb +72 -0
- data/lib/god/conditions/process_running.rb +74 -0
- data/lib/god/conditions/tries.rb +44 -0
- data/lib/god/configurable.rb +57 -0
- data/lib/god/contact.rb +106 -0
- data/lib/god/contacts/email.rb +95 -0
- data/lib/god/dependency_graph.rb +41 -0
- data/lib/god/diagnostics.rb +37 -0
- data/lib/god/driver.rb +206 -0
- data/lib/god/errors.rb +24 -0
- data/lib/god/event_handler.rb +111 -0
- data/lib/god/event_handlers/dummy_handler.rb +13 -0
- data/lib/god/event_handlers/kqueue_handler.rb +17 -0
- data/lib/god/event_handlers/netlink_handler.rb +13 -0
- data/lib/god/logger.rb +120 -0
- data/lib/god/metric.rb +59 -0
- data/lib/god/process.rb +327 -0
- data/lib/god/registry.rb +32 -0
- data/lib/god/simple_logger.rb +53 -0
- data/lib/god/socket.rb +96 -0
- data/lib/god/sugar.rb +47 -0
- data/lib/god/system/portable_poller.rb +42 -0
- data/lib/god/system/process.rb +42 -0
- data/lib/god/system/slash_proc_poller.rb +82 -0
- data/lib/god/task.rb +487 -0
- data/lib/god/timeline.rb +25 -0
- data/lib/god/trigger.rb +43 -0
- data/lib/god/watch.rb +183 -0
- data/test/configs/child_events/child_events.god +44 -0
- data/test/configs/child_events/simple_server.rb +3 -0
- data/test/configs/child_polls/child_polls.god +37 -0
- data/test/configs/child_polls/simple_server.rb +12 -0
- data/test/configs/complex/complex.god +59 -0
- data/test/configs/complex/simple_server.rb +3 -0
- data/test/configs/contact/contact.god +74 -0
- data/test/configs/contact/simple_server.rb +3 -0
- data/test/configs/daemon_events/daemon_events.god +37 -0
- data/test/configs/daemon_events/simple_server.rb +8 -0
- data/test/configs/daemon_events/simple_server_stop.rb +11 -0
- data/test/configs/daemon_polls/daemon_polls.god +17 -0
- data/test/configs/daemon_polls/simple_server.rb +6 -0
- data/test/configs/degrading_lambda/degrading_lambda.god +31 -0
- data/test/configs/degrading_lambda/tcp_server.rb +15 -0
- data/test/configs/matias/matias.god +50 -0
- data/test/configs/real.rb +59 -0
- data/test/configs/running_load/running_load.god +16 -0
- data/test/configs/stress/simple_server.rb +3 -0
- data/test/configs/stress/stress.god +15 -0
- data/test/configs/task/logs/.placeholder +0 -0
- data/test/configs/task/task.god +26 -0
- data/test/configs/test.rb +61 -0
- data/test/helper.rb +151 -0
- data/test/suite.rb +6 -0
- data/test/test_behavior.rb +21 -0
- data/test/test_condition.rb +50 -0
- data/test/test_conditions_disk_usage.rb +56 -0
- data/test/test_conditions_http_response_code.rb +109 -0
- data/test/test_conditions_process_running.rb +44 -0
- data/test/test_conditions_tries.rb +67 -0
- data/test/test_contact.rb +109 -0
- data/test/test_dependency_graph.rb +62 -0
- data/test/test_driver.rb +11 -0
- data/test/test_event_handler.rb +80 -0
- data/test/test_god.rb +598 -0
- data/test/test_handlers_kqueue_handler.rb +16 -0
- data/test/test_logger.rb +63 -0
- data/test/test_metric.rb +72 -0
- data/test/test_process.rb +246 -0
- data/test/test_registry.rb +15 -0
- data/test/test_socket.rb +42 -0
- data/test/test_sugar.rb +42 -0
- data/test/test_system_portable_poller.rb +17 -0
- data/test/test_system_process.rb +30 -0
- data/test/test_task.rb +262 -0
- data/test/test_timeline.rb +37 -0
- data/test/test_trigger.rb +59 -0
- data/test/test_watch.rb +279 -0
- metadata +186 -0
@@ -0,0 +1,57 @@
|
|
1
|
+
module God
|
2
|
+
|
3
|
+
module Configurable
|
4
|
+
# Override this method in your Configurable (optional)
|
5
|
+
#
|
6
|
+
# Called once after the Configurable has been sent to the block and attributes have been
|
7
|
+
# set. Do any post-processing on attributes here
|
8
|
+
def prepare
|
9
|
+
|
10
|
+
end
|
11
|
+
|
12
|
+
def reset
|
13
|
+
|
14
|
+
end
|
15
|
+
|
16
|
+
# Override this method in your Configurable (optional)
|
17
|
+
#
|
18
|
+
# Called once during evaluation of the config file. Return true if valid, false otherwise
|
19
|
+
#
|
20
|
+
# A convenience method 'complain' is available that will print out a message and return false,
|
21
|
+
# making it easy to report multiple validation errors:
|
22
|
+
#
|
23
|
+
# def valid?
|
24
|
+
# valid = true
|
25
|
+
# valid &= complain("You must specify the 'pid_file' attribute for :memory_usage") if self.pid_file.nil?
|
26
|
+
# valid &= complain("You must specify the 'above' attribute for :memory_usage") if self.above.nil?
|
27
|
+
# valid
|
28
|
+
# end
|
29
|
+
def valid?
|
30
|
+
true
|
31
|
+
end
|
32
|
+
|
33
|
+
def base_name
|
34
|
+
x = 1 # fix for MRI's local scope optimization bug DO NOT REMOVE!
|
35
|
+
self.class.name.split('::').last
|
36
|
+
end
|
37
|
+
|
38
|
+
def friendly_name
|
39
|
+
base_name
|
40
|
+
end
|
41
|
+
|
42
|
+
def self.complain(text, c = nil)
|
43
|
+
watch = c.watch rescue nil
|
44
|
+
msg = ""
|
45
|
+
msg += "#{watch.name}: " if watch
|
46
|
+
msg += text
|
47
|
+
msg += " for #{c.friendly_name}" if c
|
48
|
+
applog(watch, :error, msg)
|
49
|
+
false
|
50
|
+
end
|
51
|
+
|
52
|
+
def complain(text, c = nil)
|
53
|
+
Configurable.complain(text, c)
|
54
|
+
end
|
55
|
+
end
|
56
|
+
|
57
|
+
end
|
data/lib/god/contact.rb
ADDED
@@ -0,0 +1,106 @@
|
|
1
|
+
module God
|
2
|
+
|
3
|
+
class Contact
|
4
|
+
include Configurable
|
5
|
+
|
6
|
+
attr_accessor :name, :group, :info
|
7
|
+
|
8
|
+
def self.generate(kind)
|
9
|
+
sym = kind.to_s.capitalize.gsub(/_(.)/){$1.upcase}.intern
|
10
|
+
c = God::Contacts.const_get(sym).new
|
11
|
+
|
12
|
+
unless c.kind_of?(Contact)
|
13
|
+
abort "Contact '#{c.class.name}' must subclass God::Contact"
|
14
|
+
end
|
15
|
+
|
16
|
+
c
|
17
|
+
rescue NameError
|
18
|
+
raise NoSuchContactError.new("No Contact found with the class name God::Contacts::#{sym}")
|
19
|
+
end
|
20
|
+
|
21
|
+
def self.valid?(contact)
|
22
|
+
valid = true
|
23
|
+
valid &= Configurable.complain("Attribute 'name' must be specified", contact) if contact.name.nil?
|
24
|
+
valid
|
25
|
+
end
|
26
|
+
|
27
|
+
# Normalize the given notify specification into canonical form.
|
28
|
+
# +spec+ is the notify spec as a String, Array of Strings, or Hash
|
29
|
+
#
|
30
|
+
# Canonical form looks like:
|
31
|
+
# {:contacts => ['fred', 'john'], :priority => '1', :category => 'awesome'}
|
32
|
+
# Where :contacts will be present and point to an Array of Strings. Both
|
33
|
+
# :priority and :category may not be present but if they are, they will each
|
34
|
+
# contain a single String.
|
35
|
+
#
|
36
|
+
# Returns normalized notify spec
|
37
|
+
# Raises ArgumentError on invalid spec (message contains details)
|
38
|
+
def self.normalize(spec)
|
39
|
+
case spec
|
40
|
+
when String
|
41
|
+
{:contacts => Array(spec)}
|
42
|
+
when Array
|
43
|
+
unless spec.select { |x| !x.instance_of?(String) }.empty?
|
44
|
+
raise ArgumentError.new("contains non-String elements")
|
45
|
+
end
|
46
|
+
{:contacts => spec}
|
47
|
+
when Hash
|
48
|
+
copy = spec.dup
|
49
|
+
|
50
|
+
# check :contacts
|
51
|
+
if contacts = copy.delete(:contacts)
|
52
|
+
case contacts
|
53
|
+
when String
|
54
|
+
# valid
|
55
|
+
when Array
|
56
|
+
unless contacts.select { |x| !x.instance_of?(String) }.empty?
|
57
|
+
raise ArgumentError.new("has a :contacts key containing non-String elements")
|
58
|
+
end
|
59
|
+
# valid
|
60
|
+
else
|
61
|
+
raise ArgumentError.new("must have a :contacts key pointing to a String or Array of Strings")
|
62
|
+
end
|
63
|
+
else
|
64
|
+
raise ArgumentError.new("must have a :contacts key")
|
65
|
+
end
|
66
|
+
|
67
|
+
# remove priority and category
|
68
|
+
copy.delete(:priority)
|
69
|
+
copy.delete(:category)
|
70
|
+
|
71
|
+
# check for invalid keys
|
72
|
+
unless copy.empty?
|
73
|
+
raise ArgumentError.new("contains extra elements: #{copy.inspect}")
|
74
|
+
end
|
75
|
+
|
76
|
+
# normalize
|
77
|
+
spec[:contacts] &&= Array(spec[:contacts])
|
78
|
+
spec[:priority] &&= spec[:priority].to_s
|
79
|
+
spec[:category] &&= spec[:category].to_s
|
80
|
+
|
81
|
+
spec
|
82
|
+
else
|
83
|
+
raise ArgumentError.new("must be a String (contact name), Array (of contact names), or Hash (contact specification)")
|
84
|
+
end
|
85
|
+
end
|
86
|
+
|
87
|
+
# Abstract
|
88
|
+
# Send the message to the external source
|
89
|
+
# +message+ is the message body returned from the condition
|
90
|
+
# +time+ is the Time at which the notification was made
|
91
|
+
# +priority+ is the arbitrary priority String
|
92
|
+
# +category+ is the arbitrary category String
|
93
|
+
# +host+ is the hostname of the server
|
94
|
+
def notify(message, time, priority, category, host)
|
95
|
+
raise AbstractMethodNotOverriddenError.new("Contact#notify must be overridden in subclasses")
|
96
|
+
end
|
97
|
+
|
98
|
+
# Construct the friendly name of this Contact, looks like:
|
99
|
+
#
|
100
|
+
# Contact FooBar
|
101
|
+
def friendly_name
|
102
|
+
super + " Contact '#{self.name}'"
|
103
|
+
end
|
104
|
+
end
|
105
|
+
|
106
|
+
end
|
@@ -0,0 +1,95 @@
|
|
1
|
+
require 'time'
|
2
|
+
require 'net/smtp'
|
3
|
+
|
4
|
+
module God
|
5
|
+
module Contacts
|
6
|
+
|
7
|
+
class Email < Contact
|
8
|
+
class << self
|
9
|
+
attr_accessor :message_settings, :delivery_method, :server_settings, :sendmail_settings, :format
|
10
|
+
end
|
11
|
+
|
12
|
+
self.message_settings = {:from => 'god@example.com'}
|
13
|
+
|
14
|
+
self.delivery_method = :smtp # or :sendmail
|
15
|
+
|
16
|
+
self.server_settings = {:address => 'localhost',
|
17
|
+
:port => 25}
|
18
|
+
# :domain
|
19
|
+
# :user_name
|
20
|
+
# :password
|
21
|
+
# :authentication
|
22
|
+
|
23
|
+
self.sendmail_settings = {:location => '/usr/sbin/sendmail',
|
24
|
+
:arguments => '-i -t'
|
25
|
+
}
|
26
|
+
|
27
|
+
self.format = lambda do |name, email, message, time, priority, category, host|
|
28
|
+
<<-EOF
|
29
|
+
From: god <#{self.message_settings[:from]}>
|
30
|
+
To: #{name} <#{email}>
|
31
|
+
Subject: [god] #{message}
|
32
|
+
Date: #{Time.now.httpdate}
|
33
|
+
Message-Id: <unique.message.id.string@example.com>
|
34
|
+
|
35
|
+
Message: #{message}
|
36
|
+
Host: #{host}
|
37
|
+
Priority: #{priority}
|
38
|
+
Category: #{category}
|
39
|
+
EOF
|
40
|
+
end
|
41
|
+
|
42
|
+
attr_accessor :email
|
43
|
+
|
44
|
+
def valid?
|
45
|
+
valid = true
|
46
|
+
valid &= complain("Attribute 'email' must be specified", self) if self.email.nil?
|
47
|
+
valid
|
48
|
+
end
|
49
|
+
|
50
|
+
def notify(message, time, priority, category, host)
|
51
|
+
begin
|
52
|
+
body = Email.format.call(self.name, self.email, message, time, priority, category, host)
|
53
|
+
|
54
|
+
case Email.delivery_method
|
55
|
+
when :smtp
|
56
|
+
notify_smtp(body)
|
57
|
+
when :sendmail
|
58
|
+
notify_sendmail(body)
|
59
|
+
else
|
60
|
+
raise "unknown delivery method: #{Email.delivery_method}"
|
61
|
+
end
|
62
|
+
|
63
|
+
self.info = "sent email to #{self.email}"
|
64
|
+
rescue => e
|
65
|
+
applog(nil, :info, "failed to send email to #{self.email}: #{e.message}")
|
66
|
+
applog(nil, :debug, e.backtrace.join("\n"))
|
67
|
+
end
|
68
|
+
end
|
69
|
+
|
70
|
+
# private
|
71
|
+
|
72
|
+
def notify_smtp(mail)
|
73
|
+
args = [Email.server_settings[:address], Email.server_settings[:port]]
|
74
|
+
if Email.server_settings[:authentication]
|
75
|
+
args << Email.server_settings[:domain]
|
76
|
+
args << Email.server_settings[:user_name]
|
77
|
+
args << Email.server_settings[:password]
|
78
|
+
args << Email.server_settings[:authentication]
|
79
|
+
end
|
80
|
+
|
81
|
+
Net::SMTP.start(*args) do |smtp|
|
82
|
+
smtp.send_message mail, Email.message_settings[:from], self.email
|
83
|
+
end
|
84
|
+
end
|
85
|
+
|
86
|
+
def notify_sendmail(mail)
|
87
|
+
IO.popen("#{Email.sendmail_settings[:location]} #{Email.sendmail_settings[:arguments]}","w+") do |sm|
|
88
|
+
sm.print(mail.gsub(/\r/, ''))
|
89
|
+
sm.flush
|
90
|
+
end
|
91
|
+
end
|
92
|
+
end
|
93
|
+
|
94
|
+
end
|
95
|
+
end
|
@@ -0,0 +1,41 @@
|
|
1
|
+
module God
|
2
|
+
class DependencyGraph
|
3
|
+
attr_accessor :nodes
|
4
|
+
|
5
|
+
def initialize
|
6
|
+
self.nodes = {}
|
7
|
+
end
|
8
|
+
|
9
|
+
def add(a, b)
|
10
|
+
node_a = self.nodes[a] || Node.new(a)
|
11
|
+
node_b = self.nodes[b] || Node.new(b)
|
12
|
+
|
13
|
+
node_a.add(node_b)
|
14
|
+
|
15
|
+
self.nodes[a] ||= node_a
|
16
|
+
self.nodes[b] ||= node_b
|
17
|
+
end
|
18
|
+
end
|
19
|
+
end
|
20
|
+
|
21
|
+
module God
|
22
|
+
class DependencyGraph
|
23
|
+
class Node
|
24
|
+
attr_accessor :name
|
25
|
+
attr_accessor :dependencies
|
26
|
+
|
27
|
+
def initialize(name)
|
28
|
+
self.name = name
|
29
|
+
self.dependencies = []
|
30
|
+
end
|
31
|
+
|
32
|
+
def add(node)
|
33
|
+
self.dependencies << node unless self.dependencies.include?(node)
|
34
|
+
end
|
35
|
+
|
36
|
+
def has_node?(node)
|
37
|
+
(self == node) || self.dependencies.any { |x| x.has_node?(node) }
|
38
|
+
end
|
39
|
+
end
|
40
|
+
end
|
41
|
+
end
|
@@ -0,0 +1,37 @@
|
|
1
|
+
def start_dike
|
2
|
+
require 'dike'
|
3
|
+
Thread.new do
|
4
|
+
Dike.logfactory File.join(File.dirname(__FILE__), *%w[.. .. logs])
|
5
|
+
loop do
|
6
|
+
Dike.finger
|
7
|
+
sleep(1)
|
8
|
+
end
|
9
|
+
end
|
10
|
+
end
|
11
|
+
|
12
|
+
class BleakHouseDiagnostic
|
13
|
+
LOG_FILE = File.join(File.dirname(__FILE__), *%w[.. .. logs bleak.log])
|
14
|
+
|
15
|
+
class << self
|
16
|
+
attr_accessor :logger
|
17
|
+
end
|
18
|
+
|
19
|
+
def self.install
|
20
|
+
require 'bleak_house'
|
21
|
+
self.logger = BleakHouse::Logger.new
|
22
|
+
File.delete(LOG_FILE) rescue nil
|
23
|
+
end
|
24
|
+
|
25
|
+
def self.snapshot(name)
|
26
|
+
self.logger.snapshot(LOG_FILE, name, false) if self.logger
|
27
|
+
end
|
28
|
+
|
29
|
+
def self.spin(delay = 1)
|
30
|
+
Thread.new do
|
31
|
+
loop do
|
32
|
+
self.snapshot
|
33
|
+
sleep(delay)
|
34
|
+
end
|
35
|
+
end
|
36
|
+
end
|
37
|
+
end
|
data/lib/god/driver.rb
ADDED
@@ -0,0 +1,206 @@
|
|
1
|
+
module God
|
2
|
+
class TimedEvent
|
3
|
+
include Comparable
|
4
|
+
|
5
|
+
attr_accessor :at
|
6
|
+
|
7
|
+
# Instantiate a new TimedEvent that will be triggered after the specified delay
|
8
|
+
# +delay+ is the number of seconds from now at which to trigger
|
9
|
+
#
|
10
|
+
# Returns TimedEvent
|
11
|
+
def initialize(delay = 0)
|
12
|
+
self.at = Time.now + delay
|
13
|
+
end
|
14
|
+
|
15
|
+
def due?
|
16
|
+
Time.now >= self.at
|
17
|
+
end
|
18
|
+
|
19
|
+
def <=>(other)
|
20
|
+
self.at <=> other.at
|
21
|
+
end
|
22
|
+
end # DriverEvent
|
23
|
+
|
24
|
+
class DriverEvent < TimedEvent
|
25
|
+
attr_accessor :condition, :task
|
26
|
+
|
27
|
+
def initialize(delay, task, condition)
|
28
|
+
super delay
|
29
|
+
self.task = task
|
30
|
+
self.condition = condition
|
31
|
+
end
|
32
|
+
|
33
|
+
def handle_event
|
34
|
+
@task.handle_poll(@condition)
|
35
|
+
end
|
36
|
+
end # DriverEvent
|
37
|
+
|
38
|
+
class DriverOperation < TimedEvent
|
39
|
+
attr_accessor :task, :name, :args
|
40
|
+
|
41
|
+
def initialize(task, name, args)
|
42
|
+
super(0)
|
43
|
+
self.task = task
|
44
|
+
self.name = name
|
45
|
+
self.args = args
|
46
|
+
end
|
47
|
+
|
48
|
+
# Handle the next queued operation that was issued asynchronously
|
49
|
+
#
|
50
|
+
# Returns nothing
|
51
|
+
def handle_event
|
52
|
+
@task.send(@name, *@args)
|
53
|
+
end
|
54
|
+
end
|
55
|
+
|
56
|
+
class DriverEventQueue
|
57
|
+
def initialize
|
58
|
+
@shutdown = false
|
59
|
+
@waiting = []
|
60
|
+
@events = []
|
61
|
+
@waiting.taint
|
62
|
+
@events.taint
|
63
|
+
self.taint
|
64
|
+
end
|
65
|
+
|
66
|
+
#
|
67
|
+
# Wake any sleeping threads after setting the sentinel
|
68
|
+
#
|
69
|
+
def shutdown
|
70
|
+
@shutdown = true
|
71
|
+
begin
|
72
|
+
Thread.critical = true
|
73
|
+
@waiting.each do |t|
|
74
|
+
t.run
|
75
|
+
end
|
76
|
+
ensure
|
77
|
+
Thread.critical = false
|
78
|
+
end
|
79
|
+
end
|
80
|
+
|
81
|
+
#
|
82
|
+
# Sleep until the queue has something due
|
83
|
+
#
|
84
|
+
def pop
|
85
|
+
begin
|
86
|
+
while (Thread.critical = true; @events.empty? or !@events.first.due?)
|
87
|
+
@waiting.push Thread.current
|
88
|
+
if @events.empty?
|
89
|
+
raise ThreadError, "queue empty" if @shutdown
|
90
|
+
Thread.stop
|
91
|
+
else
|
92
|
+
Thread.critical = false
|
93
|
+
sleep @events.first.at - Time.now
|
94
|
+
Thread.critical = true
|
95
|
+
end
|
96
|
+
end
|
97
|
+
@events.shift
|
98
|
+
ensure
|
99
|
+
Thread.critical = false
|
100
|
+
end
|
101
|
+
end
|
102
|
+
|
103
|
+
alias shift pop
|
104
|
+
alias deq pop
|
105
|
+
|
106
|
+
#
|
107
|
+
# Add an event to the queue, wake any waiters if what we added needs to
|
108
|
+
# happen sooner than the next pending event
|
109
|
+
#
|
110
|
+
def push(event)
|
111
|
+
Thread.critical = true
|
112
|
+
@events << event
|
113
|
+
@events.sort!
|
114
|
+
begin
|
115
|
+
t = @waiting.shift if @events.first == event
|
116
|
+
t.wakeup if t
|
117
|
+
rescue ThreadError
|
118
|
+
retry
|
119
|
+
ensure
|
120
|
+
Thread.critical = false
|
121
|
+
end
|
122
|
+
begin
|
123
|
+
t.run if t
|
124
|
+
rescue ThreadError
|
125
|
+
end
|
126
|
+
end
|
127
|
+
|
128
|
+
alias << push
|
129
|
+
alias enq push
|
130
|
+
|
131
|
+
def empty?
|
132
|
+
@que.empty?
|
133
|
+
end
|
134
|
+
|
135
|
+
def clear
|
136
|
+
@events.clear
|
137
|
+
end
|
138
|
+
|
139
|
+
def length
|
140
|
+
@events.length
|
141
|
+
end
|
142
|
+
|
143
|
+
alias size length
|
144
|
+
|
145
|
+
def num_waiting
|
146
|
+
@waiting.size
|
147
|
+
end
|
148
|
+
end
|
149
|
+
|
150
|
+
|
151
|
+
class Driver
|
152
|
+
attr_reader :thread
|
153
|
+
|
154
|
+
# Instantiate a new Driver and start the scheduler loop to handle events
|
155
|
+
# +task+ is the Task this Driver belongs to
|
156
|
+
#
|
157
|
+
# Returns Driver
|
158
|
+
def initialize(task)
|
159
|
+
@task = task
|
160
|
+
@events = God::DriverEventQueue.new
|
161
|
+
|
162
|
+
@thread = Thread.new do
|
163
|
+
loop do
|
164
|
+
begin
|
165
|
+
@events.pop.handle_event
|
166
|
+
rescue ThreadError => e
|
167
|
+
# queue is empty
|
168
|
+
break
|
169
|
+
rescue Exception => e
|
170
|
+
message = format("Unhandled exception in driver loop - (%s): %s\n%s",
|
171
|
+
e.class, e.message, e.backtrace.join("\n"))
|
172
|
+
applog(nil, :fatal, message)
|
173
|
+
end
|
174
|
+
end
|
175
|
+
end
|
176
|
+
end
|
177
|
+
|
178
|
+
# Clear all events for this Driver
|
179
|
+
#
|
180
|
+
# Returns nothing
|
181
|
+
def clear_events
|
182
|
+
@events.clear
|
183
|
+
end
|
184
|
+
|
185
|
+
# Queue an asynchronous message
|
186
|
+
# +name+ is the Symbol name of the operation
|
187
|
+
# +args+ is an optional Array of arguments
|
188
|
+
#
|
189
|
+
# Returns nothing
|
190
|
+
def message(name, args = [])
|
191
|
+
@events.push(DriverOperation.new(@task, name, args))
|
192
|
+
end
|
193
|
+
|
194
|
+
# Create and schedule a new DriverEvent
|
195
|
+
# +condition+ is the Condition
|
196
|
+
# +delay+ is the number of seconds to delay (default: interval defined in condition)
|
197
|
+
#
|
198
|
+
# Returns nothing
|
199
|
+
def schedule(condition, delay = condition.interval)
|
200
|
+
applog(nil, :debug, "driver schedule #{condition} in #{delay} seconds")
|
201
|
+
|
202
|
+
@events.push(DriverEvent.new(delay, @task, condition))
|
203
|
+
end
|
204
|
+
end # Driver
|
205
|
+
|
206
|
+
end # God
|