在一个线程的Python应用程序中关闭socketserver serve_forever（）

Question

我知道socketserver有一个shut down（）方法，它会导致服务器关闭，但这只在多线程应用程序中有效，因为关闭需要从不同的线程调用，而不是从serve_forever（）正在运行的线程调用。

我的应用程序一次只处理一个请求，因此我不使用单独的线程来处理请求，而且我无法调用shutdown（），因为它会导致死锁（它不在文档中，但直接在socketserver的源代码中声明）。

为了更好地理解，我将在这里粘贴代码的简化版本：

import socketserver

class TCPServerV4(socketserver.TCPServer):
  address_family = socket.AF_INET
  allow_reuse_address = True

class TCPHandler(socketserver.BaseRequestHandler):
  def handle(self):
    try:
       data = self.request.recv(4096)
    except keyboardInterrupt:
       server.shutdown()

server = TCPServerV4((host, port), TCPHandler)
server.server_forever()

我知道这段代码不起作用。我只想向您展示我想完成的事情-关闭服务器并退出应用程序，同时等待用户按下Ctrl-C键时输入的数据

Answer 1

您可以在处理程序中本地启动另一个线程，并从中调用shutdown。

工作演示：

    #!/usr/bin/env python
    # -*- coding: UTF-8 -*-

    import SimpleHTTPServer
    import SocketServer
    import time
    import thread

    class MyHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):
        def do_POST(self):
            if self.path.startswith('/kill_server'):
                print "Server is going down, run it again manually!"
                def kill_me_please(server):
                    server.shutdown()
                thread.start_new_thread(kill_me_please, (httpd,))
                self.send_error(500)

    class MyTCPServer(SocketServer.TCPServer):
        def server_bind(self):
            import socket
            self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
            self.socket.bind(self.server_address)

    server_address = ('', 8000)
    httpd = MyTCPServer(server_address, MyHandler)
    try:
        httpd.serve_forever()
    except KeyboardInterrupt:
        pass
    httpd.server_close()

注意事项：

1. 要终止服务器，请执行http://localhost:8000/kill_server请求。
2. 我创建了调用server.shutdown()的函数，并从另一个线程运行它来解决我们讨论的问题。
3. 我使用来自https://stackoverflow.com/a/18858817/774971的建议使socket立即可以重用（您可以再次运行服务器，而不必使用[Errno 98]地址错误）。对于stock TCPServer，您必须等待连接超时才能再次运行服务器。

Answer 2

SocketServer库使用一些奇怪的方式来处理继承的属性（由于使用了旧样式的类而猜测）。如果创建服务器并列出其受保护的属性，则会看到：

In [4]: server = SocketServer.TCPServer(('127.0.0.1',8000),Handler)
In [5]: server._

server._BaseServer__is_shut_down
server.__init__
server._BaseServer__shutdown_request
server.__module__
server.__doc__
server._handle_request_nonblock

如果您只是在请求处理程序中添加以下内容：

self.server._BaseServer__shutdown_request = True

服务器将关闭。这与调用server.shutdown()一样，只是不需要等待（并使主线程死锁）直到它关闭。

Answer 3

我相信OP的目的是从请求处理程序关闭服务器，我认为他的代码的KeyboardInterrupt方面只是令人困惑的事情。

在运行服务器的shell中按ctrl-c将成功地在不做任何特殊操作的情况下关闭它。你不能从一个不同的shell按ctrl-c并期望它工作，我认为这个概念可能是这个混乱的代码的来源。不需要像OP尝试的那样处理handle()中的KeyboardInterrupt，或者像另一个建议的那样处理serve_forever()中的KeyboardInterrupt。如果你两个都不做，它就会按预期工作。

这里唯一的诀窍——而且很棘手——是告诉服务器在不死锁的情况下从处理程序关闭。

正如OP在他的代码中解释和显示的那样，他使用的是单线程服务器，因此建议在“其他线程”中关闭它的用户没有注意到。

我仔细研究了SocketServer代码，发现BaseServer类在处理该模块中可用的线程混合时，实际上通过在serve_forever中使用threading.Event循环，使非线程服务器的使用更加困难。

因此，我为单线程服务器编写了serve_forever的修改版本，这使得从请求处理程序关闭服务器成为可能。

import SocketServer
import socket
import select

class TCPServerV4(SocketServer.TCPServer):
    address_family = socket.AF_INET
    allow_reuse_address = True

    def __init__(self, server_address, RequestHandlerClass):
        SocketServer.TCPServer.__init__(self, server_address, RequestHandlerClass)
        self._shutdown_request = False

    def serve_forever(self, poll_interval=0.5):
        """provide an override that can be shutdown from a request handler.
        The threading code in the BaseSocketServer class prevented this from working
        even for a non-threaded blocking server.
        """
        try:
            while not self._shutdown_request:
                # XXX: Consider using another file descriptor or
                # connecting to the socket to wake this up instead of
                # polling. Polling reduces our responsiveness to a
                # shutdown request and wastes cpu at all other times.
                r, w, e = SocketServer._eintr_retry(select.select, [self], [], [],
                                       poll_interval)
                if self in r:
                    self._handle_request_noblock()
        finally:
            self._shutdown_request = False

class TCPHandler(SocketServer.BaseRequestHandler):
    def handle(self):
        data = self.request.recv(4096)
        if data == "shutdown":
            self.server._shutdown_request = True

host = 'localhost'
port = 52000
server = TCPServerV4((host, port), TCPHandler)
server.serve_forever()

如果将字符串'shutdown'发送到服务器，服务器将结束其serve_forever循环。您可以使用netcat进行测试：

printf "shutdown" | nc localhost 52000