Implement peer store. Not yet linted or tested XD
This commit is contained in:
parent
f62db09179
commit
88b7eda256
|
@ -0,0 +1,88 @@
|
|||
from .peerstore_interface import IPeerStore
|
||||
import asyncio
|
||||
|
||||
class PeerStore(IPeerStore):
|
||||
|
||||
def __init__(self, context):
|
||||
self.context = context
|
||||
self.peerMap = {}
|
||||
|
||||
def peer_info(self, peerID):
|
||||
if peerID in self.peerMap:
|
||||
peer = self.peerMap[peerID]
|
||||
return {
|
||||
"peerID": peerID,
|
||||
"addrs": peer.get_addrs()
|
||||
}
|
||||
else:
|
||||
return None
|
||||
|
||||
def get_protocols(self, peerID):
|
||||
if peerID in self.peerMap:
|
||||
return self.peerMap[peerId].get_protocols(), None
|
||||
else:
|
||||
return None, peerID + " not found"
|
||||
|
||||
def add_protocols(self, peerID, protocols):
|
||||
peer = __create_or_get_peer(peerID)
|
||||
peer.add_protocols(protocols)
|
||||
|
||||
def peers(self):
|
||||
return self.peerMap.keys()
|
||||
|
||||
def get(self, peerID, key):
|
||||
if peerID in self.peerMap:
|
||||
val, error = self.peerMap[peerID].get_metadata(key)
|
||||
return val, error
|
||||
else:
|
||||
return None, peerID + " not found"
|
||||
|
||||
def put(self, peerID, key, val):
|
||||
# <<?>>
|
||||
# This can output an error, not sure what the possible errors are
|
||||
peer = __create_or_get_peer(peerID)
|
||||
self.peerMap[peerID].put_metadata(key, val)
|
||||
|
||||
def add_addr(self, peerID, addr, ttl):
|
||||
self.add_addrs(self, peerID, [addr])
|
||||
|
||||
def add_addrs(self, peerID, addrs, ttl):
|
||||
# Ignore ttl for now
|
||||
peer = __create_or_get_peer(peerID)
|
||||
peer.add_addrs(addrs)
|
||||
|
||||
def addrs(self, peerID):
|
||||
if peerID in self.peerMap:
|
||||
return self.peerMap[peerId].get_addrs(), None
|
||||
else:
|
||||
return None, peerID + " not found"
|
||||
pass
|
||||
|
||||
def clear_addrs(self, peerID):
|
||||
# Only clear addresses if the peer is in peer map
|
||||
if peerID in self.peerMap:
|
||||
self.peerMap[peerID].clear_addrs()
|
||||
|
||||
def peers_with_addrs(self):
|
||||
# Add all peers with addrs at least 1 to output
|
||||
output = []
|
||||
|
||||
for key in self.peerMap:
|
||||
if len(self.peerMap[key].get_addrs()) >= 1:
|
||||
output.append(key)
|
||||
return output
|
||||
|
||||
def __create_or_get_peer(self, peerID):
|
||||
"""
|
||||
Returns the peer data for peerID or creates a new
|
||||
peer data (and stores it in peerMap) if peer
|
||||
data for peerID does not yet exist
|
||||
:param peerID: peer ID
|
||||
:return: peer data
|
||||
"""
|
||||
if peerId in self.peerMap:
|
||||
return self.peerMap[peerId]
|
||||
else:
|
||||
data = PeerData(self.context)
|
||||
self.peerMap[peerId] = data
|
||||
return self.peerMap[peerId]
|
Loading…
Reference in New Issue
Block a user