如何计算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属性.

要自定义协议的工厂,可以通过子类化为协议创建Factory 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()

或者,您可以在创建工厂实例后修改它,就像在文档中一样.

例:

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()