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