garbage control for context object

Author: ornariece

src/serializers/extra.rs

@@ -183,11 +183,11 @@ pub(crate) struct ExtraOwned {
     config: SerializationConfig,
     rec_guard: SerRecursionState,
     check: SerCheck,
-    model: Option<PyObject>,
+    pub model: Option<PyObject>,
     field_name: Option<String>,
     serialize_unknown: bool,
-    fallback: Option<PyObject>,
-    context: Option<PyObject>,
+    pub fallback: Option<PyObject>,
+    pub context: Option<PyObject>,
 }
 
 impl ExtraOwned {

src/serializers/type_serializers/function.rs

@@ -1,9 +1,11 @@
 use std::borrow::Cow;
 
 use pyo3::exceptions::{PyAttributeError, PyRecursionError, PyRuntimeError};
+use pyo3::gc::PyVisit;
 use pyo3::intern;
 use pyo3::prelude::*;
 use pyo3::types::PyDict;
+use pyo3::PyTraverseError;
 
 use pyo3::types::PyString;
 
@@ -440,6 +442,33 @@ impl SerializationCallable {
             exclude: exclude.map(|v| v.into_py(py)),
         }
     }
+
+    fn __traverse__(&self, visit: PyVisit<'_>) -> Result<(), PyTraverseError> {
+        if let Some(include) = &self.include {
+            visit.call(include)?;
+        }
+        if let Some(exclude) = &self.exclude {
+            visit.call(exclude)?;
+        }
+        if let Some(model) = &self.extra_owned.model {
+            visit.call(model)?;
+        }
+        if let Some(fallback) = &self.extra_owned.fallback {
+            visit.call(fallback)?;
+        }
+        if let Some(context) = &self.extra_owned.context {
+            visit.call(context)?;
+        }
+        Ok(())
+    }
+
+    fn __clear__(&mut self) {
+        self.include = None;
+        self.exclude = None;
+        self.extra_owned.model = None;
+        self.extra_owned.fallback = None;
+        self.extra_owned.context = None;
+    }
 }
 
 #[pymethods]
@@ -545,6 +574,25 @@ impl SerializationInfo {
             })
         }
     }
+
+    fn __traverse__(&self, visit: PyVisit<'_>) -> Result<(), PyTraverseError> {
+        if let Some(include) = &self.include {
+            visit.call(include)?;
+        }
+        if let Some(exclude) = &self.exclude {
+            visit.call(exclude)?;
+        }
+        if let Some(context) = &self.context {
+            visit.call(context)?;
+        }
+        Ok(())
+    }
+
+    fn __clear__(&mut self) {
+        self.include = None;
+        self.exclude = None;
+        self.context = None;
+    }
 }
 
 #[pymethods]

src/serializers/type_serializers/generator.rs

@@ -1,8 +1,10 @@
 use std::borrow::Cow;
 
+use pyo3::gc::PyVisit;
 use pyo3::intern;
 use pyo3::prelude::*;
 use pyo3::types::{PyDict, PyIterator};
+use pyo3::PyTraverseError;
 
 use serde::ser::SerializeSeq;
 
@@ -173,6 +175,33 @@ impl SerializationIterator {
             exclude: exclude.map(|v| v.into_py(py)),
         }
     }
+
+    fn __traverse__(&self, visit: PyVisit<'_>) -> Result<(), PyTraverseError> {
+        if let Some(include) = &self.include {
+            visit.call(include)?;
+        }
+        if let Some(exclude) = &self.exclude {
+            visit.call(exclude)?;
+        }
+        if let Some(model) = &self.extra_owned.model {
+            visit.call(model)?;
+        }
+        if let Some(fallback) = &self.extra_owned.fallback {
+            visit.call(fallback)?;
+        }
+        if let Some(context) = &self.extra_owned.context {
+            visit.call(context)?;
+        }
+        Ok(())
+    }
+
+    fn __clear__(&mut self) {
+        self.include = None;
+        self.exclude = None;
+        self.extra_owned.model = None;
+        self.extra_owned.fallback = None;
+        self.extra_owned.context = None;
+    }
 }
 
 #[pymethods]

src/validators/function.rs

@@ -480,6 +480,18 @@ impl ValidationInfo {
             mode: extra.input_type,
         }
     }
+
+    fn __traverse__(&self, visit: PyVisit<'_>) -> Result<(), PyTraverseError> {
+        visit.call(&self.config)?;
+        if let Some(context) = &self.context {
+            visit.call(context)?;
+        }
+        Ok(())
+    }
+
+    fn __clear__(&mut self) {
+        self.context = None;
+    }
 }
 
 #[pymethods]