Update ref counting for 3.12

ijl · Aug 3, 2023 · fa2df4d · fa2df4d
1 parent 5c8f93d
commit fa2df4d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/ffi/fragment.rs b/src/ffi/fragment.rs
@@ -71,6 +71,9 @@ pub unsafe extern "C" fn orjson_fragmenttype_new() -> *mut PyTypeObject {
     let ob = Box::new(PyTypeObject {
         ob_base: PyVarObject {
             ob_base: PyObject {
+                #[cfg(Py_3_12)]
+                ob_refcnt: pyo3_ffi::PyObjectObRefcnt { ob_refcnt: 0 },
+                #[cfg(not(Py_3_12))]
                 ob_refcnt: 0,
                 ob_type: std::ptr::addr_of_mut!(PyType_Type),
             },

diff --git a/src/util.rs b/src/util.rs
@@ -56,6 +56,16 @@ macro_rules! str_from_slice {
     };
 }
 
+#[cfg(Py_3_12)]
+macro_rules! py_decref_without_destroy {
+    ($op:expr) => {
+        unsafe {
+            (*$op).ob_refcnt.ob_refcnt -= 1;
+        }
+    };
+}
+
+#[cfg(not(Py_3_12))]
 macro_rules! py_decref_without_destroy {
     ($op:expr) => {
         unsafe {