comparison rust/hg-cpython/src/dirstate/dirstate_map.rs @ 43284:ce6dd1cee4c8

rust-cpython: put leaked reference in PyLeakedRef

The next patch will make PyLeakedRef manage the lifetime of the underlying
object. leak_handle.data.take() will be removed soon.

43283:96eb9ef777a8

            )),
        }
    }

    def keys(&self) -> PyResult<DirstateMapKeysIterator> {
        let (leak_handle, leaked_ref) =
            unsafe { self.inner_shared(py).leak_immutable()? };
        DirstateMapKeysIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.iter(),
        )
    }

    def items(&self) -> PyResult<DirstateMapItemsIterator> {
        let (leak_handle, leaked_ref) =
            unsafe { self.inner_shared(py).leak_immutable()? };
        DirstateMapItemsIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.iter(),
        )
    }

    def __iter__(&self) -> PyResult<DirstateMapKeysIterator> {
        let (leak_handle, leaked_ref) =
            unsafe { self.inner_shared(py).leak_immutable()? };
        DirstateMapKeysIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.iter(),
        )

            None => Ok(default),
        }
    }

    def copymapiter(&self) -> PyResult<CopyMapKeysIterator> {
        let (leak_handle, leaked_ref) =
            unsafe { self.inner_shared(py).leak_immutable()? };
        CopyMapKeysIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.copy_map.iter(),
        )
    }

    def copymapitemsiter(&self) -> PyResult<CopyMapItemsIterator> {
        let (leak_handle, leaked_ref) =
            unsafe { self.inner_shared(py).leak_immutable()? };
        CopyMapItemsIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.copy_map.iter(),
        )

py_shared_ref!(DirstateMap, RustDirstateMap, inner, inner_shared);

py_shared_iterator!(
    DirstateMapKeysIterator,
    PyLeakedRef,
    StateMapIter<'static>,
    DirstateMap::translate_key,
    Option<PyBytes>
);

py_shared_iterator!(
    DirstateMapItemsIterator,
    PyLeakedRef,
    StateMapIter<'static>,
    DirstateMap::translate_key_value,
    Option<(PyBytes, PyObject)>
);

43284:ce6dd1cee4c8

            )),
        }
    }

    def keys(&self) -> PyResult<DirstateMapKeysIterator> {
        let mut leak_handle =
            unsafe { self.inner_shared(py).leak_immutable()? };
        let leaked_ref = leak_handle.data.take().unwrap();
        DirstateMapKeysIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.iter(),
        )
    }

    def items(&self) -> PyResult<DirstateMapItemsIterator> {
        let mut leak_handle =
            unsafe { self.inner_shared(py).leak_immutable()? };
        let leaked_ref = leak_handle.data.take().unwrap();
        DirstateMapItemsIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.iter(),
        )
    }

    def __iter__(&self) -> PyResult<DirstateMapKeysIterator> {
        let mut leak_handle =
            unsafe { self.inner_shared(py).leak_immutable()? };
        let leaked_ref = leak_handle.data.take().unwrap();
        DirstateMapKeysIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.iter(),
        )

            None => Ok(default),
        }
    }

    def copymapiter(&self) -> PyResult<CopyMapKeysIterator> {
        let mut leak_handle =
            unsafe { self.inner_shared(py).leak_immutable()? };
        let leaked_ref = leak_handle.data.take().unwrap();
        CopyMapKeysIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.copy_map.iter(),
        )
    }

    def copymapitemsiter(&self) -> PyResult<CopyMapItemsIterator> {
        let mut leak_handle =
            unsafe { self.inner_shared(py).leak_immutable()? };
        let leaked_ref = leak_handle.data.take().unwrap();
        CopyMapItemsIterator::from_inner(
            py,
            leak_handle,
            leaked_ref.copy_map.iter(),
        )

py_shared_ref!(DirstateMap, RustDirstateMap, inner, inner_shared);

py_shared_iterator!(
    DirstateMapKeysIterator,
    PyLeakedRef<&'static RustDirstateMap>,
    StateMapIter<'static>,
    DirstateMap::translate_key,
    Option<PyBytes>
);

py_shared_iterator!(
    DirstateMapItemsIterator,
    PyLeakedRef<&'static RustDirstateMap>,
    StateMapIter<'static>,
    DirstateMap::translate_key_value,
    Option<(PyBytes, PyObject)>
);