2

単純なロードバランサーを作成しようとしています。サーバーの1つ（BalanceServer）が接続を閉じない限り、正常に機能します...クライアント（ReverseProxy）は切断されますが、BalanceServerとの接続は開いたままになります。サーバーが切断されたときに接続を閉じる（clientLoseConnection）の場合と同じように、 ReverseProxy.connectionLostにコールバック（＃3）を追加して、サーバーの1つとの接続を閉じようとしましたが、その時点でServerWriterがNullであり、＃で終了できません。 1と＃2

片側が切断されたときにすべての接続が閉じていることを確認するにはどうすればよいですか？ここでも、クライアントとサーバーの1つがハングしたときに、ある種のタイムアウトが適切だと思いますが、両方の接続で機能するようにタイムアウトを追加するにはどうすればよいですか？

from twisted.internet.protocol import Protocol, Factory, ClientCreator
from twisted.internet import reactor, defer
from collections import namedtuple

BalanceServer = namedtuple('BalanceServer', 'host port')

SERVER_LIST = [BalanceServer('127.0.0.1', 8000), BalanceServer('127.0.0.1', 8001)]

def getServer(servers):
    while True:
        for server in servers:
            yield server

# this writes to one of balance servers and responds to client with callback 'clientWrite'
class ServerWriter(Protocol):
    def sendData(self, data):
        self.transport.write(data)

    def dataReceived(self, data):
        self.clientWrite(data)

    def connectionLost( self, reason ):
        self.clientLoseConnection()

# callback for reading data from client to send it to server and get response to client again    
def transferData(serverWriter, clientWrite, clientLoseConnection, data):
    if serverWriter:
        serverWriter.clientWrite = clientWrite
        serverWriter.clientLoseConnection = clientLoseConnection
        serverWriter.sendData(data)

def closeConnection(serverWriter):
    if serverWriter: #1 this is null
        #2 So connection is not closed and hangs there, till BalanceServer close it 
        serverWriter.transport.loseConnection()

# accepts clients
class ReverseProxy(Protocol):
    def connectionMade(self):
        server = self.factory.getServer()
        self.serverWriter = ClientCreator(reactor, ServerWriter)
        self.client = self.serverWriter.connectTCP( server.host, server.port )

    def dataReceived(self, data):
        self.client.addCallback(transferData, self.transport.write, 
                    self.transport.loseConnection, data )

    def connectionLost(self, reason):
        self.client.addCallback(closeConnection) #3 adding close doesn't work


class ReverseProxyFactory(Factory):
    protocol = ReverseProxy
    def __init__(self, serverGenerator):
        self.getServer = serverGenerator

plainFactory = ReverseProxyFactory( getServer(SERVER_LIST).next )
reactor.listenTCP( 7777, plainFactory )
reactor.run()
4

1 に答える 1

1

twisted.internet.protocols.portforward2つの接続を接続してから、それらを切断する例を確認することをお勧めします。または、txloadbalancerを使用して、独自のコードを記述しないでください。

ただし、loseConnectionトラフィックが通過しない場合は、接続を強制的に終了することはありません。そのため、アプリケーションレベルのpingや接続を介したデータがない場合でも、それらがシャットダウンすることはありません。 これはTwistedの長年のバグです。 実際、最も長く続いているバグです。おそらくあなたは修正に取り組むのを手伝いたいと思うでしょう:)。

于 2010-12-05T22:26:13.013 に答える