Question

我有一个在Debian中运行的Python（2.7.13）HTTP Server，我想停止任何耗时超过10秒但无法在任何地方找到解决方案的GET请求。

我已经尝试过以下问题中发布的所有代码段：How to implement Timeout in BaseHTTPServer.BaseHTTPRequestHandler Python

#!/usr/bin/env python
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
import os

class handlr(BaseHTTPRequestHandler):
    def do_GET(self):
        self.send_response(200)
        self.send_header('Content-type','text-html')
        self.end_headers()
        self.wfile.write(os.popen('sleep 20 & echo "this took 20 seconds"').read())

def run():
    server_address = ('127.0.0.1', 8080)
    httpd = HTTPServer(server_address, handlr)
    httpd.serve_forever()

if __name__ == '__main__':
    run()

作为测试，我正在运行一个需要20秒才能执行的shell命令，因此我需要在此之前停止服务器。

Answer 1

将操作放在后台线程上，然后等待后台线程完成。没有通用的安全方法来中止线程，因此，不幸的是，这种实现使该函数即使在已经放弃的情况下仍在后台运行。

如果可以的话，您可以考虑将代理服务器（例如说nginx）放在服务器之前，并让它为您处理超时，或者使用更强大的HTTP服务器实现，将其作为配置选项。但是下面的答案应该可以解决这个问题。

#!/usr/bin/env python
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
import os
import threading

class handlr(BaseHTTPRequestHandler):
    def do_GET(self):
        result, succeeded = run_with_timeout(lambda: os.popen('sleep 20 & echo "this took 20 seconds"').read(), timeout=3)
        if succeeded:
            self.send_response(200)
            self.send_header('Content-type','text-html')
            self.end_headers()
            self.wfile.write(os.popen('sleep 20 & echo "this took 20 seconds"').read())
        else:
            self.send_response(500)
            self.send_header('Content-type','text-html')
            self.end_headers()
            self.wfile.write('<html><head></head><body>Sad panda</body></html>')
        self.wfile.close()

def run():
    server_address = ('127.0.0.1', 8080)
    httpd = HTTPServer(server_address, handlr)
    httpd.serve_forever()

def run_with_timeout(fn, timeout):
    lock = threading.Lock()
    result = [None, False]
    def run_callback():
        r = fn()
        with lock:
            result[0] = r
            result[1] = True

    t = threading.Thread(target=run_callback)
    t.daemon = True
    t.start()
    t.join(timeout)
    with lock:
        return tuple(result)

if __name__ == '__main__':
    run()

如何在Python HTTPServer中设置超时

1 个答案: