区分双绞线中的传出和传入连接

Question

我正在开发基于 Twisted 框架的 P2P 应用程序。因此，我可以同时拥有传入和传出连接。有没有简单的方法来区分它们？目前我只是创建另一个工厂，将连接标记为传出并将所有工厂调用委托给原始工厂，但必须有更简单的方法。

class OutgoingProtocolFactory(MyProtocolFactory):
    """
    A rather simple factory that is used to earmark connections as outgoing.
    """
    def __init__(self, parentFactory):
        self.factory = parentFactory

    def buildProtocol(self, addr):
        connection = MyProtocolFactory.buildProtocol(self.factory, addr)
        connection.factory = self.factory
        connection.incoming = False
        return connection

    def clientConnectionFailed(self, connector, reason):
        self.factory.clientConnectionFailed(connector, reason)

    def clientConnectionLost(self, connector, reason):
        self.factory.clientConnectionLost(connector, reason)

有什么想法吗？

Answer 1

由 Twisted 中包含的任何工厂构造的每个协议实例都会获得一个免费的 factory 属性，该属性引用创建它们的工厂。

按照惯例，接受的（服务器）连接有一个ServerFactory（或子类）实例作为它们的工厂，连接的（客户端）连接有一个ClientFactory（或子类）实例。

但是，如果您真的希望您的代码完全通用，则不能依赖它，最好自己跟踪它，就像您目前正在做的那样。

如果您目前用于将调用从一个工厂转移到另一个工厂的代码过于繁琐，这里有一个例子来说明它的简单程度：

from twisted.internet.protocol import ClientFactory
from twisted.protocols.policies import WrappingFactory

IN, OUT = 1, 2

class YourFactory(ClientFactory):
    # Your application logic
    ...

class Incoming(WrappingFactory):
    direction = IN

    def buildProtocol(self, addr):
        protocol = self.wrappedFactory.buildProtocol(addr)
        protocol.direction = self.direction
        return protocol

class Outgoing(Incoming):
    direction = OUT

yourFactory = YourFactory(...)
reactor.listenTCP(0, Incoming(yourFactory))
reactor.connectTCP('example.com', 1234, Outgoing(yourFactory))

如果这对你来说仍然太复杂，那么我不知道该说什么。我不够聪明，无法想出更简单的解决方案。