我是 Twisted 的新手,我正在尝试弄清楚如何实现以下内容。我有一个服务器,它从客户端接收消息。但是,此服务器在收到消息后将消息从客户端发送到另一台服务器。所以它看起来像:
Client ---> Server1 ---> Server2
因此,Server1 本质上既充当服务器又充当客户端。但是,在 Server1 向 Server2 发送信息后,我想断开 Server1 与 Server2 的连接。我不确定我该怎么做。
我现在工作的是客户端向 Server1 发送信息。然后我稍微修改输入,然后reactor.connectTCP()
成功连接并将信息发送到 Server2。我的问题是如何关闭连接而不必完全关闭 Server1。我尝试使用transport.loseConnection( )
,但这会在与 Server2 断开连接时关闭 Server1。
我正在考虑reactor.spawnProcess()
以某种方式使用,但我无法让它工作。据我所知,当我关闭连接时,它会关闭进程,所以如果我可以与另一个进程进行 connectTCP,它不应该影响其他进程。
这是我的代码
import time, datetime
import re
from twisted.internet import stdio, reactor, protocol
from twisted.protocols import basic
result = 'The AT message is unavailable (no previous talk with client)'
class DataForwardingProtocol(protocol.Protocol):
def __init__(self):
self.output = None
self.normalizeNewlines = False
def dataReceived(self, data):
if self.normalizeNewlines:
data = re.sub(r"(\r\n|\n)", "\r\n", data)
if self.output:
self.output.write(data)
class StdioProxyProtocol(DataForwardingProtocol):
global result
def connectionMade(self):
inputForwarder = DataForwardingProtocol()
inputForwarder.output = self.transport
inputForwarder.normalizeNewlines = True
stdioWrapper = stdio.StandardIO(inputForwarder)
self.output = stdioWrapper
self.transport.write(result)
self.transport.loseConnection( )
class StdioProxyFactory(protocol.ClientFactory):
protocol = StdioProxyProtocol
def clientConnectionLost(self, transport, reason):
reactor.stop()
def clientConnectionFailed(self, transport, reason):
print reason.getErrorMessage()
reactor.stop()
class EchoProtocol(basic.LineReceiver):
def dataReceived(self, line):
#Do stuff with the input sent from the client. This is irrelevant to my problem.
#UPDATE OTHER SERVERS
reactor.connectTCP('localhost', 12771, StdioProxyFactory())
class EchoServerFactory(protocol.ServerFactory):
protocol = EchoProtocol
if __name__ == "__main__":
port = 12770
reactor.listenTCP(port, EchoServerFactory( ))
reactor.run( )
谢谢!