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