diff options
Diffstat (limited to 'src/waitress/buffers.py')
-rw-r--r-- | src/waitress/buffers.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/src/waitress/buffers.py b/src/waitress/buffers.py index 0086fe8..386eb40 100644 --- a/src/waitress/buffers.py +++ b/src/waitress/buffers.py @@ -234,11 +234,23 @@ class OverflowableBuffer: return buf def _set_small_buffer(self): - self.buf = BytesIOBasedBuffer(self.buf) + oldbuf = self.buf + self.buf = BytesIOBasedBuffer(oldbuf) + + # Attempt to close the old buffer + if hasattr(oldbuf, "close"): + oldbuf.close() + self.overflowed = False def _set_large_buffer(self): - self.buf = TempfileBasedBuffer(self.buf) + oldbuf = self.buf + self.buf = TempfileBasedBuffer(oldbuf) + + # Attempt to close the old buffer + if hasattr(oldbuf, "close"): + oldbuf.close() + self.overflowed = True def append(self, s): |