From be0637843766b4c6118da9af2bb5c49019418a45 Mon Sep 17 00:00:00 2001 From: Victor Shyba Date: Fri, 10 Jun 2022 17:15:13 -0300 Subject: [PATCH] add method for getting the node_id from a known peer on peer manager --- lbry/dht/peer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lbry/dht/peer.py b/lbry/dht/peer.py index c5a9c9e84..8fef68181 100644 --- a/lbry/dht/peer.py +++ b/lbry/dht/peer.py @@ -100,6 +100,9 @@ class PeerManager: self._node_id_reverse_mapping[node_id] = (address, udp_port) self.peer_manager_keys_metric.labels("global").set(self.count_cache_keys()) + def get_node_id_for_endpoint(self, address, port): + return self._node_id_mapping.get((address, port)) + def prune(self): # TODO: periodically call this now = self._loop.time() to_pop = []