async def handle_connection(self, reader: StreamReader,
writer: StreamWriter):
while True:
try:
# Wait forever on new data to arrive
data = await reader.readuntil(b"\n") # <3>
decoded_data = data.decode("utf8").strip() # <4>
try:
message = BaseSchema().loads(decoded_data) # <5>
except MarshmallowError:
logger.info("Received unreadable message", peer=writer)
break
# Extract the address from the message, add it to the writer object
writer.address = message["meta"]["address"]
# Let's add the peer to our connection pool
self.connection_pool.add_peer(writer)
# ...and handle the message
await self.p2p_protocol.handle_message(message, writer) # <6>
await writer.drain()
if writer.is_closing():
break
except (asyncio.exceptions.IncompleteReadError, ConnectionError):
# An error happened, break out of the wait loop
break
# The connection has closed. Let's clean up...
writer.close()
await writer.wait_closed()
self.connection_pool.remove_peer(writer) # <7>
