mercurial-scm/hg: rust/hg-pyo3/src/revlog/mod.rs annotate

annotate rust/hg-pyo3/src/revlog/mod.rs @ 52799:798355e46d03

rust-pyo3-revlog: _index_findsnapshots We still need to enclose the cache because of the foreign type limitation and of the error treatment, but overall this is simpler than the `rust-cpython` counterpart This one was still creating a `c_cache` object dating back from the times when we were comparing with the C index (but is was not filling its `PySet` values).

author	Georges Racinet <georges.racinet@cloudcrane.io>
date	Mon, 23 Dec 2024 20:44:26 +0100
parents	88d62995a65b
children	ebcbd2b7a3b6

rev	line source
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1 // revlog.rs
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	2 //
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	3 // Copyright 2019-2020 Georges Racinet <georges.racinet@octobus.net>
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	4 // 2020-2024 Raphaël Gomès <raphael.gomes@octobus.net>
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	5 // 2024 Georges Racinet <georges.racinet@cloudcrane.io>
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	6 //
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	7 // This software may be used and distributed according to the terms of the
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	8 // GNU General Public License version 2 or any later version.
52787 e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	9 #![allow(non_snake_case)]
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	10 use pyo3::buffer::PyBuffer;
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	11 use pyo3::conversion::IntoPyObject;
52797 f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	12 use pyo3::exceptions::{PyIndexError, PyTypeError, PyValueError};
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	13 use pyo3::prelude::*;
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	14 use pyo3::types::{PyBytes, PyBytesMethods, PyDict, PyList, PySet, PyTuple};
52792 acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	15 use pyo3_sharedref::{PyShareable, SharedByPyObject};
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	16
52795 adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	17 use std::collections::HashSet;
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	18 use std::sync::{
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	19 atomic::{AtomicUsize, Ordering},
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	20 RwLock, RwLockReadGuard, RwLockWriteGuard,
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	21 };
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	22
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	23 use hg::{
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	24 errors::HgError,
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	25 revlog::{
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	26 index::{Index, RevisionDataParams, SnapshotsCache, INDEX_ENTRY_SIZE},
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	27 inner_revlog::InnerRevlog as CoreInnerRevlog,
52789 34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	28 nodemap::{NodeMap, NodeMapError, NodeTree as CoreNodeTree},
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	29 options::RevlogOpenOptions,
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	30 RevlogError, RevlogIndex, RevlogType,
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	31 },
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	32 utils::files::get_path_from_bytes,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	33 vfs::FnCacheVfs,
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	34 BaseRevision, Revision, UncheckedRevision, NULL_REVISION,
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	35 };
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	36
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	37 use crate::{
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	38 exceptions::{
52795 adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	39 graph_error, map_lock_error, map_try_lock_error, nodemap_error,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	40 rev_not_in_index, revlog_error_bare, revlog_error_from_msg,
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	41 },
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	42 node::{node_from_py_bytes, node_prefix_from_py_bytes, py_node_for_rev},
52796 670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	43 revision::{
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	44 check_revision, rev_pyiter_collect, rev_pyiter_collect_or_else,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	45 revs_py_list, PyRevision,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	46 },
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	47 store::PyFnCache,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	48 util::{new_submodule, take_buffer_with_slice},
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	49 };
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	50
52778 523ca3d225f5 rust-pyo3-revlog: config extraction functions Georges Racinet <georges.racinet@cloudcrane.io> parents: 52775 diff changeset	51 mod config;
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	52 use config::*;
52789 34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	53 mod index;
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	54 use index::{
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	55 py_tuple_to_revision_data_params, revision_data_params_to_py_tuple,
52792 acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	56 PySharedIndex,
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	57 };
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	58
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	59 #[pyclass]
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	60 struct ReadingContextManager {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	61 inner_revlog: Py<InnerRevlog>,
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	62 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	63
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	64 #[pymethods]
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	65 impl ReadingContextManager {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	66 fn __enter__(slf: PyRef<'_, Self>) -> PyResult<()> {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	67 let inner_bound = slf.inner_revlog.bind(slf.py());
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	68 let shareable = &inner_bound.borrow().irl;
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	69 // Safety: the owner is correct and we won't use `share()` anyway
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	70 let core_irl =
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	71 unsafe { shareable.borrow_with_owner(inner_bound) }.read();
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	72 core_irl
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	73 .enter_reading_context()
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	74 .map_err(revlog_error_from_msg)
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	75 .inspect_err(\|_e\| {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	76 // `__exit__` is not called from Python if `__enter__` fails
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	77 core_irl.exit_reading_context();
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	78 })
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	79 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	80
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	81 #[pyo3(signature = (*_args))]
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	82 fn __exit__(slf: PyRef<'_, Self>, _args: &Bound<'_, PyTuple>) {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	83 let inner_bound = slf.inner_revlog.bind(slf.py());
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	84 let shareable = &inner_bound.borrow().irl;
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	85 // Safety: the owner is correct and we won't use `share()` anyway
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	86 let core_irl_ref = unsafe { shareable.borrow_with_owner(inner_bound) };
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	87 core_irl_ref.read().exit_reading_context();
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	88 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	89 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	90
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	91 struct PySnapshotsCache<'a, 'py: 'a>(&'a Bound<'py, PyDict>);
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	92
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	93 impl<'a, 'py> PySnapshotsCache<'a, 'py> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	94 fn insert_for_with_py_result(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	95 &self,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	96 rev: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	97 value: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	98 ) -> PyResult<()> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	99 match self.0.get_item(rev)? {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	100 Some(obj) => obj.downcast::<PySet>()?.add(value),
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	101 None => {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	102 let set = PySet::new(self.0.py(), vec![value])?;
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	103 self.0.set_item(rev, set)
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	104 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	105 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	106 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	107 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	108
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	109 impl<'a, 'py> SnapshotsCache for PySnapshotsCache<'a, 'py> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	110 fn insert_for(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	111 &mut self,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	112 rev: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	113 value: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	114 ) -> Result<(), RevlogError> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	115 self.insert_for_with_py_result(rev, value).map_err(\|_\| {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	116 RevlogError::Other(HgError::unsupported(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	117 "Error in Python caches handling",
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	118 ))
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	119 })
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	120 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	121 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	122
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	123 #[pyclass]
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	124 #[allow(dead_code)]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	125 struct InnerRevlog {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	126 irl: PyShareable<CoreInnerRevlog>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	127 nt: RwLock<Option<CoreNodeTree>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	128 docket: Option<PyObject>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	129 // Holds a reference to the mmap'ed persistent nodemap data
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	130 nodemap_mmap: Option<PyBuffer<u8>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	131 // Holds a reference to the mmap'ed persistent index data
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	132 index_mmap: Option<PyBuffer<u8>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	133 revision_cache: Option<PyObject>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	134 head_revs_py_list: Option<Py<PyList>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	135 head_node_ids_py_list: Option<Py<PyList>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	136 use_persistent_nodemap: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	137 nodemap_queries: AtomicUsize,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	138 }
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	139
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	140 #[pymethods]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	141 impl InnerRevlog {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	142 #[new]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	143 // The Python side has authority on this signature.
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	144 #[allow(clippy::too_many_arguments)]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	145 fn new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	146 vfs_base: &Bound<'_, PyBytes>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	147 fncache: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	148 vfs_is_readonly: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	149 index_data: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	150 index_file: &Bound<'_, PyBytes>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	151 data_file: &Bound<'_, PyBytes>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	152 sidedata_file: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	153 inline: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	154 data_config: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	155 delta_config: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	156 feature_config: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	157 chunk_cache: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	158 default_compression_header: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	159 revlog_type: usize,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	160 use_persistent_nodemap: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	161 ) -> PyResult<Self> {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	162 // Let clippy accept the unused arguments. This is a bit better than
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	163 // a blank `allow` directive
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	164 let _ = sidedata_file;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	165 let _ = chunk_cache;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	166 let _ = default_compression_header;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	167
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	168 let index_file = get_path_from_bytes(index_file.as_bytes()).to_owned();
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	169 let data_file = get_path_from_bytes(data_file.as_bytes()).to_owned();
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	170 let revlog_type = RevlogType::try_from(revlog_type)
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	171 .map_err(revlog_error_from_msg)?;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	172 let data_config = extract_data_config(data_config, revlog_type)?;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	173 let delta_config = extract_delta_config(delta_config, revlog_type)?;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	174 let feature_config =
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	175 extract_feature_config(feature_config, revlog_type)?;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	176 let options = RevlogOpenOptions::new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	177 inline,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	178 data_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	179 delta_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	180 feature_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	181 );
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	182
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	183 // Safety: we keep the buffer around inside the returned instance as
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	184 // `index_mmap`
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	185 let (buf, bytes) = unsafe { take_buffer_with_slice(index_data)? };
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	186 let index = Index::new(bytes, options.index_header())
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	187 .map_err(revlog_error_from_msg)?;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	188
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	189 let base = get_path_from_bytes(vfs_base.as_bytes()).to_owned();
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	190 let core = CoreInnerRevlog::new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	191 Box::new(FnCacheVfs::new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	192 base,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	193 vfs_is_readonly,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	194 Box::new(PyFnCache::new(fncache.clone().unbind())),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	195 )),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	196 index,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	197 index_file,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	198 data_file,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	199 data_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	200 delta_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	201 feature_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	202 );
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	203 Ok(Self {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	204 irl: core.into(),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	205 nt: None.into(),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	206 docket: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	207 nodemap_mmap: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	208 index_mmap: buf.into(),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	209 head_revs_py_list: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	210 head_node_ids_py_list: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	211 revision_cache: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	212 use_persistent_nodemap,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	213 nodemap_queries: AtomicUsize::new(0),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	214 })
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	215 }
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	216
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	217 fn reading(slf: &Bound<'_, Self>) -> PyResult<ReadingContextManager> {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	218 Ok(ReadingContextManager {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	219 inner_revlog: slf.clone().unbind(),
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	220 })
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	221 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	222
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	223 //
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	224 // -- forwarded index methods --
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	225 //
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	226
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	227 fn _index_get_rev(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	228 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	229 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	230 ) -> PyResult<Option<PyRevision>> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	231 let node = node_from_py_bytes(node)?;
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	232
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	233 // Do not rewrite this with `Self::with_index_nt_read`: it makes
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	234 // inconditionally a volatile nodetree, and that is not the intent
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	235 // here: the code below specifically avoids that.
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	236 Self::with_core_read(slf, \|self_ref, irl\| {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	237 let idx = &irl.index;
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	238
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	239 let prev_queries =
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	240 self_ref.nodemap_queries.fetch_add(1, Ordering::Relaxed);
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	241 // Filelogs have no persistent nodemaps and are often small,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	242 // use a brute force lookup from the end
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	243 // backwards. If there is a very large filelog
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	244 // (automation file that changes every
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	245 // commit etc.), it also seems to work quite well for
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	246 // all measured purposes so far.
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	247 if !self_ref.use_persistent_nodemap && prev_queries <= 3 {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	248 return Ok(idx
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	249 .rev_from_node_no_persistent_nodemap(node.into())
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	250 .ok()
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	251 .map(Into::into));
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	252 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	253
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	254 let opt =
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	255 self_ref.get_nodetree(idx)?.read().map_err(map_lock_error)?;
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	256 let nt = opt.as_ref().expect("nodetree should be set");
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	257
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	258 let rust_rev =
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	259 nt.find_bin(idx, node.into()).map_err(nodemap_error)?;
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	260 Ok(rust_rev.map(Into::into))
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	261 })
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	262 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	263
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	264 /// same as `_index_get_rev()` but raises a bare `error.RevlogError` if
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	265 /// node is not found.
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	266 ///
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	267 /// No need to repeat `node` in the exception, `mercurial/revlog.py`
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	268 /// will catch and rewrap with it
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	269 fn _index_rev(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	270 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	271 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	272 ) -> PyResult<PyRevision> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	273 Self::_index_get_rev(slf, node)?.ok_or_else(revlog_error_bare)
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	274 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	275
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	276 /// return True if the node exist in the index
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	277 fn _index_has_node(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	278 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	279 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	280 ) -> PyResult<bool> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	281 Self::_index_get_rev(slf, node).map(\|opt\| opt.is_some())
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	282 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	283
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	284 /// find length of shortest hex nodeid of a binary ID
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	285 fn _index_shortest(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	286 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	287 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	288 ) -> PyResult<usize> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	289 Self::with_index_nt_read(slf, \|idx, nt\| {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	290 match nt.unique_prefix_len_node(idx, &node_from_py_bytes(node)?) {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	291 Ok(Some(l)) => Ok(l),
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	292 Ok(None) => Err(revlog_error_bare()),
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	293 Err(e) => Err(nodemap_error(e)),
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	294 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	295 })
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	296 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	297
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	298 fn _index_partialmatch<'py>(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	299 slf: &Bound<'py, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	300 node: &Bound<'py, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	301 ) -> PyResult<Option<Bound<'py, PyBytes>>> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	302 Self::with_index_nt_read(slf, \|idx, nt\| {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	303 Ok(nt
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	304 .find_bin(idx, node_prefix_from_py_bytes(node)?)
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	305 .map_err(nodemap_error)?
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	306 .map(\|rev\| py_node_for_rev(slf.py(), idx, rev)))
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	307 })
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	308 }
52787 e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	309
52789 34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	310 /// append an index entry
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	311 fn _index_append(
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	312 slf: &Bound<'_, Self>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	313 tup: &Bound<'_, PyTuple>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	314 ) -> PyResult<()> {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	315 // no need to check length: in PyO3 tup.get_item() does return
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	316 // proper errors
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	317 let node_bytes = tup.get_item(7)?.extract()?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	318 let node = node_from_py_bytes(&node_bytes)?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	319
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	320 Self::with_index_nt_write(slf, \|idx, nt\| {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	321 let rev = idx.len() as BaseRevision;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	322 // This is ok since we will immediately add the revision to the
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	323 // index
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	324 let rev = Revision(rev);
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	325 idx.append(py_tuple_to_revision_data_params(tup)?)
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	326 .map_err(revlog_error_from_msg)?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	327
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	328 nt.insert(idx, &node, rev).map_err(nodemap_error)?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	329 Ok(())
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	330 })
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	331 }
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	332
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	333 /// Removes one or several entries from the index.
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	334 ///
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	335 /// Historically, on the Mercurial revlog index, `__delitem__` has always
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	336 /// been both for `del idx[r1]` and `del idx[r1:r2]`. In both cases,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	337 /// all entries starting from `r1` are removed anyway.
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	338 fn _index___delitem__(
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	339 slf: &Bound<'_, Self>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	340 arg: &Bound<'_, PyAny>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	341 ) -> PyResult<()> {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	342 let start = if let Ok(rev) = arg.extract() {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	343 UncheckedRevision(rev)
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	344 } else {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	345 // here we could downcast to `PySlice` and use `indices()`, but
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	346 // the rust-cpython based version could not do that, and
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	347 // `indices()` does some resolving that makes it not equivalent,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	348 // e.g., `idx[-1::]` has `start=0`. As we are currently in
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	349 // transition, we keep it the old way (hoping it was consistent
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	350 // with the C index).
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	351 let start = arg.getattr("start")?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	352 UncheckedRevision(start.extract()?)
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	353 };
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	354
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	355 Self::with_index_nt_write(slf, \|idx, nt\| {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	356 // In the case of a slice, the check is possibly already done by
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	357 // `slice.indices`, which is itself an FFI wrapper for CPython's
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	358 // `PySlice_GetIndicesEx`
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	359 // (Python integration tests will tell us)
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	360 let start = idx.check_revision(start).ok_or_else(\|\| {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	361 nodemap_error(NodeMapError::RevisionNotInIndex(start))
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	362 })?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	363 idx.remove(start).map_err(revlog_error_from_msg)?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	364 nt.invalidate_all();
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	365 Self::fill_nodemap(idx, nt)?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	366 Ok(())
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	367 })
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	368 }
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	369
52796 670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	370 /// reachableroots
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	371 #[pyo3(signature = (*args))]
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	372 fn _index_reachableroots2(
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	373 slf: &Bound<'_, Self>,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	374 py: Python<'_>,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	375 args: &Bound<'_, PyTuple>,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	376 ) -> PyResult<Py<PyList>> {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	377 // TODO what was the point of having a signature with variable args?
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	378 let min_root = UncheckedRevision(args.get_item(0)?.extract()?);
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	379 let heads = args.get_item(1)?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	380 let roots = args.get_item(2)?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	381 let include_path: bool = args.get_item(3)?.extract()?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	382
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	383 let as_set = Self::with_index_read(slf, \|idx\| {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	384 let heads = rev_pyiter_collect_or_else(&heads, idx, \|_rev\| {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	385 PyIndexError::new_err("head out of range")
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	386 })?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	387 let roots: Result<_, _> = roots
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	388 .try_iter()?
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	389 .map(\|r\| {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	390 r.and_then(\|o\| match o.extract::<PyRevision>() {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	391 Ok(r) => Ok(UncheckedRevision(r.0)),
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	392 Err(e) => Err(e),
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	393 })
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	394 })
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	395 .collect();
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	396 idx.reachable_roots(min_root, heads, roots?, include_path)
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	397 .map_err(graph_error)
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	398 })?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	399
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	400 revs_py_list(py, as_set)
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	401 }
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	402
52795 adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	403 #[pyo3(signature = (*args))]
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	404 fn _index_headrevs(
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	405 slf: &Bound<'_, Self>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	406 py: Python<'_>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	407 args: &Bound<'_, PyTuple>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	408 ) -> PyResult<Py<PyList>> {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	409 let (filtered_revs, stop_rev) = match args.len() {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	410 0 => Ok((None, None)),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	411 1 => Ok((Some(args.get_item(0)?), None)),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	412 2 => Ok((Some(args.get_item(0)?), Some(args.get_item(1)?))),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	413 _ => Err(PyTypeError::new_err("too many arguments")),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	414 }?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	415 let stop_rev = stop_rev
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	416 .map(\|o\| o.extract::<Option<i32>>())
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	417 .transpose()?
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	418 .flatten();
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	419 let filtered_revs = filtered_revs.filter(\|o\| !o.is_none());
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	420
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	421 let (from_core, stop_rev) = Self::with_index_read(slf, \|idx\| {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	422 let stop_rev = stop_rev
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	423 // should this not just be the normal checking?
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	424 .filter(\|rev\| 0 <= rev && rev < idx.len() as BaseRevision)
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	425 .map(Revision);
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	426
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	427 let from_core = if let Some(filtered_revs) = filtered_revs {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	428 let filtered_revs = rev_pyiter_collect(&filtered_revs, idx)?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	429 idx.head_revs_advanced(
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	430 &filtered_revs,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	431 stop_rev,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	432 stop_rev.is_none(),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	433 )
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	434 } else if stop_rev.is_some() {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	435 idx.head_revs_advanced(&HashSet::new(), stop_rev, false)
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	436 } else {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	437 idx.head_revs_shortcut()
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	438 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	439 .map_err(graph_error)?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	440 Ok((from_core, stop_rev))
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	441 })?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	442
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	443 if stop_rev.is_some() {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	444 // we don't cache result for now
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	445 let new_heads =
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	446 from_core.expect("this case should not be cached yet");
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	447
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	448 revs_py_list(py, new_heads)
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	449 } else {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	450 if let Some(new_heads) = from_core {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	451 Self::cache_new_heads_py_list(slf, new_heads)?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	452 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	453
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	454 Ok(slf
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	455 .borrow()
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	456 .head_revs_py_list
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	457 .as_ref()
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	458 .expect("head revs should be cached")
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	459 .clone_ref(py))
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	460 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	461 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	462
52797 f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	463 /// True if the object is a snapshot
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	464 fn _index_issnapshot(
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	465 slf: &Bound<'_, Self>,
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	466 rev: PyRevision,
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	467 ) -> PyResult<bool> {
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	468 let rev: UncheckedRevision = rev.into();
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	469 let rev = Self::with_index_read(slf, \|idx\| {
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	470 idx.check_revision(rev).ok_or_else(\|\| rev_not_in_index(rev))
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	471 })?;
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	472 Self::with_core_read(slf, \|_self_ref, irl\| {
52797 f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	473 irl.is_snapshot(rev)
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	474 .map_err(\|e\| PyValueError::new_err(e.to_string()))
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	475 })
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	476 }
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	477
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	478 /// Gather snapshot data in a cache dict
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	479 fn _index_findsnapshots(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	480 slf: &Bound<'_, Self>,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	481 cache: &Bound<'_, PyDict>,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	482 start_rev: PyRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	483 end_rev: PyRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	484 ) -> PyResult<()> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	485 let mut cache = PySnapshotsCache(cache);
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	486 Self::with_index_read(slf, \|idx\| {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	487 idx.find_snapshots(start_rev.into(), end_rev.into(), &mut cache)
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	488 .map_err(\|_\| revlog_error_bare())
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	489 })?;
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	490 Ok(())
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	491 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	492
52787 e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	493 fn _index___len__(slf: &Bound<'_, Self>) -> PyResult<usize> {
e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	494 Self::with_index_read(slf, \|idx\| Ok(idx.len()))
e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	495 }
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	496
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	497 fn _index___getitem__(
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	498 slf: &Bound<'_, Self>,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	499 py: Python<'_>,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	500 key: &Bound<'_, PyAny>,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	501 ) -> PyResult<PyObject> {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	502 Self::with_index_read(slf, \|idx\| {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	503 match key.extract::<BaseRevision>() {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	504 Ok(key_as_int) => {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	505 let entry_params = if key_as_int == NULL_REVISION.0 {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	506 RevisionDataParams::default()
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	507 } else {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	508 let rev = UncheckedRevision(key_as_int);
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	509 match idx.entry_as_params(rev) {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	510 Some(e) => e,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	511 None => {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	512 return Err(PyIndexError::new_err(
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	513 "revlog index out of range",
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	514 ));
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	515 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	516 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	517 };
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	518 Ok(revision_data_params_to_py_tuple(py, entry_params)?
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	519 .into_any()
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	520 .unbind())
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	521 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	522 // Case when key is a binary Node ID (lame: we're re-unlocking)
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	523 _ => Self::_index_get_rev(slf, key.downcast::<PyBytes>()?)?
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	524 .map_or_else(
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	525 \|\| Ok(py.None()),
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	526 \|py_rev\| Ok(py_rev.into_pyobject(py)?.unbind().into()),
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	527 ),
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	528 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	529 })
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	530 }
52798 88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	531
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	532 #[getter]
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	533 fn _index_entry_size(&self) -> usize {
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	534 INDEX_ENTRY_SIZE
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	535 }
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	536
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	537 #[getter]
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	538 fn _index_rust_ext_compat(&self) -> i32 {
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	539 1
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	540 }
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	541
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	542 #[getter]
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	543 fn _index_is_rust(&self) -> bool {
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	544 true
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	545 }
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	546 }
52778 523ca3d225f5 rust-pyo3-revlog: config extraction functions Georges Racinet <georges.racinet@cloudcrane.io> parents: 52775 diff changeset	547
52782 827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	548 impl InnerRevlog {
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	549 /// Take the lock on `slf.irl` for reading and call a closure.
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	550 ///
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	551 /// This serves the purpose to keep the needed intermediate [`PyRef`]
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	552 /// that must be obtained to access the data from the [`Bound`] reference
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	553 /// and of which the locked [`CoreInnerRevlog`] depends.
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	554 /// This also provides releasing of the [`PyRef`] as soon as the closure
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	555 /// is done, which is crucial if the caller needs to obtain a [`PyRefMut`]
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	556 /// later on.
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	557 ///
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	558 /// In the closure, we hand back the intermediate [`PyRef`] that
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	559 /// has been generated so that the closure can access more attributes.
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	560 fn with_core_read<'py, T>(
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	561 slf: &Bound<'py, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	562 f: impl FnOnce(
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	563 &PyRef<'py, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	564 RwLockReadGuard<CoreInnerRevlog>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	565 ) -> PyResult<T>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	566 ) -> PyResult<T> {
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	567 let self_ref = slf.borrow();
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	568 // Safety: the owner is the right one. We will anyway
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	569 // not actually `share` it. Perhaps pyo3-sharedref should provide
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	570 // something less scary for this kind of usage.
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	571 let shareable_ref = unsafe { self_ref.irl.borrow_with_owner(slf) };
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	572 let guard = shareable_ref.try_read().map_err(map_try_lock_error)?;
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	573 f(&self_ref, guard)
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	574 }
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	575
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	576 /// Take the lock on `slf.irl` for writing and call a closure.
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	577 ///
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	578 /// See [`Self::with_core_read`] for more explanations.
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	579 fn with_core_write<'py, T>(
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	580 slf: &Bound<'py, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	581 f: impl FnOnce(
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	582 &PyRef<'py, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	583 RwLockWriteGuard<CoreInnerRevlog>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	584 ) -> PyResult<T>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	585 ) -> PyResult<T> {
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	586 let self_ref = slf.borrow();
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	587 // Safety: the owner is the right one. We will anyway
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	588 // not actually `share` it. Perhaps pyo3-sharedref should provide
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	589 // something less scary for this kind of usage.
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	590 let shareable_ref = unsafe { self_ref.irl.borrow_with_owner(slf) };
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	591 let guard = shareable_ref.try_write().map_err(map_try_lock_error)?;
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	592 f(&self_ref, guard)
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	593 }
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	594
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	595 fn with_index_read<T>(
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	596 slf: &Bound<'_, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	597 f: impl FnOnce(&Index) -> PyResult<T>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	598 ) -> PyResult<T> {
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	599 Self::with_core_read(slf, \|_, guard\| f(&guard.index))
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	600 }
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	601
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	602 #[allow(dead_code)]
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	603 fn with_index_write<T>(
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	604 slf: &Bound<'_, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	605 f: impl FnOnce(&mut Index) -> PyResult<T>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	606 ) -> PyResult<T> {
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	607 Self::with_core_write(slf, \|_, mut guard\| f(&mut guard.index))
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	608 }
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	609
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	610 /// Lock `slf` for reading and execute a closure on its [`Index`] and
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	611 /// [`NodeTree`]
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	612 ///
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	613 /// The [`NodeTree`] is initialized an filled before hand if needed.
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	614 fn with_index_nt_read<T>(
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	615 slf: &Bound<'_, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	616 f: impl FnOnce(&Index, &CoreNodeTree) -> PyResult<T>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	617 ) -> PyResult<T> {
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	618 Self::with_core_read(slf, \|self_ref, guard\| {
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	619 let idx = &guard.index;
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	620 let nt =
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	621 self_ref.get_nodetree(idx)?.read().map_err(map_lock_error)?;
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	622 let nt = nt.as_ref().expect("nodetree should be set");
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	623 f(idx, nt)
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	624 })
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	625 }
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	626
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	627 fn with_index_nt_write<T>(
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	628 slf: &Bound<'_, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	629 f: impl FnOnce(&mut Index, &mut CoreNodeTree) -> PyResult<T>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	630 ) -> PyResult<T> {
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	631 Self::with_core_write(slf, \|self_ref, mut guard\| {
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	632 let idx = &mut guard.index;
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	633 let mut nt = self_ref
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	634 .get_nodetree(idx)?
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	635 .write()
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	636 .map_err(map_lock_error)?;
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	637 let nt = nt.as_mut().expect("nodetree should be set");
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	638 f(idx, nt)
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	639 })
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	640 }
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	641
52782 827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	642 /// Fill a [`CoreNodeTree`] by doing a full iteration on the given
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	643 /// [`Index`]
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	644 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	645 /// # Python exceptions
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	646 /// Raises `ValueError` if `nt` has existing data that is inconsistent
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	647 /// with `idx`.
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	648 fn fill_nodemap(idx: &Index, nt: &mut CoreNodeTree) -> PyResult<()> {
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	649 for r in 0..idx.len() {
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	650 let rev = Revision(r as BaseRevision);
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	651 // in this case node() won't ever return None
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	652 nt.insert(idx, idx.node(rev).expect("node should exist"), rev)
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	653 .map_err(nodemap_error)?
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	654 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	655 Ok(())
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	656 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	657
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	658 /// Return a working NodeTree of this InnerRevlog
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	659 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	660 /// In case the NodeTree has not been initialized yet (in particular
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	661 /// not from persistent data at instantiation), it is created and
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	662 /// filled right away from the index.
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	663 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	664 /// Technically, the returned NodeTree is still behind the lock of
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	665 /// the `nt` field, hence still wrapped in an [`Option`]. Callers
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	666 /// will need to take the lock and unwrap with `expect()`.
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	667 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	668 /// # Python exceptions
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	669 /// The case mentioned in [`Self::fill_nodemap()`] cannot happen, as the
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	670 /// NodeTree is empty when it is called.
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	671 fn get_nodetree(
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	672 &self,
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	673 idx: &Index,
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	674 ) -> PyResult<&RwLock<Option<CoreNodeTree>>> {
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	675 if self.nt.read().map_err(map_lock_error)?.is_none() {
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	676 let readonly = Box::<Vec<_>>::default();
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	677 let mut nt = CoreNodeTree::load_bytes(readonly, 0);
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	678 Self::fill_nodemap(idx, &mut nt)?;
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	679 self.nt.write().map_err(map_lock_error)?.replace(nt);
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	680 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	681 Ok(&self.nt)
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	682 }
52795 adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	683
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	684 fn cache_new_heads_py_list(
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	685 slf: &Bound<'_, Self>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	686 new_heads: Vec<Revision>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	687 ) -> PyResult<Py<PyList>> {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	688 let py = slf.py();
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	689 let new_heads_py_list = revs_py_list(py, new_heads)?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	690 slf.borrow_mut().head_revs_py_list =
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	691 Some(new_heads_py_list.clone_ref(py));
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	692 // TODO is returning really useful?
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	693 Ok(new_heads_py_list)
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	694 }
52782 827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	695 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	696
52792 acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	697 #[pyclass]
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	698 struct NodeTree {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	699 nt: RwLock<CoreNodeTree>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	700 index: SharedByPyObject<PySharedIndex>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	701 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	702
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	703 #[pymethods]
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	704 impl NodeTree {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	705 #[new]
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	706 // The share/mapping should be set apart to become the PyO3 homolog of
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	707 // `py_rust_index_to_graph`
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	708 fn new(index_proxy: &Bound<'_, PyAny>) -> PyResult<Self> {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	709 let py_irl = index_proxy.getattr("inner")?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	710 let py_irl_ref = py_irl.downcast::<InnerRevlog>()?.borrow();
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	711 let shareable_irl = &py_irl_ref.irl;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	712
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	713 // Safety: the owner is the actual one and we do not leak any
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	714 // internal reference.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	715 let index = unsafe {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	716 shareable_irl.share_map(&py_irl, \|irl\| (&irl.index).into())
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	717 };
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	718 let nt = CoreNodeTree::default(); // in-RAM, fully mutable
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	719
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	720 Ok(Self {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	721 nt: nt.into(),
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	722 index,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	723 })
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	724 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	725
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	726 /// Tell whether the NodeTree is still valid
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	727 ///
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	728 /// In case of mutation of the index, the given results are not
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	729 /// guaranteed to be correct, and in fact, the methods borrowing
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	730 /// the inner index would fail because of `PySharedRef` poisoning
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	731 /// (generation-based guard), same as iterating on a `dict` that has
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	732 /// been meanwhile mutated.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	733 fn is_invalidated(&self, py: Python<'_>) -> PyResult<bool> {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	734 // Safety: we don't leak any reference derived from self.index, as
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	735 // we only check errors
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	736 let result = unsafe { self.index.try_borrow(py) };
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	737 // two cases for result to be an error:
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	738 // - the index has previously been mutably borrowed
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	739 // - there is currently a mutable borrow
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	740 // in both cases this amounts for previous results related to
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	741 // the index to still be valid.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	742 Ok(result.is_err())
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	743 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	744
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	745 fn insert(&self, py: Python<'_>, rev: PyRevision) -> PyResult<()> {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	746 // Safety: we don't leak any reference derived from self.index,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	747 // as `nt.insert` does not store direct references
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	748 let idx = &*unsafe { self.index.try_borrow(py)? };
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	749
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	750 let rev = check_revision(idx, rev)?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	751 if rev == NULL_REVISION {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	752 return Err(rev_not_in_index(rev.into()));
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	753 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	754
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	755 let entry = idx.inner().get_entry(rev).expect("entry should exist");
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	756 let mut nt = self.nt.write().map_err(map_lock_error)?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	757 nt.insert(idx, entry.hash(), rev).map_err(nodemap_error)
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	758 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	759
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	760 fn shortest(
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	761 &self,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	762 py: Python<'_>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	763 node: &Bound<'_, PyBytes>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	764 ) -> PyResult<usize> {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	765 let nt = self.nt.read().map_err(map_lock_error)?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	766 // Safety: we don't leak any reference derived from self.index
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	767 // as returned type is Copy
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	768 let idx = &*unsafe { self.index.try_borrow(py)? };
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	769 nt.unique_prefix_len_node(idx, &node_from_py_bytes(node)?)
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	770 .map_err(nodemap_error)?
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	771 .ok_or_else(revlog_error_bare)
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	772 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	773
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	774 /// Lookup by node hex prefix in the NodeTree, returning revision number.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	775 ///
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	776 /// This is not part of the classical NodeTree API, but is good enough
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	777 /// for unit testing, as in `test-rust-revlog.py`.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	778 fn prefix_rev_lookup(
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	779 &self,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	780 py: Python<'_>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	781 node_prefix: &Bound<'_, PyBytes>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	782 ) -> PyResult<Option<PyRevision>> {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	783 let prefix = node_prefix_from_py_bytes(node_prefix)?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	784 let nt = self.nt.read().map_err(map_lock_error)?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	785 // Safety: we don't leak any reference derived from self.index
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	786 // as returned type is Copy
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	787 let idx = &*unsafe { self.index.try_borrow(py)? };
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	788 Ok(nt
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	789 .find_bin(idx, prefix)
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	790 .map_err(nodemap_error)?
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	791 .map(\|r\| r.into()))
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	792 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	793 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	794
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	795 pub fn init_module<'py>(
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	796 py: Python<'py>,
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	797 package: &str,
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	798 ) -> PyResult<Bound<'py, PyModule>> {
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	799 let m = new_submodule(py, package, "revlog")?;
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	800 m.add_class::<InnerRevlog>()?;
52792 acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	801 m.add_class::<NodeTree>()?;
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	802 m.add_class::<ReadingContextManager>()?;
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	803 Ok(m)
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	804 }

Mercurial > public > mercurial-scm > hg

annotate rust/hg-pyo3/src/revlog/mod.rs @ 52799:798355e46d03