AaronFiore/electrum
1
0
Fork 0
Code Activity

channel_db: update channels_for_node when removing channel

Browse Source
This commit is contained in:
ThomasV
2019-07-04 10:50:23 +02:00
parent 2be68ac4d2
commit 32fcad5bc3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
electrum/channel_db.py
View File

@@ -477,7 +477,7 @@ class ChannelDB(SqlDB):
def prune_orphaned_channels(self): def prune_orphaned_channels(self):
l = self.get_orphaned_channels() l = self.get_orphaned_channels()
for k in l: for k in l:
self._channels.pop(k) self.remove_channel(k)
self.update_counts() self.update_counts()
if l: if l:
self.logger.info(f'Deleting {len(l)} orphaned channels') self.logger.info(f'Deleting {len(l)} orphaned channels')
@@ -490,7 +490,10 @@ class ChannelDB(SqlDB):
self._channel_updates_for_private_channels[(start_node_id, short_channel_id)] = msg_payload self._channel_updates_for_private_channels[(start_node_id, short_channel_id)] = msg_payload
def remove_channel(self, short_channel_id): def remove_channel(self, short_channel_id):
self._channels.pop(short_channel_id, None) channel_info = self._channels.pop(short_channel_id, None)
if channel_info:
self._channels_for_node[channel_info.node1_id].remove(channel_info.short_channel_id)
self._channels_for_node[channel_info.node2_id].remove(channel_info.short_channel_id)
def get_node_addresses(self, node_id): def get_node_addresses(self, node_id):
return self._addresses.get(node_id) return self._addresses.get(node_id)