py-libp2p/libp2p/protocol_muxer/multiselect_communicator.py

from libp2p.io.abc import ReadWriteCloser
from libp2p.utils import encode_delim, read_delim

from .multiselect_communicator_interface import IMultiselectCommunicator


class MultiselectCommunicator(IMultiselectCommunicator):
    read_writer: ReadWriteCloser

    def __init__(self, read_writer: ReadWriteCloser) -> None:
        self.read_writer = read_writer

    async def write(self, msg_str: str) -> None:
        msg_bytes = encode_delim(msg_str.encode())
        await self.read_writer.write(msg_bytes)

    async def read(self) -> str:
        data = await read_delim(self.read_writer)
        return data.decode()
Use `ReadWriteCloser` for conns and streams 2019-09-05 23:24:17 +08:00			`from libp2p.io.abc import ReadWriteCloser`
Move varint and delim read/write to toplevel To `libp2p.utils`. 2019-08-17 00:19:37 +08:00			`from libp2p.utils import encode_delim, read_delim`
add typing to protocol_muxer 2019-08-07 18:02:30 +08:00
Protocol muxing (#82) * 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 2018-11-29 02:51:50 +08:00			`from .multiselect_communicator_interface import IMultiselectCommunicator`

Migrate to new project structure. 2019-01-10 02:38:56 +08:00
Use `ReadWriteCloser` for conns and streams 2019-09-05 23:24:17 +08:00			`class MultiselectCommunicator(IMultiselectCommunicator):`
			`read_writer: ReadWriteCloser`
Add `delim_encode` and `delim_read` - Add `StreamCommunicator` and `RawConnectionCommunicator`, read/write messages with delim codec, with `IMuxedStream` and `IRawConnection` respectively. - Use it in `Multiselect` and `MultiselectClient`. 2019-08-15 23:31:26 +08:00
Use `ReadWriteCloser` for conns and streams 2019-09-05 23:24:17 +08:00			`def __init__(self, read_writer: ReadWriteCloser) -> None:`
			`self.read_writer = read_writer`
Add `delim_encode` and `delim_read` - Add `StreamCommunicator` and `RawConnectionCommunicator`, read/write messages with delim codec, with `IMuxedStream` and `IRawConnection` respectively. - Use it in `Multiselect` and `MultiselectClient`. 2019-08-15 23:31:26 +08:00
			`async def write(self, msg_str: str) -> None:`
Add lock to `RawConnection` To avoid `self.writer.drain()` is called in parallel. Reference: https://bugs.python.org/issue29930 2019-08-20 17:09:38 +08:00			`msg_bytes = encode_delim(msg_str.encode())`
Use `ReadWriteCloser` for conns and streams 2019-09-05 23:24:17 +08:00			`await self.read_writer.write(msg_bytes)`
Add `delim_encode` and `delim_read` - Add `StreamCommunicator` and `RawConnectionCommunicator`, read/write messages with delim codec, with `IMuxedStream` and `IRawConnection` respectively. - Use it in `Multiselect` and `MultiselectClient`. 2019-08-15 23:31:26 +08:00
			`async def read(self) -> str:`
Use `ReadWriteCloser` for conns and streams 2019-09-05 23:24:17 +08:00			`data = await read_delim(self.read_writer)`
Add lock to `RawConnection` To avoid `self.writer.drain()` is called in parallel. Reference: https://bugs.python.org/issue29930 2019-08-20 17:09:38 +08:00			`return data.decode()`