Signed-off-by: Guido Trotter <ultrot...@google.com>
---
 daemons/ganeti-masterd |   14 +++++++-------
 1 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/daemons/ganeti-masterd b/daemons/ganeti-masterd
index 57fa9ec..0021b07 100755
--- a/daemons/ganeti-masterd
+++ b/daemons/ganeti-masterd
@@ -108,14 +108,14 @@ class IOServer(SocketServer.UnixStreamServer):
     """
     self.request_workers.AddTask(self, request, client_address)
 
-  def serve_forever(self):
+  @utils.SignalHandled([signal.SIGINT, signal.SIGTERM])
+  def serve_forever(self, signal_handlers=None):
     """Handle one request at a time until told to quit."""
-    sighandler = utils.SignalHandler([signal.SIGINT, signal.SIGTERM])
-    try:
-      while not sighandler.called:
-        self.handle_request()
-    finally:
-      sighandler.Reset()
+    # We know we only have one SignalHandled decorator, so the SignalHandler
+    # object is common.
+    sighandler = signal_handlers.values()[0]
+    while not sighandler.called:
+      self.handle_request()
 
   def server_cleanup(self):
     """Cleanup the server.
-- 
1.5.6.5

Reply via email to