关闭SIG *上的SocketServer

时间:2013-07-09 14:07:09

标签: python multithreading socketserver

我对如何正确关闭我正在使用的非常简单的服务器感到困惑。

我认为这应该足够了:

#!/usr/bin/python

import signal
import myhandler
import SocketServer

def terminate(signal, frame):
    print "terminating on %s at %s"
    server.shutdown()

if __name__ == "__main__":

    signal.signal(signal.SIGTERM, terminate)

    server = SocketServer.TCPServer(("localhost", 9999), myhandler.MyHandler)

    server.serve_forever()

服务器工作正常,但是当我向其发出SIGTERM时,它只会打印terminating on 15 at ...但不会真正关闭(即关闭所有套接字并退出)。

现在pydoc解释了它

 shutdown(self)
     Stops the serve_forever loop.

     Blocks until the loop has finished. This must be called while
     serve_forever() is running in another thread, or it will
     deadlock.

但这是我迷路的地方,因为我甚至无法绕过线程编程。现在我需要一个简单的TCP echo服务器,我能够killall并且可以随时启动(由于剩余的LISTENING套接字现在无法启动)。

那么实现这个目标的正确方法是什么?

2 个答案:

答案 0 :(得分:3)

免责声明:我有0,nil,null,none,没有使用python的经验。 免责声明2 :我绝不认为您的服务器是“走的路”......任何与服务器相关的东西,甚至不是最基本的东西或学校作业以外的任何东西东西;它可能是一个很好的样本,可以帮助人们学习基础知识,但同时,在我失去数量的那么多层次上,它会产生误导和错误。

回到你的问题。我拿了你的代码并将其修改为按预期工作:

#!/usr/bin/python

import signal
import SocketServer
import threading
import thread

class DummyServer(SocketServer.BaseRequestHandler):
    def handle(self):
        data = self.request.recv(1024)
        self.request.send(data)
        return

def shutdownHandler(msg,evt):
    print "shutdown handler called. shutting down on thread id:%x"%(id(threading.currentThread()))
    server.shutdown()
    print "shutdown complete"
    evt.set()
    return

def terminate(signal,frame):
    print "terminate handle on thread id:%x"%(id(threading.currentThread()))
    t = threading.Thread(target = shutdownHandler,args = ('SIGTERM received',doneEvent))
    t.start()

if __name__ == "__main__":
    doneEvent = threading.Event()
    print "main thread id:%x"%(id(threading.currentThread()))
    signal.signal(signal.SIGTERM, terminate)
    server = SocketServer.TCPServer(("localhost",9999), DummyServer)
    server.serve_forever()
    doneEvent.wait()

您应该检查SocketServer的代码,尤其是server_forever()和shutdown()方法。您还应该尝试了解线程以及如何在它们之间进行任何类型的通信/信令。这些主题有很多好的来源。

关于线程要记住的基本要点是,一般来说,一个线程一次只能做一件事 - 你的信号处理程序是其中一个例外:)如果一个线程卡在server_forever()中,你可以'期望同一个线程也可以运行对shutdown()的调用。 Python(检查信号文档)将在主线程上运行你的信号处理程序 - 从你的代码运行server_forever()循环的那个:从信号处理程序中调用shutdown()会导致死锁,正如你所注意到的那样。

绕过它的方法是创建一个新线程,仅用于运行shutdown()。新线程的shutdown()调用将通知主线程的server_forever(),它是时候打破循环并退出。主线程甚至可以在线程运行shutdown()完成之前结束 - 一般来说,当主线程结束时,任何其他线程也会突然被杀死而没有机会完成他们正在做的任何事情。

doneEvent甚至可以确保主线程等待(doneEvent.wait())直到关闭线程完成它的工作 - 在退出之前打印“shutdown complete”。

答案 1 :(得分:1)

作为一个简单的解决方案,您可以在server_close()之后调用serve_forever()

import socketserver

class StoppableServer(socketserver.TCPServer):
    def run(self):
        try:
            self.serve_forever()
        except KeyboardInterrupt:
            pass
        finally:
            # Clean-up server (close socket, etc.)
            self.server_close()

服务器可以使用 Ctrl + C 或SIGTERM停止:

server = StoppableServer(("127.0.0.1", 8080), socketserver.BaseRequestHandler)
server.run()

服务器在线程中运行:

server = StoppableServer(("127.0.0.1", 8080), socketserver.BaseRequestHandler)

thread = threading.Thread(None, server.run)
thread.start()

# ... do things ...

server.shutdown()
thread.join()