freelist for range iterators

Author: eendebakpt

Include/internal/pycore_freelist_state.h

@@ -17,6 +17,7 @@ extern "C" {
 #  define Py_ints_MAXFREELIST 100
 #  define Py_slices_MAXFREELIST 1
 #  define Py_ranges_MAXFREELIST 10
+#  define Py_range_iters_MAXFREELIST 10
 #  define Py_shared_iters_MAXFREELIST 24
 #  define Py_class_method_MAXFREELIST 10
 #  define Py_contexts_MAXFREELIST 255
@@ -46,6 +47,7 @@ struct _Py_freelists {
     struct _Py_freelist dictkeys;
     struct _Py_freelist slices;
     struct _Py_freelist ranges;
+    struct _Py_freelist range_iters;
     struct _Py_freelist shared_iters;
     struct _Py_freelist class_method;
     struct _Py_freelist contexts;

Objects/iterobject.c

@@ -23,7 +23,7 @@ PySeqIter_New(PyObject *seq)
         return NULL;
     }
 
-    it = _Py_FREELIST_POP(seqiterobject, shared_iters);
+    it = 0; //_Py_FREELIST_POP(seqiterobject, shared_iters);
     if (it == NULL) {
         it = PyObject_GC_New(seqiterobject, &PySeqIter_Type);
         if (it == NULL)

Objects/listobject.c

@@ -3914,7 +3914,7 @@ list_iter(PyObject *seq)
         PyErr_BadInternalCall();
         return NULL;
     }
-    _PyListIterObject *it = _Py_FREELIST_POP(_PyListIterObject, shared_iters);
+    _PyListIterObject *it = 0; //_Py_FREELIST_POP(_PyListIterObject, shared_iters);
     if (it == NULL) {
         it = PyObject_GC_New(_PyListIterObject, &PyListIter_Type);
         if (it == NULL) {

Objects/object.c

@@ -939,6 +939,7 @@ _PyObject_ClearFreeLists(struct _Py_freelists *freelists, int is_finalization)
     clear_freelist(&freelists->ints, is_finalization, free_object);
     clear_freelist(&freelists->shared_iters, is_finalization, free_object);
     clear_freelist(&freelists->ranges, is_finalization, free_object);
+    clear_freelist(&freelists->range_iters, is_finalization, free_object);
     clear_freelist(&freelists->class_method, is_finalization, free_object);
 }
 

Objects/rangeobject.c

@@ -886,7 +886,7 @@ rangeiter_setstate(_PyRangeIterObject *r, PyObject *state)
 static void
 rangeiter_dealloc(_PyRangeIterObject *r)
 {
-    _Py_FREELIST_FREE(shared_iters, r, PyObject_Free);
+    _Py_FREELIST_FREE(range_iters, r, PyObject_Free);
 }
 
 PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
@@ -969,15 +969,14 @@ get_len_of_range(long lo, long hi, long step)
 static PyObject *
 fast_range_iter(long start, long stop, long step, long len)
 {
-    _PyRangeIterObject *it = _Py_FREELIST_POP(_PyRangeIterObject, shared_iters);
+    _PyRangeIterObject *it = _Py_FREELIST_POP(_PyRangeIterObject, range_iters);
     if (it == NULL) {
         it = PyObject_New(_PyRangeIterObject, &PyRangeIter_Type);
         if (it == NULL)
             return NULL;
     }
     else {
         Py_SET_TYPE(it, &PyRangeIter_Type);
-        //_PyObject_GC_TRACK(it); // need to track with GC again
     }
     it->start = start;
     it->step = step;

Objects/tupleobject.c

@@ -1123,7 +1123,7 @@ tuple_iter(PyObject *seq)
         PyErr_BadInternalCall();
         return NULL;
     }
-    it = _Py_FREELIST_POP(_PyTupleIterObject, shared_iters);
+    it = 0; // _Py_FREELIST_POP(_PyTupleIterObject, shared_iters);
 
     if (it == NULL) {
         it = PyObject_GC_New(_PyTupleIterObject, &PyTupleIter_Type);