Class: God::Process
- Inherits:
-
Object
- Object
- God::Process
- Defined in:
- lib/god/process.rb
Constant Summary collapse
- WRITES_PID =
[:start, :restart]
Instance Attribute Summary collapse
-
#gid ⇒ Object
Returns the value of attribute gid.
-
#log ⇒ Object
Returns the value of attribute log.
-
#name ⇒ Object
Returns the value of attribute name.
-
#restart ⇒ Object
Returns the value of attribute restart.
-
#start ⇒ Object
Returns the value of attribute start.
-
#stop ⇒ Object
Returns the value of attribute stop.
-
#uid ⇒ Object
Returns the value of attribute uid.
Instance Method Summary collapse
- #alive? ⇒ Boolean
- #call_action(action) ⇒ Object
- #default_pid_file ⇒ Object
-
#ensure_stop ⇒ Object
Ensure that a stop command actually stops the process.
- #file_writable?(file) ⇒ Boolean
-
#initialize ⇒ Process
constructor
A new instance of Process.
-
#pid ⇒ Object
Fetch the PID from pid_file.
- #pid_file ⇒ Object
-
#pid_file=(value) ⇒ Object
DON’T USE THIS INTERNALLY.
- #restart! ⇒ Object
-
#spawn(command) ⇒ Object
Fork/exec the given command, returns immediately
command
is the String containing the shell command. - #start! ⇒ Object
- #stop! ⇒ Object
- #valid? ⇒ Boolean
Constructor Details
#initialize ⇒ Process
Returns a new instance of Process.
7 8 9 10 11 12 13 14 |
# File 'lib/god/process.rb', line 7 def initialize self.log = '/dev/null' @pid_file = nil @tracking_pid = true @user_log = false @pid = nil end |
Instance Attribute Details
#gid ⇒ Object
Returns the value of attribute gid.
5 6 7 |
# File 'lib/god/process.rb', line 5 def gid @gid end |
#log ⇒ Object
Returns the value of attribute log.
5 6 7 |
# File 'lib/god/process.rb', line 5 def log @log end |
#name ⇒ Object
Returns the value of attribute name.
5 6 7 |
# File 'lib/god/process.rb', line 5 def name @name end |
#restart ⇒ Object
Returns the value of attribute restart.
5 6 7 |
# File 'lib/god/process.rb', line 5 def restart @restart end |
#start ⇒ Object
Returns the value of attribute start.
5 6 7 |
# File 'lib/god/process.rb', line 5 def start @start end |
#stop ⇒ Object
Returns the value of attribute stop.
5 6 7 |
# File 'lib/god/process.rb', line 5 def stop @stop end |
#uid ⇒ Object
Returns the value of attribute uid.
5 6 7 |
# File 'lib/god/process.rb', line 5 def uid @uid end |
Instance Method Details
#alive? ⇒ Boolean
16 17 18 19 20 21 22 |
# File 'lib/god/process.rb', line 16 def alive? if self.pid System::Process.new(self.pid).exists? else false end end |
#call_action(action) ⇒ Object
158 159 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 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 |
# File 'lib/god/process.rb', line 158 def call_action(action) command = send(action) if action == :stop && command.nil? pid = self.pid name = self.name command = lambda do applog(self, :info, "#{self.name} stop: default lambda killer") ::Process.kill('TERM', pid) rescue nil applog(self, :info, "#{self.name} sent SIGTERM") # Poll to see if it's dead 5.times do begin ::Process.kill(0, pid) rescue Errno::ESRCH # It died. Good. applog(self, :info, "#{self.name} process stopped") return end sleep 1 end ::Process.kill('KILL', pid) rescue nil applog(self, :info, "#{self.name} still alive; sent SIGKILL") end end if command.kind_of?(String) pid = nil if @tracking_pid # double fork god-daemonized processes # we don't want to wait for them to finish r, w = IO.pipe begin opid = fork do STDOUT.reopen(w) r.close pid = self.spawn(command) puts pid.to_s # send pid back to forker end ::Process.waitpid(opid, 0) w.close pid = r.gets.chomp ensure # make sure the file descriptors get closed no matter what r.close rescue nil w.close rescue nil end else # single fork self-daemonizing processes # we want to wait for them to finish pid = self.spawn(command) status = ::Process.waitpid2(pid, 0) exit_code = status[1] >> 8 if exit_code != 0 applog(self, :warn, "#{self.name} #{action} command exited with non-zero code = #{exit_code}") end ensure_stop if action == :stop end if @tracking_pid or (@pid_file.nil? and WRITES_PID.include?(action)) File.open(default_pid_file, 'w') do |f| f.write pid end @tracking_pid = true @pid_file = default_pid_file end elsif command.kind_of?(Proc) # lambda command command.call else raise NotImplementedError end end |
#default_pid_file ⇒ Object
154 155 156 |
# File 'lib/god/process.rb', line 154 def default_pid_file File.join(God.pid_file_directory, "#{self.name}.pid") end |
#ensure_stop ⇒ Object
Ensure that a stop command actually stops the process. Force kill if necessary.
Returns nothing
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 |
# File 'lib/god/process.rb', line 267 def ensure_stop unless self.pid applog(self, :warn, "#{self.name} stop called but pid is uknown") return end # Poll to see if it's dead 10.times do begin ::Process.kill(0, self.pid) rescue Errno::ESRCH # It died. Good. return end sleep 1 end # last resort ::Process.kill('KILL', self.pid) rescue nil applog(self, :warn, "#{self.name} process still running 10 seconds after stop command returned. Force killing.") end |
#file_writable?(file) ⇒ Boolean
24 25 26 27 28 29 30 31 32 33 34 |
# File 'lib/god/process.rb', line 24 def file_writable?(file) pid = fork do ::Process::Sys.setgid(Etc.getgrnam(self.gid).gid) if self.gid ::Process::Sys.setuid(Etc.getpwnam(self.uid).uid) if self.uid File.writable?(file) ? exit(0) : exit(1) end wpid, status = ::Process.waitpid2(pid) status.exitstatus == 0 ? true : false end |
#pid ⇒ Object
Fetch the PID from pid_file. If the pid_file does not exist, then use the PID from the last time it was read. If it has never been read, then return nil.
Returns Integer(pid) or nil
130 131 132 133 134 135 136 137 138 139 140 |
# File 'lib/god/process.rb', line 130 def pid contents = File.read(self.pid_file).strip rescue '' real_pid = contents =~ /^\d+$/ ? contents.to_i : nil if real_pid @pid = real_pid real_pid else @pid end end |
#pid_file ⇒ Object
121 122 123 |
# File 'lib/god/process.rb', line 121 def pid_file @pid_file ||= default_pid_file end |
#pid_file=(value) ⇒ Object
DON’T USE THIS INTERNALLY. Use the instance variable. – Kev No really, trust me. Use the instance variable.
110 111 112 113 114 115 116 117 118 119 |
# File 'lib/god/process.rb', line 110 def pid_file=(value) # if value is nil, do the right thing if value @tracking_pid = false else @tracking_pid = true end @pid_file = value end |
#restart! ⇒ Object
150 151 152 |
# File 'lib/god/process.rb', line 150 def restart! call_action(:restart) end |
#spawn(command) ⇒ Object
Fork/exec the given command, returns immediately
+command+ is the String containing the shell command
Returns nothing
245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 |
# File 'lib/god/process.rb', line 245 def spawn(command) fork do ::Process.setsid ::Process::Sys.setgid(Etc.getgrnam(self.gid).gid) if self.gid ::Process::Sys.setuid(Etc.getpwnam(self.uid).uid) if self.uid Dir.chdir "/" $0 = command STDIN.reopen "/dev/null" STDOUT.reopen self.log, "a" STDERR.reopen STDOUT # close any other file descriptors 3.upto(256){|fd| IO::new(fd).close rescue nil} exec command unless command.empty? end end |
#start! ⇒ Object
142 143 144 |
# File 'lib/god/process.rb', line 142 def start! call_action(:start) end |
#stop! ⇒ Object
146 147 148 |
# File 'lib/god/process.rb', line 146 def stop! call_action(:stop) end |
#valid? ⇒ Boolean
36 37 38 39 40 41 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 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
# File 'lib/god/process.rb', line 36 def valid? # determine if we're tracking pid or not self.pid_file valid = true # a start command must be specified if self.start.nil? valid = false applog(self, :error, "No start command was specified") end # self-daemonizing processes must specify a stop command if !@tracking_pid && self.stop.nil? valid = false applog(self, :error, "No stop command was specified") end # uid must exist if specified if self.uid begin Etc.getpwnam(self.uid) rescue ArgumentError valid = false applog(self, :error, "UID for '#{self.uid}' does not exist") end end # gid must exist if specified if self.gid begin Etc.getgrnam(self.gid) rescue ArgumentError valid = false applog(self, :error, "GID for '#{self.gid}' does not exist") end end # pid dir must exist if specified if !@tracking_pid && !File.exist?(File.dirname(self.pid_file)) valid = false applog(self, :error, "PID file directory '#{File.dirname(self.pid_file)}' does not exist") end # pid dir must be writable if specified if !@tracking_pid && File.exist?(File.dirname(self.pid_file)) && !file_writable?(File.dirname(self.pid_file)) valid = false applog(self, :error, "PID file directory '#{File.dirname(self.pid_file)}' is not writable by #{self.uid || Etc.getlogin}") end # log dir must exist if !File.exist?(File.dirname(self.log)) valid = false applog(self, :error, "Log directory '#{File.dirname(self.log)}' does not exist") end # log file or dir must be writable if File.exist?(self.log) unless file_writable?(self.log) valid = false applog(self, :error, "Log file '#{self.log}' exists but is not writable by #{self.uid || Etc.getlogin}") end else unless file_writable?(File.dirname(self.log)) valid = false applog(self, :error, "Log directory '#{File.dirname(self.log)}' is not writable by #{self.uid || Etc.getlogin}") end end valid end |