register producers/consumers for TCP and WebSocket

This commit is contained in:
meejah 2022-09-27 15:48:26 -06:00
parent 80e02d4a77
commit 01510de60e

View File

@ -47,6 +47,7 @@ class TransitConnection(LineReceiver):
ITransitClient API
"""
self._buddy = other
self._buddy._client.transport.registerProducer(self.transport, True)
def disconnect_partner(self):
"""
@ -198,6 +199,7 @@ class WebSocketTransitConnection(WebSocketServerProtocol):
ITransitClient API
"""
self._buddy = other
self._buddy._client.transport.registerProducer(self.transport, True)
def disconnect_partner(self):
"""