masscollaborationlabs/p2pfs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Handle REQUEST_REGISTER.

Browse source
This commit is contained in:
Yuxin Wang 2018-09-25 14:28:11 -04:00
parent 993b1050de
commit 998daaaa59
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
p2pfs/core/server.py
View file

@ -1,4 +1,5 @@
from p2pfs.core.utils import MessageServer
from p2pfs.core.message import MessageType
import socket
import logging
logger = logging.getLogger(__name__)
@ -17,7 +18,16 @@ class CentralServer(MessageServer):
def _process_message(self, client, message):
assert isinstance(client, socket.socket)
print(message)
if message['type'] == MessageType.REQUEST_REGISTER:
assert client in self._peers
self._write_message(client, {
'type': MessageType.REPLY_REGISTER,
'id': self._id,
'peer_list': tuple(filter(lambda x: x is not None, self._peers.values()))
})
self._peers[client] = (self._id, message['address'])
self._id += 1
logger.debug(self._peers.values())
def _client_closed(self, client):
assert isinstance(client, socket.socket)