7fa674dee2
* Implement protocol muxing * Integrate protocol muxing into new stream and listen's conn handler * Fix bugs in protocol muxing * Remove blank line * Add comments and fix linting issues * Fix order of parameters to select_one_of to match interface * Use array of protocol ids in new stream instead of protocol id * Add basic protocol muxer tests * Add todo * Modify new stream to take in protocol ids * Add check to all tests to ensure protocol id is saved to net stream properly * Lint tests * Fix lint issues * Add todo * Modify port numbers in tests * Fix linting issues * Add more documentation to functions * Add docs describing classes and fix indent error
26 lines
757 B
Python
26 lines
757 B
Python
from .multiselect_communicator_interface import IMultiselectCommunicator
|
|
|
|
class MultiselectCommunicator(IMultiselectCommunicator):
|
|
"""
|
|
Communicator helper class that ensures both the client
|
|
and multistream module will follow the same multistream protocol,
|
|
which is necessary for them to work
|
|
"""
|
|
|
|
def __init__(self, stream):
|
|
self.stream = stream
|
|
|
|
async def write(self, msg_str):
|
|
"""
|
|
Write message to stream
|
|
:param msg_str: message to write
|
|
"""
|
|
await self.stream.write(msg_str.encode())
|
|
|
|
async def read_stream_until_eof(self):
|
|
"""
|
|
Reads message from stream until EOF
|
|
"""
|
|
read_str = (await self.stream.read()).decode()
|
|
return read_str
|