Merge pull request #28 from zixuanzh/raw-connection

Raw connection

network/multiaddr.py

@@ -88,7 +88,8 @@ class MultiAddr:
         Gives back a dictionary with access to transport information from this multiaddr.
         Example: MultiAddr('/ip4/127.0.0.1/tcp/4001').to_options()
         = { family: 'ipv4', host: '127.0.0.1', transport: 'tcp', port: '4001' }
-        :return: {{family: String, host: String, transport: String, port: String}} with None if field does not exist
+        :return: {{family: String, host: String, transport: String, port: String}}
+        with None if field does not exist
         """
         options = dict()
 

network/network_interface.py

@@ -2,10 +2,6 @@ from abc import ABC, abstractmethod
 
 class INetwork(ABC):
 
-    def __init__(self, my_peer_id, peer_store):
-        self.my_peer_id = my_peer_id
-        self.peer_store = peer_store
-
     @abstractmethod
     def set_stream_handler(self, stream_handler):
         """
@@ -15,9 +11,10 @@ class INetwork(ABC):
         pass
 
     @abstractmethod
-    def new_stream(self, peer_id):
+    def new_stream(self, peer_id, protocol_id):
         """
         :param peer_id: peer_id of destination
+        :param protocol_id: protocol id
         :return: stream instance
         """
         pass
@@ -28,4 +25,4 @@ class INetwork(ABC):
         :param *args: one or many multiaddrs to start listening on
         :return: True if at least one success
         """
-        pass
+        pass

network/swarm.py

@@ -1,10 +1,13 @@
 from .network_interface import INetwork
+from ..connection.muxed_connection import MuxedConnection
+from ..connection.raw_connection import RawConnection
 
 class Swarm(INetwork):
 
     def __init__(self, my_peer_id, peer_store):
         self.my_peer_id = my_peer_id
         self.peer_store = peer_store
+        self.connections = {}
 
     def set_stream_handler(self, stream_handler):
         """
@@ -13,12 +16,32 @@ class Swarm(INetwork):
         """
         pass
 
-    def new_stream(self, peer_id):
+    def new_stream(self, peer_id, protocol_id):
         """
+        Determine if a connection to peer_id already exists
+        If a connection to peer_id exists, then
+        c = existing connection,
+        otherwise c = new muxed connection to peer_id
+        s = c.open_stream(protocol_id)
+        return s
+
         :param peer_id: peer_id of destination
+        :param protocol_id: protocol id
         :return: stream instance
         """
-        pass
+        muxed_connection = None
+        if peer_id in self.connections:
+            muxed_connection = self.connections[peer_id]
+        else:
+            addrs = self.peer_store.addrs(peer_id)
+            stream_ip = addrs.get_protocol_value("ip")
+            stream_port = addrs.get_protocol_value("port")
+            if len(addrs) > 0:
+                conn = RawConnection(stream_ip, stream_port)
+                muxed_connection = MuxedConnection(conn, True)
+            else:
+                raise Exception("No IP and port in addr")
+        return muxed_connection.open_stream(protocol_id, "")
 
     def listen(self, *args):
         """

network/walkthrough.txt

@@ -0,0 +1,4 @@
+host.go --> config.go 
+    config.go: newNode --> swarm.go: newSwarm
+        newSwarm | initializes data stores
+