Skip to content

Commit 1cfe4fb

Browse files
committed
remove source_stream.closed from _EncryptionStream.read()
1 parent 1942f59 commit 1cfe4fb

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/aws_encryption_sdk/streaming_client.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -243,10 +243,11 @@ def read(self, b=-1):
243243
output.write(self.output_buffer[:b])
244244
self.output_buffer = self.output_buffer[b:]
245245
else:
246-
while not self.source_stream.closed:
247-
self._read_bytes(LINE_LENGTH)
248-
output.write(self.output_buffer)
249-
self.output_buffer = b""
246+
while True:
247+
line = self.readline()
248+
if not line:
249+
break
250+
output.write(line)
250251

251252
self.bytes_read += output.tell()
252253
_LOGGER.debug("Returning %d bytes of %d bytes requested", output.tell(), b)

0 commit comments

Comments
 (0)