def handleSigTerm(*args):
global listeners
global globalIsTerminating
# sys.stderr.write('CALLED\n')
if globalIsTerminating is True:
return # Already terminating
globalIsTerminating = True
logerr('Caught signal, shutting down listeners...\n')
for listener in listeners:
try:
os.kill(listener.pid, signal.SIGTERM)
except:
pass
logerr('Sent signal to children, waiting up to 4 seconds then trying to clean up\n')
time.sleep(1)
startTime = time.time()
remainingListeners = listeners
remainingListeners2 = []
for listener in remainingListeners:
logerr('Waiting on %d...\n' %(listener.pid,))
listener.join(.05)
if listener.is_alive() is True:
remainingListeners2.append(listener)
remainingListeners = remainingListeners2
logerr('Remaining (%d) listeners are: %s\n' %(len(remainingListeners), [listener.pid for listener in remainingListeners]))
afterJoinTime = time.time()
if remainingListeners:
delta = afterJoinTime - startTime
remainingSleep = int(GRACEFUL_SHUTDOWN_TIME - math.floor(afterJoinTime - startTime))
if remainingSleep > 0:
anyAlive = False
# If we still have time left, see if we are just done or if there are children to clean up using remaining time allotment
if threading.activeCount() > 1 or len(multiprocessing.active_children()) > 0:
logerr('Listener closed in %1.2f seconds. Waiting up to %d seconds before terminating.\n' %(delta, remainingSleep))
thisThread = threading.current_thread()
for i in range(remainingSleep):
allThreads = threading.enumerate()
anyAlive = False
for thread in allThreads:
if thread is thisThread or thread.name == 'MainThread':
continue
thread.join(.05)
if thread.is_alive() == True:
anyAlive = True
allChildren = multiprocessing.active_children()
for child in allChildren:
child.join(.05)
if child.is_alive() == True:
anyAlive = True
if anyAlive is False:
break
time.sleep(1)
if anyAlive is True:
logerr('Could not kill in time.\n')
else:
logerr('Shutdown successful after %1.2f seconds.\n' %( time.time() - startTime))
else:
logerr('Listener timed out in closing, exiting uncleanly.\n')
time.sleep(.05) # Why not? :P
logmsg('exiting...\n')
signal.signal(signal.SIGTERM, signal.SIG_DFL)
signal.signal(signal.SIGINT, signal.SIG_DFL)
sys.exit(0)
os.kill(os.getpid(), signal.SIGTERM)
return 0
def handleSigTerm(*args):
global listeners
global globalIsTerminating
# sys.stderr.write('CALLED\n')
if globalIsTerminating is True:
return # Already terminating
globalIsTerminating = True
logerr('Caught signal, shutting down listeners...\n')
for listener in listeners:
try:
os.kill(listener.pid, signal.SIGTERM)
except:
pass
logerr('Sent signal to children, waiting up to 4 seconds then trying to clean up\n')
time.sleep(1)
startTime = time.time()
remainingListeners = listeners
remainingListeners2 = []
for listener in remainingListeners:
logerr('Waiting on %d...\n' %(listener.pid,))
listener.join(.05)
if listener.is_alive() is True:
remainingListeners2.append(listener)
remainingListeners = remainingListeners2
logerr('Remaining (%d) listeners are: %s\n' %(len(remainingListeners), [listener.pid for listener in remainingListeners]))
afterJoinTime = time.time()
if remainingListeners:
delta = afterJoinTime - startTime
remainingSleep = int(GRACEFUL_SHUTDOWN_TIME - math.floor(afterJoinTime - startTime))
if remainingSleep > 0:
anyAlive = False
# If we still have time left, see if we are just done or if there are children to clean up using remaining time allotment
if threading.activeCount() > 1 or len(multiprocessing.active_children()) > 0:
logerr('Listener closed in %1.2f seconds. Waiting up to %d seconds before terminating.\n' %(delta, remainingSleep))
thisThread = threading.current_thread()
for i in range(remainingSleep):
allThreads = threading.enumerate()
anyAlive = False
for thread in allThreads:
if thread is thisThread or thread.name == 'MainThread':
continue
thread.join(.05)
if thread.is_alive() == True:
anyAlive = True
allChildren = multiprocessing.active_children()
for child in allChildren:
child.join(.05)
if child.is_alive() == True:
anyAlive = True
if anyAlive is False:
break
time.sleep(1)
if anyAlive is True:
logerr('Could not kill in time.\n')
else:
logerr('Shutdown successful after %1.2f seconds.\n' %( time.time() - startTime))
else:
logerr('Listener timed out in closing, exiting uncleanly.\n')
time.sleep(.05) # Why not? :P
logmsg('exiting...\n')
signal.signal(signal.SIGTERM, signal.SIG_DFL)
signal.signal(signal.SIGINT, signal.SIG_DFL)
sys.exit(0)
os.kill(os.getpid(), signal.SIGTERM)
return 0
if not configFilename:
sys.stderr.write('No config file provided\n\n')
printUsage(sys.stderr)
sys.exit(1)
pumpkinConfig = PumpkinConfig(configFilename)
try:
pumpkinConfig.parse()
except:
traceback.print_exc(file=sys.stderr)
printConfigHelp(sys.stderr)
sys.exit(1)
bufferSize = pumpkinConfig.getOptionValue('buffer_size')
logmsg('Configured buffer size = %d bytes\n' %(bufferSize,))
mappings = pumpkinConfig.getMappings()
listeners = []
for mappingAddr, mapping in mappings.iteritems():
logmsg('Starting up listener on %s:%d with mappings: %s\n' %(mapping.localAddr, mapping.localPort, str(mapping.workers)))
listener = PumpkinListener(mapping.localAddr, mapping.localPort, mapping.workers, bufferSize)
listener.start()
listeners.append(listener)
globalIsTerminating = False
def handleSigTerm(*args):
global listeners
global globalIsTerminating
sys.exit(1)
pumpkinConfig = PumpkinConfig(configFilename)
try:
pumpkinConfig.parse()
except PumpkinConfigException as configError:
sys.stderr.write(str(configError) + '\n\n\n')
printConfigHelp()
sys.exit(1)
except Exception as e:
traceback.print_exc(file=sys.stderr)
printConfigHelp(sys.stderr)
sys.exit(1)
bufferSize = pumpkinConfig.getOptionValue('buffer_size')
logmsg('Configured buffer size = %d bytes\n' %(bufferSize,))
mappings = pumpkinConfig.getMappings()
listeners = []
for mappingAddr, mapping in list(mappings.items()):
logmsg('Starting up listener on %s:%d with mappings: %s\n' %(mapping.localAddr, mapping.localPort, str(mapping.workers)))
listener = PumpkinListener(mapping.localAddr, mapping.localPort, mapping.workers, bufferSize)
listener.start()
listeners.append(listener)
globalIsTerminating = False
def handleSigTerm(*args):
global listeners
global globalIsTerminating
