如何在python Klein中设置服务器超时?

时间:2016-09-20 15:04:09

标签: python web-services klein-mvc

我正在使用python Klein http://klein.readthedocs.io/en/latest/来设置Web服务。我检查了文档,但我仍然不知道如何设置服务的超时。任何更熟悉工具的人都能说明如何将超时设置为15秒吗?谢谢!

1 个答案:

答案 0 :(得分:1)

在设置的超时间隔后,您可以调用Request.loseConnection()删除与客户端的请求连接。这是一个简单的例子:

from twisted.internet import reactor, task, defer
from klein import Klein

app = Klein()
request_timeout = 10 # seconds

@app.route('/delayed/<int:n>')
@defer.inlineCallbacks
def timeoutRequest(request, n):
    work = serverTask(n)       # work that might take too long

    drop = reactor.callLater(
        request_timeout,    # drop request connection after n seconds
        dropRequest,        # function to drop request connection
            request,        # pass request obj into dropRequest()
            work)           # pass worker deferred obj to dropRequest()

    try:
        result = yield work     # work has completed, get result
        drop.cancel()           # cancel the task to drop the request connection
    except:
        result = 'Request dropped'

    defer.returnValue(result)

def serverTask(n):
    """
    A simulation of a task that takes n number of seconds to complete.
    """
    d = task.deferLater(reactor, n, lambda: 'delayed for %d seconds' % (n))
    return d

def dropRequest(request, deferred):
    """
    Drop the request connection and cancel any deferreds
    """
    request.loseConnection()
    deferred.cancel()

app.run('localhost', 9000)

要尝试此操作,请转到http://localhost:9000/delayed/2然后http://localhost:9000/delayed/20以测试任务未及时完成的​​方案。不要忘记取消与此请求相关的所有任务,延期,线程等,否则可能会浪费大量内存。

代码说明

服务器端任务:客户端以指定的延迟值转到/delayed/<n>端点。服务器端任务(serverTask())启动,为了简单起见并模拟繁忙任务,deferLater用于在n秒后返回字符串。

请求超时:使用callLater函数,在request_timeout间隔后,调用dropRequest函数并传递request并将所有工作延迟需要取消(在这种情况下只有work)。当request_timeout通过后,请求连接将被关闭(request.loseConnection()),延迟将被取消(deferred.cancel)。

Yield服务器任务结果:在try / except块中,当值可用时将产生结果,或者如果超时已经过去且连接断开,则会发生错误并且将返回Request dropped消息。

替代

这似乎不是一个理想的场景,如果可能应该避免,但我可以看到需要这种功能。此外,尽管很少见,但请记住loseConnection并不总是完全关闭连接(这是由于TCP实现并没有那么多Twisted)。更好的解决方案是在客户端断开连接时取消服务器端任务(这可能更容易捕获)。这可以通过将addErrback附加到Request.notifyFinish()来完成。以下是仅使用Twisted(http://twistedmatrix.com/documents/current/web/howto/web-in-60/interrupted.html)的示例。