[bitbake-devel] server/process: Deal more gracefully with SIGTERM

Submitted by Richard Purdie on March 9, 2014, 5:04 p.m.


Message ID 1394384681.7883.11.camel@ted
State New
Headers show

Commit Message

Richard Purdie March 9, 2014, 5:04 p.m.
Currently a SIGTERM to the UI process causes the UI simply to lock up.

By setting an exit flag, the waitEvent can raise a SIGINT, allowing the
UI to break out the event loop and exit. Currently this is results in a
traceback but that is more desirable than a hanging process.

Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>

Patch hide | download patch | download mbox

diff --git a/bitbake/lib/bb/server/process.py b/bitbake/lib/bb/server/process.py
index 6db6a23..f9074fb 100644
--- a/bitbake/lib/bb/server/process.py
+++ b/bitbake/lib/bb/server/process.py
@@ -157,6 +157,10 @@  class BitBakeProcessServerConnection(BitBakeBaseServerConnection):
         self.connection = ServerCommunicator(self.ui_channel, self.procserver.event_handle)
         self.events = self.event_queue
+    def sigterm_terminate(self):
+        bb.error("UI received SIGTERM")
+        self.terminate()
     def terminate(self):
         def flushevents():
             while True:
@@ -176,10 +180,20 @@  class BitBakeProcessServerConnection(BitBakeBaseServerConnection):
+        self.event_queue.setexit()
 # Wrap Queue to provide API which isn't server implementation specific
 class ProcessEventQueue(multiprocessing.queues.Queue):
+    def __init__(self, maxsize):
+        multiprocessing.queues.Queue.__init__(self, maxsize)
+        self.exit = False
+    def setexit(self):
+        self.exit = True
     def waitEvent(self, timeout):
+        if self.exit:
+            raise KeyboardInterrupt
             return self.get(True, timeout)
         except Empty:
@@ -214,5 +228,5 @@  class BitBakeServer(BitBakeBaseServer):
         if error:
             logger.error("Unable to set the cooker to the correct featureset: %s" % error)
             raise BaseException(error)
-        signal.signal(signal.SIGTERM, lambda i, s: self.connection.terminate())
+        signal.signal(signal.SIGTERM, lambda i, s: self.connection.sigterm_terminate())
         return self.connection