StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Catch PeerDataError in PeerStore.get

Browse Source
This commit is contained in:
NIC619 2019-10-17 15:52:57 +08:00
parent 5063f0e2a6
commit 211e2f6dd0
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libp2p/peer/peerstore.py
View File

@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional, Sequence
from multiaddr import Multiaddr from multiaddr import Multiaddr
from .id import ID from .id import ID
from .peerdata import PeerData from .peerdata import PeerData, PeerDataError
from .peerinfo import PeerInfo from .peerinfo import PeerInfo
from .peerstore_interface import IPeerStore from .peerstore_interface import IPeerStore
@ -54,7 +54,10 @@ class PeerStore(IPeerStore):
def get(self, peer_id: ID, key: str) -> Any: def get(self, peer_id: ID, key: str) -> Any:
if peer_id in self.peer_map: if peer_id in self.peer_map:
val = self.peer_map[peer_id].get_metadata(key) try:
val = self.peer_map[peer_id].get_metadata(key)
except PeerDataError as error:
raise PeerStoreError(error)
return val return val
raise PeerStoreError("peer ID not found") raise PeerStoreError("peer ID not found")