open.source/magic-wormhole-transit-relay
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

register producers/consumers for TCP and WebSocket

Browse Source
This commit is contained in:
meejah 2022-09-27 15:48:26 -06:00
parent 80e02d4a77
commit 01510de60e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/wormhole_transit_relay/transit_server.py
View File

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