refactor for sprint
This commit is contained in:
parent
fd0958aa4f
commit
2bde260f5f
|
@ -1,14 +0,0 @@
|
||||||
import asyncio
|
|
||||||
from .raw_connection_interface import IRawConnection
|
|
||||||
|
|
||||||
class RawConnection(IRawConnection):
|
|
||||||
|
|
||||||
def __init__(self, ip, port):
|
|
||||||
self.conn_ip = ip
|
|
||||||
self.conn_port = port
|
|
||||||
self.reader = None
|
|
||||||
self.writer = None
|
|
||||||
|
|
||||||
async def open_connection(self):
|
|
||||||
self.reader, self.writer = \
|
|
||||||
await asyncio.open_connection(self.conn_ip, self.conn_port)
|
|
|
@ -1,8 +0,0 @@
|
||||||
from abc import ABC, abstractmethod
|
|
||||||
|
|
||||||
class IRawConnection(ABC):
|
|
||||||
|
|
||||||
@abstractmethod
|
|
||||||
async def open_connection(self):
|
|
||||||
pass
|
|
||||||
|
|
|
@ -0,0 +1,17 @@
|
||||||
|
from .config import Config
|
||||||
|
from ..peer.peerstore import PeerStore
|
||||||
|
|
||||||
|
class Libp2p(object):
|
||||||
|
|
||||||
|
def __init__(self, idOpt, \
|
||||||
|
transportOpt = ["/ip4/0.0.0.0/tcp/0"], \
|
||||||
|
muxerOpt = ["mplex/6.7.0"], \
|
||||||
|
secOpt = ["secio"], \
|
||||||
|
peerstoreOpt = new PeerStore()):
|
||||||
|
|
||||||
|
if idOpt:
|
||||||
|
self.idOpt = idOpt
|
||||||
|
else:
|
||||||
|
# TODO generate RSA public key pair
|
||||||
|
|
||||||
|
# TODO initialize
|
|
@ -1,83 +0,0 @@
|
||||||
import asyncio
|
|
||||||
from .transport_interface import ITransport
|
|
||||||
from .listener_interface import IListener
|
|
||||||
|
|
||||||
class TCP(ITransport):
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self.listener = self.Listener()
|
|
||||||
|
|
||||||
class Listener(IListener):
|
|
||||||
|
|
||||||
def __init__(self, handler_function=None):
|
|
||||||
self.multiaddrs = []
|
|
||||||
self.server = None
|
|
||||||
self.handler = staticmethod(handler_function)
|
|
||||||
|
|
||||||
def listen(self, multiaddr):
|
|
||||||
"""
|
|
||||||
put listener in listening mode and wait for incoming connections
|
|
||||||
:param multiaddr: multiaddr of peer
|
|
||||||
:return: return True if successful
|
|
||||||
"""
|
|
||||||
# TODO check for exceptions
|
|
||||||
_multiaddr = multiaddr
|
|
||||||
if "ipfs" in multiaddr.get_protocols():
|
|
||||||
# ipfs_id = multiaddr.get_ipfs_id()
|
|
||||||
_multiaddr = multiaddr.remove_protocol("ipfs")
|
|
||||||
|
|
||||||
self.multiaddrs.append(_multiaddr)
|
|
||||||
_multiaddr_dict = _multiaddr.to_dict()
|
|
||||||
_loop = asyncio.get_event_loop()
|
|
||||||
_coroutine = asyncio.start_server(self.handler, _multiaddr_dict.host,\
|
|
||||||
_multiaddr_dict.port, loop=_loop)
|
|
||||||
self.server = _loop.run_until_complete(_coroutine)
|
|
||||||
return True
|
|
||||||
|
|
||||||
def get_addrs(self):
|
|
||||||
"""
|
|
||||||
retrieve list of addresses the listener is listening on
|
|
||||||
:return: return list of addrs
|
|
||||||
"""
|
|
||||||
# TODO check if server is listening
|
|
||||||
return self.multiaddrs
|
|
||||||
|
|
||||||
def close(self, options=None):
|
|
||||||
"""
|
|
||||||
close the listener such that no more connections
|
|
||||||
can be open on this transport instance
|
|
||||||
:param options: optional object potential with timeout
|
|
||||||
a timeout value in ms that fires and destroy all connections
|
|
||||||
:return: return True if successful
|
|
||||||
"""
|
|
||||||
if self.server is None:
|
|
||||||
return False
|
|
||||||
self.server.close()
|
|
||||||
_loop = asyncio.get_event_loop()
|
|
||||||
_loop.run_until_complete(self.server.wait_closed())
|
|
||||||
_loop.close()
|
|
||||||
self.server = None
|
|
||||||
return True
|
|
||||||
|
|
||||||
def dial(self, multiaddr, options=None):
|
|
||||||
"""
|
|
||||||
dial a transport to peer listening on multiaddr
|
|
||||||
:param multiaddr: multiaddr of peer
|
|
||||||
:param options: optional object
|
|
||||||
:return: True if successful
|
|
||||||
"""
|
|
||||||
_multiaddr_dict = multiaddr.to_dict()
|
|
||||||
reader, writer = await asyncio.open_connection(_multiaddr_dict.host,\
|
|
||||||
_multiaddr_dict.port)
|
|
||||||
return False
|
|
||||||
# TODO dial behavior not fully understood
|
|
||||||
|
|
||||||
def create_listener(self, handler_function, options=None):
|
|
||||||
"""
|
|
||||||
create listener on transport
|
|
||||||
:param options: optional object with properties the listener must have
|
|
||||||
:param handler_function: a function called when a new conntion is received
|
|
||||||
that takes a connection as argument which implements interface-connection
|
|
||||||
:return: a listener object that implements listener_interface.py
|
|
||||||
"""
|
|
||||||
return self.Listener(handler_function)
|
|
0
transport/connection/__init__.py
Normal file
0
transport/connection/__init__.py
Normal file
16
transport/connection/raw_connection.py
Normal file
16
transport/connection/raw_connection.py
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
import asyncio
|
||||||
|
from .raw_connection_interface import IRawConnection
|
||||||
|
|
||||||
|
class RawConnection(IRawConnection):
|
||||||
|
|
||||||
|
def __init__(self, ip, port):
|
||||||
|
self.conn_ip = ip
|
||||||
|
self.conn_port = port
|
||||||
|
self.reader, self.writer = self.open_connection()
|
||||||
|
|
||||||
|
async def open_connection(self):
|
||||||
|
"""
|
||||||
|
opens a connection on self.ip and self.port
|
||||||
|
:return: a raw connection
|
||||||
|
"""
|
||||||
|
return await asyncio.open_connection(self.conn_ip, self.conn_port)
|
15
transport/connection/raw_connection_interface.py
Normal file
15
transport/connection/raw_connection_interface.py
Normal file
|
@ -0,0 +1,15 @@
|
||||||
|
from abc import ABC, abstractmethod
|
||||||
|
|
||||||
|
class IRawConnection(ABC):
|
||||||
|
"""
|
||||||
|
A Raw Connection provides a Reader and a Writer
|
||||||
|
open_connection should return such a connection
|
||||||
|
"""
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
async def open_connection(self):
|
||||||
|
"""
|
||||||
|
opens a connection on ip and port
|
||||||
|
:return: a raw connection
|
||||||
|
"""
|
||||||
|
pass
|
0
transport/stream/__init__.py
Normal file
0
transport/stream/__init__.py
Normal file
|
@ -5,24 +5,42 @@ from .listener_interface import IListener
|
||||||
class TCP(ITransport):
|
class TCP(ITransport):
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.multiaddr = None
|
self.listener = self.Listener()
|
||||||
|
|
||||||
class Listener(IListener):
|
class Listener(IListener):
|
||||||
|
|
||||||
|
def __init__(self, handler_function=None):
|
||||||
|
self.multiaddrs = []
|
||||||
|
self.server = None
|
||||||
|
self.handler = staticmethod(handler_function)
|
||||||
|
|
||||||
def listen(self, multiaddr):
|
def listen(self, multiaddr):
|
||||||
"""
|
"""
|
||||||
put listener in listening mode and wait for incoming connections
|
put listener in listening mode and wait for incoming connections
|
||||||
:param multiaddr: multiaddr of peer
|
:param multiaddr: multiaddr of peer
|
||||||
:return: return True if successful
|
:return: return True if successful
|
||||||
"""
|
"""
|
||||||
pass
|
# TODO check for exceptions
|
||||||
|
_multiaddr = multiaddr
|
||||||
|
if "ipfs" in multiaddr.get_protocols():
|
||||||
|
# ipfs_id = multiaddr.get_ipfs_id()
|
||||||
|
_multiaddr = multiaddr.remove_protocol("ipfs")
|
||||||
|
|
||||||
|
self.multiaddrs.append(_multiaddr)
|
||||||
|
_multiaddr_dict = _multiaddr.to_dict()
|
||||||
|
_loop = asyncio.get_event_loop()
|
||||||
|
_coroutine = asyncio.start_server(self.handler, _multiaddr_dict.host,\
|
||||||
|
_multiaddr_dict.port, loop=_loop)
|
||||||
|
self.server = _loop.run_until_complete(_coroutine)
|
||||||
|
return True
|
||||||
|
|
||||||
def get_addrs(self):
|
def get_addrs(self):
|
||||||
"""
|
"""
|
||||||
retrieve list of addresses the listener is listening on
|
retrieve list of addresses the listener is listening on
|
||||||
:return: return list of addrs
|
:return: return list of addrs
|
||||||
"""
|
"""
|
||||||
pass
|
# TODO check if server is listening
|
||||||
|
return self.multiaddrs
|
||||||
|
|
||||||
def close(self, options=None):
|
def close(self, options=None):
|
||||||
"""
|
"""
|
||||||
|
@ -32,16 +50,27 @@ class TCP(ITransport):
|
||||||
a timeout value in ms that fires and destroy all connections
|
a timeout value in ms that fires and destroy all connections
|
||||||
:return: return True if successful
|
:return: return True if successful
|
||||||
"""
|
"""
|
||||||
pass
|
if self.server is None:
|
||||||
|
return False
|
||||||
|
self.server.close()
|
||||||
|
_loop = asyncio.get_event_loop()
|
||||||
|
_loop.run_until_complete(self.server.wait_closed())
|
||||||
|
_loop.close()
|
||||||
|
self.server = None
|
||||||
|
return True
|
||||||
|
|
||||||
def dial(self, multiaddr, options=None):
|
def dial(self, multiaddr, options=None):
|
||||||
"""
|
"""
|
||||||
dial a transport to peer listening on multiaddr
|
dial a transport to peer listening on multiaddr
|
||||||
:param multiaddr: multiaddr of peer
|
:param multiaddr: multiaddr of peer
|
||||||
:param options: optional object
|
:param options: optional object
|
||||||
:return: list of multiaddrs
|
:return: True if successful
|
||||||
"""
|
"""
|
||||||
pass
|
_multiaddr_dict = multiaddr.to_dict()
|
||||||
|
reader, writer = await asyncio.open_connection(_multiaddr_dict.host,\
|
||||||
|
_multiaddr_dict.port)
|
||||||
|
return False
|
||||||
|
# TODO dial behavior not fully understood
|
||||||
|
|
||||||
def create_listener(self, handler_function, options=None):
|
def create_listener(self, handler_function, options=None):
|
||||||
"""
|
"""
|
||||||
|
@ -51,4 +80,4 @@ class TCP(ITransport):
|
||||||
that takes a connection as argument which implements interface-connection
|
that takes a connection as argument which implements interface-connection
|
||||||
:return: a listener object that implements listener_interface.py
|
:return: a listener object that implements listener_interface.py
|
||||||
"""
|
"""
|
||||||
pass
|
return self.Listener(handler_function)
|
||||||
|
|
18
transport/upgrader.py
Normal file
18
transport/upgrader.py
Normal file
|
@ -0,0 +1,18 @@
|
||||||
|
class TransportUpgrader(object):
|
||||||
|
|
||||||
|
def __init__(self, secOpt, muxerOpt):
|
||||||
|
self.sec = secOpt
|
||||||
|
self.muxer = muxerOpt
|
||||||
|
|
||||||
|
def upgrade_listener(self, transport, listeners):
|
||||||
|
"""
|
||||||
|
upgrade multiaddr listeners to libp2p-transport listeners
|
||||||
|
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
def upgrade_security(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def upgrade_muxer(self):
|
||||||
|
pass
|
Loading…
Reference in New Issue
Block a user