diff --git a/libp2p/peer/peerstore.py b/libp2p/peer/peerstore.py index 8111f08..03edffb 100644 --- a/libp2p/peer/peerstore.py +++ b/libp2p/peer/peerstore.py @@ -203,8 +203,8 @@ class PeerStore(IPeerStore): :param peer_id: peer ID to add private key for :param key_pair: """ - self.add_pubkey(key_pair.public_key) - self.add_privkey(key_pair.private_key) + self.add_pubkey(peer_id, key_pair.public_key) + self.add_privkey(peer_id, key_pair.private_key) class PeerStoreError(KeyError): diff --git a/libp2p/peer/peerstore_interface.py b/libp2p/peer/peerstore_interface.py index 618bbf5..a9790b0 100644 --- a/libp2p/peer/peerstore_interface.py +++ b/libp2p/peer/peerstore_interface.py @@ -1,5 +1,5 @@ from abc import abstractmethod -from typing import Any, List, Sequence, Set +from typing import Any, List, Sequence from multiaddr import Multiaddr @@ -138,9 +138,3 @@ class IPeerStore(IAddrBook, IPeerMetadata): :param key_pair: :raise PeerStoreError: if peer ID already has pubkey or privkey set """ - - @abstractmethod - def peers_with_keys(self) -> Set[ID]: - """ - :return: all of the peer IDs which has pubkey/privkey stored in peer store - """