refactor for sprint

This commit is contained in:
zixuanzh 2018-11-11 09:56:44 -05:00
parent fd0958aa4f
commit 2bde260f5f
20 changed files with 102 additions and 112 deletions

View File

@ -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)

View File

@ -1,8 +0,0 @@
from abc import ABC, abstractmethod
class IRawConnection(ABC):
@abstractmethod
async def open_connection(self):
pass

View File

@ -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

View File

@ -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)

View File

View 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)

View 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

View File

View 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
View 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