python - 如何统计 Python 扭曲框架中当前连接的协议数量

Question

我试图计算扭曲的活动协议的数量，但出现错误：

exceptions.AttributeError: Factory instance has no attribute 'numProtocols'

下面是代码：

class EchoPro(Protocol):
    def connectionMade(self):
        self.factory.numProtocols = self.factory.numProtocols+1
        if self.factory.numProtocols > 100:
            self.transport.write("Too many connections, try later")
            self.transport.loseConnection()
    def connectionLost(self, reason):
        self.factory.numProtocols = self.factory.numProtocols-1

    def dataReceived(self, data):
        self.transport.write(data)

Answer 1

那是因为self.factory不包含该numProtocols属性。

要自定义协议的工厂，您可以通过子类化为您的协议创建工厂twisted.internet.protocol.Factory。

例子：

from twisted.internet.protocol import Protocol, Factory
from twisted.internet import reactor

class Echo(Protocol):
    # ... your implementation as it is now ...

class EchoFactory(Factory):  # Factory for your protocol
    protocol = Echo
    numProtocols = 0

factory = EchoFactory()
factory.protocol = Echo

reactor.listenTCP(8007, factory)
reactor.run()

或者，您可以在创建工厂实例后对其进行修改，如 docs 中所做的那样。

例子：

from twisted.internet.protocol import Protocol, Factory
from twisted.internet import reactor

class Echo(Protocol):
    # ... your implementation as it is now ...

def getEchoFactory():
    factory = Factory()
    factory.protocol = Echo
    factory.numProtocols = 0
    return factory

reactor.listenTCP(8007, getEchoFactory())
reactor.run()