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

better comment

Browse Source
This commit is contained in:
meejah 2021-04-14 16:53:37 -06:00
parent 319145608d
commit 95a72e6ac9
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/wormhole_transit_relay/server_state.py
View File

@ -246,6 +246,10 @@ class TransitServerState(object):
def _remember_client(self, client): def _remember_client(self, client):
self._client = client self._client = client
# note that there is no corresponding "_forget_client" because we
# may still want to access it after it is gone .. for example, to
# get the .started_time for logging purposes
@_machine.output() @_machine.output()
def _register_token(self, token): def _register_token(self, token):
return self._real_register_token_for_side(token, None) return self._real_register_token_for_side(token, None)
@ -388,13 +392,10 @@ class TransitServerState(object):
Terminal state Terminal state
""" """
# need a listening.upon(connection_lost) for special websocket
# case where handshake fails?
listening.upon( listening.upon(
connection_made, connection_made,
enter=wait_relay, enter=wait_relay,
outputs=[_remember_client], # XXX need _forget_client ? outputs=[_remember_client],
) )
listening.upon( listening.upon(
connection_lost, connection_lost,