kill osrm-routed if it refuses to shutdown in cuke tests

This commit is contained in:
Emil Tin 2012-12-15 17:01:24 +01:00
parent 7f7055a9e7
commit 7e9614b9ec

View File

@ -1,83 +1,90 @@
require 'socket' require 'socket'
require 'sys/proctable' require 'sys/proctable'
LAUNCH_TIMEOUT = 5 LAUNCH_TIMEOUT = 2
SHUTDOWN_TIMEOUT = 5 SHUTDOWN_TIMEOUT = 2
class OSRMLauncher class OSRMLauncher
def initialize &block def initialize &block
Dir.chdir TEST_FOLDER do Dir.chdir TEST_FOLDER do
begin begin
begin launch
Timeout.timeout(LAUNCH_TIMEOUT) do
osrm_up
wait_for_connection
end
rescue Timeout::Error
raise "*** Launching osrm-routed timed out."
end
yield yield
ensure ensure
begin shutdown
Timeout.timeout(SHUTDOWN_TIMEOUT) do
osrm_down
end
rescue Timeout::Error
raise "*** Shutting down osrm-routed timed out."
end
end end
end end
end end
end
def osrm_up? private
if @pipe
begin def launch
Process.getpgid @pipe.pid Timeout.timeout(LAUNCH_TIMEOUT) do
true osrm_up
rescue Errno::ESRCH wait_for_connection
end
rescue Timeout::Error
raise "*** Launching osrm-routed timed out."
end
def shutdown
Timeout.timeout(SHUTDOWN_TIMEOUT) do
osrm_down
end
rescue Timeout::Error
kill
raise "*** Shutting down osrm-routed timed out."
end
def osrm_up?
if @pipe
begin
Process.getpgid @pipe.pid
true
rescue Errno::ESRCH
false
end
else
false false
end end
else
false
end end
end
def osrm_up def osrm_up
return if osrm_up? return if osrm_up?
#exec avoids popen running osrm-routed inside a shell #exec avoids popen running osrm-routed inside a shell
#if the cmd is run inside a shell, popen returns the pid for the shell, and if we try to kill it, #if the cmd is run inside a shell, popen returns the pid for the shell, and if we try to kill it,
#the child process is orphaned, and we can't terminate it. #the child process is orphaned, and we can't terminate it.
@pipe = IO.popen('exec ../osrm-routed 1>>osrm-routed.log 2>>osrm-routed.log') @pipe = IO.popen('exec ../osrm-routed 1>>osrm-routed.log 2>>osrm-routed.log')
end
def osrm_down
if @pipe
Process.kill 'TERM', @pipe.pid
wait_for_shutdown
@pipe = nil
end end
end
def kill def osrm_down
if @pipe if @pipe
Process.kill 'KILL', @pipe.pid Process.kill 'TERM', @pipe.pid
wait_for_shutdown
end
end end
end
def wait_for_connection def kill
while true if @pipe
begin Process.kill 'KILL', @pipe.pid
socket = TCPSocket.new('localhost', OSRM_PORT) end
return end
rescue Errno::ECONNREFUSED
def wait_for_connection
while true
begin
socket = TCPSocket.new('localhost', OSRM_PORT)
return
rescue Errno::ECONNREFUSED
sleep 0.1
end
end
end
def wait_for_shutdown
while osrm_up?
sleep 0.1 sleep 0.1
end end
end end
end end
def wait_for_shutdown
while osrm_up?
sleep 0.1
end
end