Class: God::CLI::Command
- Inherits:
-
Object
- Object
- God::CLI::Command
- Defined in:
- lib/god/cli/command.rb
Instance Method Summary collapse
- #check_command ⇒ Object
- #dispatch ⇒ Object
-
#initialize(command, options, args) ⇒ Command
constructor
A new instance of Command.
- #lifecycle_command ⇒ Object
- #load_command ⇒ Object
- #log_command ⇒ Object
- #quit_command ⇒ Object
- #setup ⇒ Object
- #signal_command ⇒ Object
- #status_command ⇒ Object
- #terminate_command ⇒ Object
Constructor Details
permalink #initialize(command, options, args) ⇒ Command
Returns a new instance of Command.
5 6 7 8 9 10 11 |
# File 'lib/god/cli/command.rb', line 5 def initialize(command, , args) @command = command @options = @args = args dispatch end |
Instance Method Details
permalink #check_command ⇒ Object
[View source]
160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 |
# File 'lib/god/cli/command.rb', line 160 def check_command Thread.new do begin event_system = God::EventHandler.event_system puts "using event system: #{event_system}" if God::EventHandler.loaded? puts "starting event handler" God::EventHandler.start else puts "[fail] event system did not load" exit(1) end puts 'forking off new process' pid = fork do loop { sleep(1) } end puts "forked process with pid = #{pid}" God::EventHandler.register(pid, :proc_exit) do puts "[ok] process exit event received" exit!(0) end sleep(1) puts "killing process" ::Process.kill('KILL', pid) rescue => e puts e. puts e.backtrace.join("\n") end end sleep(2) puts "[fail] never received process exit event" exit(1) end |
permalink #dispatch ⇒ Object
[View source]
27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
# File 'lib/god/cli/command.rb', line 27 def dispatch if %w{load status signal log quit terminate}.include?(@command) setup send("#{@command}_command") elsif %w{start stop restart monitor unmonitor remove}.include?(@command) setup lifecycle_command elsif @command == 'check' check_command else puts "Command '#{@command}' is not valid. Run 'god --help' for usage" abort end end |
permalink #lifecycle_command ⇒ Object
[View source]
204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 |
# File 'lib/god/cli/command.rb', line 204 def lifecycle_command # get the name of the watch/group name = @args[1] puts "Sending '#{@command}' command" t = Thread.new { loop { sleep(1); STDOUT.print('.'); STDOUT.flush; sleep(1) } } # send @command watches = @server.control(name, @command) # output response t.kill; STDOUT.puts unless watches.empty? puts 'The following watches were affected:' watches.each do |w| puts ' ' + w end else puts 'No matching task or group' end end |
permalink #load_command ⇒ Object
[View source]
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
# File 'lib/god/cli/command.rb', line 42 def load_command file = @args[1] puts "Sending '#{@command}' command" puts unless File.exist?(file) abort "File not found: #{file}" end names, errors = *@server.running_load(File.read(file), File.(file)) # output response unless names.empty? puts 'The following tasks were affected:' names.each do |w| puts ' ' + w end end unless errors.empty? puts errors exit(1) end end |
permalink #log_command ⇒ Object
[View source]
110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 |
# File 'lib/god/cli/command.rb', line 110 def log_command begin Signal.trap('INT') { exit } name = @args[1] unless name puts "You must specify a Task or Group name" exit! end t = Time.at(0) loop do print @server.running_log(name, t) t = Time.now sleep 1 end rescue God::NoSuchWatchError puts "No such watch" rescue DRb::DRbConnError puts "The server went away" end end |
permalink #quit_command ⇒ Object
[View source]
133 134 135 136 137 138 139 140 |
# File 'lib/god/cli/command.rb', line 133 def quit_command begin @server.terminate abort 'Could not stop god' rescue DRb::DRbConnError puts 'Stopped god' end end |
permalink #setup ⇒ Object
[View source]
13 14 15 16 17 18 19 20 21 22 23 24 25 |
# File 'lib/god/cli/command.rb', line 13 def setup # connect to drb unix socket DRb.start_service("druby://127.0.0.1:0") @server = DRbObject.new(nil, God::Socket.socket(@options[:port])) # ping server to ensure that it is responsive begin @server.ping rescue DRb::DRbConnError puts "The server is not available (or you do not have permissions to access it)" abort end end |
permalink #signal_command ⇒ Object
[View source]
87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
# File 'lib/god/cli/command.rb', line 87 def signal_command # get the name of the watch/group name = @args[1] signal = @args[2] puts "Sending signal '#{signal}' to '#{name}'" t = Thread.new { loop { sleep(1); STDOUT.print('.'); STDOUT.flush; sleep(1) } } watches = @server.signal(name, signal) # output response t.kill; STDOUT.puts unless watches.empty? puts 'The following watches were affected:' watches.each do |w| puts ' ' + w end else puts 'No matching task or group' end end |
permalink #status_command ⇒ Object
[View source]
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
# File 'lib/god/cli/command.rb', line 68 def status_command watches = {} @server.status.each do |name, status| g = status[:group] || '' unless watches.has_key?(g) watches[g] = {} end watches[g][name] = status end watches.keys.sort.each do |group| puts "#{group}:" unless group.empty? watches[group].keys.sort.each do |name| state = watches[group][name][:state] print " " unless group.empty? puts "#{name}: #{state}" end end end |
permalink #terminate_command ⇒ Object
[View source]
142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 |
# File 'lib/god/cli/command.rb', line 142 def terminate_command t = Thread.new { loop { STDOUT.print('.'); STDOUT.flush; sleep(1) } } if @server.stop_all t.kill; STDOUT.puts puts 'Stopped all watches' else t.kill; STDOUT.puts puts 'Could not stop all watches within 10 seconds' end begin @server.terminate abort 'Could not stop god' rescue DRb::DRbConnError puts 'Stopped god' end end |