kopia lustrzana https://github.com/Yakifo/amqtt
BufferWriter uses provided buffer
rodzic
2cb5a7d9dc
commit
a5c34ed315
|
@ -173,8 +173,8 @@ class BufferReader(ReaderAdapter):
|
||||||
Byte Buffer reader adapter
|
Byte Buffer reader adapter
|
||||||
This adapter simply adapt reading a byte buffer.
|
This adapter simply adapt reading a byte buffer.
|
||||||
"""
|
"""
|
||||||
def __init__(self, data: bytes):
|
def __init__(self, buffer: bytes):
|
||||||
self._stream = io.BytesIO(data)
|
self._stream = io.BytesIO(buffer)
|
||||||
|
|
||||||
@asyncio.coroutine
|
@asyncio.coroutine
|
||||||
def read(self, n=-1) -> bytes:
|
def read(self, n=-1) -> bytes:
|
||||||
|
@ -186,8 +186,8 @@ class BufferWriter(WriterAdapter):
|
||||||
ByteBuffer writer adapter
|
ByteBuffer writer adapter
|
||||||
This adapter simply adapt writing to a byte buffer
|
This adapter simply adapt writing to a byte buffer
|
||||||
"""
|
"""
|
||||||
def __init__(self):
|
def __init__(self, buffer=b''):
|
||||||
self._stream = io.BytesIO(b'')
|
self._stream = io.BytesIO(buffer)
|
||||||
|
|
||||||
def write(self, data):
|
def write(self, data):
|
||||||
"""
|
"""
|
||||||
|
@ -197,7 +197,7 @@ class BufferWriter(WriterAdapter):
|
||||||
|
|
||||||
@asyncio.coroutine
|
@asyncio.coroutine
|
||||||
def drain(self):
|
def drain(self):
|
||||||
self._stream = io.BytesIO(b'')
|
pass
|
||||||
|
|
||||||
def get_buffer(self):
|
def get_buffer(self):
|
||||||
return self._stream.getvalue()
|
return self._stream.getvalue()
|
||||||
|
|
Ładowanie…
Reference in New Issue