python - 是否可以在 Twisted 中的套接字上设置超时？

Question

我意识到我可能只是愚蠢并且错过了一些重要而重要的事情，但我无法弄清楚如何使用reactor.listenUDP指定扭曲的超时。我的目标是能够指定超时，并且在所述时间之后，如果 DatagramProtocol.datagramReceived 没有被执行，让它执行一个回调或者我可以用来调用 reactor.stop() 的东西。任何帮助或建议表示赞赏。谢谢

Answer 1

我认为reactor.callLater会比LoopingCall. 像这样的东西：

class Protocol(DatagramProtocol):
    def __init__(self, timeout):
        self.timeout = timeout

    def datagramReceived(self, datagram):
        self.timeout.cancel()
        # ...

timeout = reactor.callLater(5, timedOut)
reactor.listenUDP(Protocol(timeout))

Answer 2

由于 Twisted 是事件驱动的，因此您本身不需要超时。您只需要在收到数据报时设置一个状态变量（如 datagramRecieved）并注册一个检查状态变量的循环调用，在适当的情况下停止反应器然后清除状态变量：

from twisted.internet import task
from twisted.internet import reactor

datagramRecieved = False
timeout = 1.0 # One second

# UDP code here

def testTimeout():
    global datagramRecieved
    if not datagramRecieved:
        reactor.stop()
    datagramRecieved = False


l = task.LoopingCall(testTimeout)
l.start(timeout) # call every second

# l.stop() will stop the looping calls
reactor.run()

Answer 3

对于反应堆，我们必须使用 callLater。connectionMade 时开始超时倒计时。当 lineReceived 时重置超时倒计时。

这是

# -*- coding: utf-8 -*-

from twisted.internet.protocol import Factory
from twisted.protocols.basic import LineReceiver
from twisted.internet import reactor, defer

_timeout = 27


class ServiceProtocol(LineReceiver):

    def __init__(self, users):
        self.users = users


    def connectionLost(self, reason):
        if self.users.has_key(self.name):
            del self.users[self.name]

    def timeOut(self):
        if self.users.has_key(self.name):
            del self.users[self.name]
        self.sendLine("\nOUT: 9 - Disconnected, reason: %s" % 'Connection Timed out')
        print "%s - Client disconnected: %s. Reason: %s" % (datetime.now(), self.client_ip, 'Connection Timed out' )
        self.transport.loseConnection()

    def connectionMade(self):
        self.timeout = reactor.callLater(_timeout, self.timeOut)

        self.sendLine("\nOUT: 7 - Welcome to CAED")

    def lineReceived(self, line):
        # a simple timeout procrastination
        self.timeout.reset(_timeout)

class ServFactory(Factory):

    def __init__(self):
        self.users = {} # maps user names to Chat instances

    def buildProtocol(self, addr):
        return ServiceProtocol(self.users)

port = 8123
reactor.listenTCP(port, ServFactory())
print "Started service at port %d\n" % port
reactor.run()

Answer 4

一个更好的方法是使用twisted.protocols.policies.TimeoutMixin. 它本质上是在做 acallLater但抽象成Mixin.