From ffde646050fa2fa126b775402c98c77e4d629352 Mon Sep 17 00:00:00 2001
From: Pieter Eendebak <pieter.eendebak@gmail.com>
Date: Fri, 8 Nov 2024 12:59:25 +0100
Subject: [PATCH] Use freelist for rangeiter object

---
 Include/internal/pycore_freelist_state.h |  2 ++
 Objects/rangeobject.c                    | 17 +++++++++++++----
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/Include/internal/pycore_freelist_state.h b/Include/internal/pycore_freelist_state.h
index 70c0403b32ba83..169fd62f9ee45b 100644
--- a/Include/internal/pycore_freelist_state.h
+++ b/Include/internal/pycore_freelist_state.h
@@ -16,6 +16,7 @@ extern "C" {
 #  define Py_floats_MAXFREELIST 100
 #  define Py_slices_MAXFREELIST 1
 #  define Py_ranges_MAXFREELIST 10
+#  define Py_rangeiters_MAXFREELIST 10
 #  define Py_contexts_MAXFREELIST 255
 #  define Py_async_gens_MAXFREELIST 80
 #  define Py_async_gen_asends_MAXFREELIST 80
@@ -42,6 +43,7 @@ struct _Py_freelists {
     struct _Py_freelist dictkeys;
     struct _Py_freelist slices;
     struct _Py_freelist ranges;
+    struct _Py_freelist rangeiters;
     struct _Py_freelist contexts;
     struct _Py_freelist async_gens;
     struct _Py_freelist async_gen_asends;
diff --git a/Objects/rangeobject.c b/Objects/rangeobject.c
index 959b4f91733aed..8df6d95df22047 100644
--- a/Objects/rangeobject.c
+++ b/Objects/rangeobject.c
@@ -885,6 +885,12 @@ rangeiter_setstate(_PyRangeIterObject *r, PyObject *state)
     Py_RETURN_NONE;
 }
 
+static void
+rangeiter_dealloc(_PyRangeIterObject *r)
+{
+    _Py_FREELIST_FREE(rangeiters, r, PyObject_Free);
+}
+
 PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
 PyDoc_STRVAR(setstate_doc, "Set state information for unpickling.");
 
@@ -904,7 +910,7 @@ PyTypeObject PyRangeIter_Type = {
         sizeof(_PyRangeIterObject),             /* tp_basicsize */
         0,                                      /* tp_itemsize */
         /* methods */
-        (destructor)PyObject_Free,              /* tp_dealloc */
+        (destructor)rangeiter_dealloc,              /* tp_dealloc */
         0,                                      /* tp_vectorcall_offset */
         0,                                      /* tp_getattr */
         0,                                      /* tp_setattr */
@@ -965,9 +971,12 @@ 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 = PyObject_New(_PyRangeIterObject, &PyRangeIter_Type);
-    if (it == NULL)
-        return NULL;
+    _PyRangeIterObject *it = _Py_FREELIST_POP(_PyRangeIterObject, rangeiters);
+    if (it == NULL) {
+        it = PyObject_New(_PyRangeIterObject, &PyRangeIter_Type);
+        if (it == NULL)
+            return NULL;
+    }
     it->start = start;
     it->step = step;
     it->len = len;