self.buffer_size = buffer_size
self._write_buf = bytearray()
self._write_lock = Lock()
- _register_writer(self)
def writable(self):
return self.raw.writable()
def detach(self):
# This doesn't make sense on StringIO.
self._unsupported("detach")
-
-
-# ____________________________________________________________
-
-import atexit, weakref
-
-_all_writers = weakref.WeakSet()
-
-def _register_writer(w):
- # keep weak-ref to buffered writer
- _all_writers.add(w)
-
-def _flush_all_writers():
- # Ensure all buffered writers are flushed before proceeding with
- # normal shutdown. Otherwise, if the underlying file objects get
- # finalized before the buffered writer wrapping it then any buffered
- # data will be lost.
- for w in _all_writers:
- try:
- w.flush()
- except:
- pass
-atexit.register(_flush_all_writers)
}
-typedef struct _buffered {
+typedef struct {
PyObject_HEAD
PyObject *raw;
PyObject *dict;
PyObject *weakreflist;
-
- /* a doubly-linked chained list of "buffered" objects that need to
- be flushed when the process exits */
- struct _buffered *next, *prev;
} buffered;
-/* the actual list of buffered objects */
-static buffered buffer_list_end = {
- .next = &buffer_list_end,
- .prev = &buffer_list_end
-};
-
/*
Implementation notes:
(self->buffer_size * (size / self->buffer_size)))
-static void
-remove_from_linked_list(buffered *self)
-{
- self->next->prev = self->prev;
- self->prev->next = self->next;
- self->prev = NULL;
- self->next = NULL;
-}
-
static void
buffered_dealloc(buffered *self)
{
return;
_PyObject_GC_UNTRACK(self);
self->ok = 0;
- if (self->next != NULL)
- remove_from_linked_list(self);
if (self->weakreflist != NULL)
PyObject_ClearWeakRefs((PyObject *)self);
Py_CLEAR(self->raw);
self->fast_closed_checks = (Py_TYPE(self) == &PyBufferedWriter_Type &&
Py_TYPE(raw) == &PyFileIO_Type);
- if (self->next == NULL) {
- self->prev = &buffer_list_end;
- self->next = buffer_list_end.next;
- buffer_list_end.next->prev = self;
- buffer_list_end.next = self;
- }
-
self->ok = 1;
return 0;
}
-/*
-* Ensure all buffered writers are flushed before proceeding with
-* normal shutdown. Otherwise, if the underlying file objects get
-* finalized before the buffered writer wrapping it then any buffered
-* data will be lost.
-*/
-void _PyIO_atexit_flush(void)
-{
- while (buffer_list_end.next != &buffer_list_end) {
- buffered *buf = buffer_list_end.next;
- remove_from_linked_list(buf);
- buffered_flush(buf, NULL);
- PyErr_Clear();
- }
-}
-
static Py_ssize_t
_bufferedwriter_raw_write(buffered *self, char *start, Py_ssize_t len)
{