GlcoinCore/squeaknode
1
0
Fork 0
mirror of https://github.com/yzernik/squeaknode.git synced 2026-08-18 13:09:08 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix remove peer on disconnect (#912)

Browse source
This commit is contained in:
Jonathan Zernik 2021-08-11 15:39:27 -07:00 • committed by GitHub
parent afa8ccd144
commit 1fb353b5eb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
squeaknode/network/connection.py
View file

@ -86,12 +86,14 @@ class Connection():
def open_connection(self):
logger.debug(
'Starting handshake connection with peer ... {}'.format(self.peer))
self.connection_manager.add_peer(self.peer)
self.handshake()
logger.debug('Peer connection added... {}'.format(self.peer))
yield self
self.connection_manager.remove_peer(self.peer)
logger.debug('Peer connection removed... {}'.format(self.peer))
try:
self.connection_manager.add_peer(self.peer)
self.handshake()
logger.debug('Peer connection added... {}'.format(self.peer))
yield self
finally:
self.connection_manager.remove_peer(self.peer)
logger.debug('Peer connection removed... {}'.format(self.peer))
# def __enter__(self):
# logger.debug(