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

annotate rust/hg-pyo3/src/revlog/mod.rs @ 52832:c72d8df0c080

rust-pyo3-revlog: clear_cache

author	Rapha?l Gom?s <rgomes@octobus.net>
date	Sun, 05 Jan 2025 23:20:01 +0100
parents	9e6b50837b37
children	b7dd7af61488

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)]
52820 503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	10 use hg::revlog::index::IndexHeader;
52812 6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	11 use hg::revlog::nodemap::Block;
52814 3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	12 use hg::utils::files::get_bytes_from_path;
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	13 use pyo3::buffer::PyBuffer;
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	14 use pyo3::conversion::IntoPyObject;
52797 f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	15 use pyo3::exceptions::{PyIndexError, PyTypeError, PyValueError};
52805 a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	16 use pyo3::types::{
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	17 PyBool, PyBytes, PyBytesMethods, PyDict, PyList, PySet, PyTuple,
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	18 };
52812 6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	19 use pyo3::{prelude::*, IntoPyObjectExt};
52792 acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	20 use pyo3_sharedref::{PyShareable, SharedByPyObject};
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	21
52808 6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	22 use std::collections::{HashMap, HashSet};
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	23 use std::sync::{
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	24 atomic::{AtomicUsize, Ordering},
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	25 RwLock, RwLockReadGuard, RwLockWriteGuard,
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	26 };
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	27
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	28 use hg::{
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	29 errors::HgError,
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	30 revlog::{
52808 6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	31 index::{
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	32 Index, Phase, RevisionDataParams, SnapshotsCache, INDEX_ENTRY_SIZE,
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	33 },
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	34 inner_revlog::InnerRevlog as CoreInnerRevlog,
52789 34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	35 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	36 options::RevlogOpenOptions,
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	37 RevlogError, RevlogIndex, RevlogType,
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	38 },
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	39 utils::files::get_path_from_bytes,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	40 vfs::FnCacheVfs,
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	41 BaseRevision, Revision, UncheckedRevision, NULL_REVISION,
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	42 };
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	43
52831 9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	44 use crate::util::PyBytesDeref;
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	45 use crate::{
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	46 exceptions::{
52795 adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	47 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	48 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	49 },
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	50 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	51 revision::{
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	52 check_revision, rev_pyiter_collect, rev_pyiter_collect_or_else,
52808 6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	53 revs_py_list, revs_py_set, PyRevision,
52796 670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	54 },
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	55 store::PyFnCache,
52828 e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	56 transaction::PyTransaction,
52823 9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	57 util::{new_submodule, take_buffer_with_slice, with_pybytes_buffer},
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	58 };
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	59
52778 523ca3d225f5 rust-pyo3-revlog: config extraction functions Georges Racinet <georges.racinet@cloudcrane.io> parents: 52775 diff changeset	60 mod config;
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	61 use config::*;
52789 34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	62 mod index;
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	63 use index::{
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	64 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	65 PySharedIndex,
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	66 };
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	67
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	68 #[pyclass]
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	69 struct ReadingContextManager {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	70 inner_revlog: Py<InnerRevlog>,
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	71 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	72
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	73 #[pymethods]
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	74 impl ReadingContextManager {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	75 fn __enter__(slf: PyRef<'_, Self>) -> PyResult<()> {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	76 let inner_bound = slf.inner_revlog.bind(slf.py());
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	77 let shareable = &inner_bound.borrow().irl;
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	78 // 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	79 let core_irl =
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	80 unsafe { shareable.borrow_with_owner(inner_bound) }.read();
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	81 core_irl
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	82 .enter_reading_context()
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	83 .map_err(revlog_error_from_msg)
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	84 .inspect_err(\|_e\| {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	85 // `__exit__` is not called from Python if `__enter__` fails
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	86 core_irl.exit_reading_context();
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	87 })
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 #[pyo3(signature = (*_args))]
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	91 fn __exit__(slf: PyRef<'_, Self>, _args: &Bound<'_, PyTuple>) {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	92 let inner_bound = slf.inner_revlog.bind(slf.py());
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	93 let shareable = &inner_bound.borrow().irl;
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	94 // 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	95 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	96 core_irl_ref.read().exit_reading_context();
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	97 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	98 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	99
52829 c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	100 #[pyclass]
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	101 struct WritingContextManager {
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	102 inner_revlog: Py<InnerRevlog>,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	103 transaction: RwLock<PyTransaction>,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	104 data_end: Option<usize>,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	105 }
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	106
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	107 #[pymethods]
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	108 impl WritingContextManager {
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	109 fn __enter__(slf: PyRefMut<'_, Self>) -> PyResult<()> {
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	110 let inner_bound = slf.inner_revlog.bind(slf.py());
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	111 let shareable = &inner_bound.borrow_mut().irl;
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	112 // Safety: the owner is correct and we won't use `share()` anyway
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	113 let mut core_irl =
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	114 unsafe { shareable.borrow_with_owner(inner_bound) }.write();
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	115 core_irl
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	116 .enter_writing_context(
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	117 slf.data_end,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	118 &mut *slf
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	119 .transaction
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	120 .try_write()
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	121 .expect("transaction should be protected by the GIL"),
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	122 )
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	123 .map_err(revlog_error_from_msg)
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	124 .inspect_err(\|_e\| {
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	125 // `__exit__` is not called from Python if `__enter__` fails
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	126 core_irl.exit_writing_context();
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	127 })
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	128 }
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	129
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	130 #[pyo3(signature = (*_args))]
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	131 fn __exit__(slf: PyRef<'_, Self>, _args: &Bound<'_, PyTuple>) {
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	132 let inner_bound = slf.inner_revlog.bind(slf.py());
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	133 let shareable = &inner_bound.borrow().irl;
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	134 // Safety: the owner is correct and we won't use `share()` anyway
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	135 let core_irl_ref = unsafe { shareable.borrow_with_owner(inner_bound) };
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	136 core_irl_ref.write().exit_writing_context();
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	137 }
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	138 }
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	139
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	140 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	141
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	142 impl<'a, 'py> PySnapshotsCache<'a, 'py> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	143 fn insert_for_with_py_result(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	144 &self,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	145 rev: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	146 value: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	147 ) -> PyResult<()> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	148 match self.0.get_item(rev)? {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	149 Some(obj) => obj.downcast::<PySet>()?.add(value),
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	150 None => {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	151 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	152 self.0.set_item(rev, set)
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	153 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	154 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	155 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	156 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	157
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	158 impl<'a, 'py> SnapshotsCache for PySnapshotsCache<'a, 'py> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	159 fn insert_for(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	160 &mut self,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	161 rev: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	162 value: BaseRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	163 ) -> Result<(), RevlogError> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	164 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	165 RevlogError::Other(HgError::unsupported(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	166 "Error in Python caches handling",
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	167 ))
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	168 })
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	169 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	170 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	171
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	172 #[pyclass]
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	173 #[allow(dead_code)]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	174 struct InnerRevlog {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	175 irl: PyShareable<CoreInnerRevlog>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	176 nt: RwLock<Option<CoreNodeTree>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	177 docket: Option<PyObject>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	178 // 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	179 nodemap_mmap: Option<PyBuffer<u8>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	180 // 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	181 index_mmap: Option<PyBuffer<u8>>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	182 revision_cache: Option<PyObject>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	183 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	184 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	185 use_persistent_nodemap: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	186 nodemap_queries: AtomicUsize,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	187 }
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 #[pymethods]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	190 impl InnerRevlog {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	191 #[new]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	192 // 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	193 #[allow(clippy::too_many_arguments)]
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	194 fn new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	195 vfs_base: &Bound<'_, PyBytes>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	196 fncache: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	197 vfs_is_readonly: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	198 index_data: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	199 index_file: &Bound<'_, PyBytes>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	200 data_file: &Bound<'_, PyBytes>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	201 sidedata_file: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	202 inline: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	203 data_config: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	204 delta_config: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	205 feature_config: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	206 chunk_cache: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	207 default_compression_header: &Bound<'_, PyAny>,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	208 revlog_type: usize,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	209 use_persistent_nodemap: bool,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	210 ) -> PyResult<Self> {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	211 // 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	212 // a blank `allow` directive
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	213 let _ = sidedata_file;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	214 let _ = chunk_cache;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	215 let _ = default_compression_header;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	216
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	217 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	218 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	219 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	220 .map_err(revlog_error_from_msg)?;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	221 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	222 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	223 let feature_config =
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	224 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	225 let options = RevlogOpenOptions::new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	226 inline,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	227 data_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	228 delta_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	229 feature_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	230 );
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	231
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	232 // 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	233 // `index_mmap`
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	234 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	235 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	236 .map_err(revlog_error_from_msg)?;
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	237
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	238 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	239 let core = CoreInnerRevlog::new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	240 Box::new(FnCacheVfs::new(
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	241 base,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	242 vfs_is_readonly,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	243 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	244 )),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	245 index,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	246 index_file,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	247 data_file,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	248 data_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	249 delta_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	250 feature_config,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	251 );
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	252 Ok(Self {
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	253 irl: core.into(),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	254 nt: None.into(),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	255 docket: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	256 nodemap_mmap: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	257 index_mmap: buf.into(),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	258 head_revs_py_list: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	259 head_node_ids_py_list: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	260 revision_cache: None,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	261 use_persistent_nodemap,
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	262 nodemap_queries: AtomicUsize::new(0),
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	263 })
42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	264 }
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	265
52814 3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	266 #[getter]
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	267 fn canonical_index_file(
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	268 slf: &Bound<'_, Self>,
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	269 py: Python<'_>,
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	270 ) -> PyResult<Py<PyBytes>> {
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	271 Self::with_core_read(slf, \|_self_ref, irl\| {
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	272 let path = irl.canonical_index_file();
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	273 Ok(PyBytes::new(py, &get_bytes_from_path(path)).into())
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	274 })
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	275 }
3fcd86374074 rust-pyo3-revlog: canonical_index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52813 diff changeset	276
52815 4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	277 #[getter]
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	278 fn is_delaying(slf: &Bound<'_, Self>) -> PyResult<bool> {
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	279 Self::with_core_read(slf, \|_self_ref, irl\| Ok(irl.is_delaying()))
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	280 }
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	281
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	282 #[getter]
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	283 fn inline(slf: &Bound<'_, Self>) -> PyResult<bool> {
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	284 Self::with_core_read(slf, \|_self_ref, irl\| Ok(irl.is_inline()))
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	285 }
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	286
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	287 #[setter]
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	288 fn set_inline(slf: &Bound<'_, Self>, inline: bool) -> PyResult<()> {
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	289 Self::with_core_write(slf, \|_self_ref, mut irl\| {
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	290 irl.inline = inline;
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	291 Ok(())
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	292 })
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	293 }
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	294
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	295 #[getter]
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	296 fn is_writing(slf: &Bound<'_, Self>) -> PyResult<bool> {
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	297 Self::with_core_read(slf, \|_self_ref, irl\| Ok(irl.is_writing()))
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	298 }
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	299
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	300 #[getter]
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	301 fn is_open(slf: &Bound<'_, Self>) -> PyResult<bool> {
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	302 Self::with_core_read(slf, \|_self_ref, irl\| Ok(irl.is_open()))
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	303 }
4e58167a72a5 rust-pyo3-revlog: trivial properties Rapha?l Gom?s <rgomes@octobus.net> parents: 52814 diff changeset	304
52816 6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	305 #[getter]
52831 9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	306 fn _revisioncache(&self, py: Python<'_>) -> PyResult<PyObject> {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	307 match &self.revision_cache {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	308 None => Ok(py.None()),
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	309 Some(cache) => Ok(cache.clone_ref(py)),
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	310 }
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	311 }
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	312
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	313 #[setter]
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	314 fn set__revisioncache(
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	315 slf: &Bound<'_, Self>,
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	316 py: Python<'_>,
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	317 value: Option<PyObject>,
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	318 ) -> PyResult<()> {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	319 let mut self_ref = slf.borrow_mut();
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	320 self_ref.revision_cache = value.as_ref().map(\|v\| v.clone_ref(py));
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	321
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	322 match value {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	323 None => {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	324 // This means the property has been deleted, not that the
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	325 // property has been set to `None`. Whatever happens is up
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	326 // to the implementation. Here we just set it to `None`.
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	327 self_ref.revision_cache.take();
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	328 }
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	329 Some(tuple) => {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	330 if tuple.is_none(py) {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	331 self_ref.revision_cache.take();
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	332 return Ok(());
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	333 }
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	334 drop(self_ref);
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	335 let tuple: &Bound<'_, PyTuple> = tuple.downcast_bound(py)?;
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	336 let node = tuple.get_item(0)?;
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	337 let node = node_from_py_bytes(node.downcast()?)?;
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	338 let rev: BaseRevision = tuple.get_item(1)?.extract()?;
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	339 // Ok because Python only sets this if the revision has been
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	340 // checked
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	341 let rev = Revision(rev);
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	342 let data = tuple.get_item(2)?;
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	343 let bytes = data.downcast_into::<PyBytes>()?.unbind();
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	344 Self::with_core_read(slf, \|_self_ref, irl\| {
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	345 let mut last_revision_cache = irl
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	346 .last_revision_cache
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	347 .lock()
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	348 .expect("lock should not be held");
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	349 *last_revision_cache = Some((
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	350 node,
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	351 rev,
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	352 Box::new(PyBytesDeref::new(py, bytes)),
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	353 ));
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	354 Ok(())
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	355 })?;
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	356 }
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	357 }
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	358 Ok(())
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	359 }
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	360
9e6b50837b37 rust-pyo3-revlog: _revisioncache Rapha?l Gom?s <rgomes@octobus.net> parents: 52829 diff changeset	361 #[getter]
52816 6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	362 fn index_file(
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	363 slf: &Bound<'_, Self>,
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	364 py: Python<'_>,
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	365 ) -> PyResult<Py<PyBytes>> {
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	366 Self::with_core_read(slf, \|_self_ref, irl\| {
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	367 let path = get_bytes_from_path(&irl.index_file);
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	368 Ok(PyBytes::new(py, &path).unbind())
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	369 })
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	370 }
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	371
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	372 #[setter]
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	373 fn set_index_file(
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	374 slf: &Bound<'_, Self>,
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	375 path: &Bound<'_, PyBytes>,
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	376 ) -> PyResult<()> {
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	377 Self::with_core_write(slf, \|_self_ref, mut irl\| {
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	378 let path = get_path_from_bytes(path.as_bytes());
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	379 path.clone_into(&mut irl.index_file);
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	380 Ok(())
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	381 })
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	382 }
6c121468b026 rust-pyo3-revlog: index_file property Rapha?l Gom?s <rgomes@octobus.net> parents: 52815 diff changeset	383
52832 c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	384 fn clear_cache(slf: &Bound<'_, Self>) -> PyResult<PyObject> {
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	385 assert!(!Self::is_delaying(slf)?);
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	386 let mut self_ref = slf.borrow_mut();
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	387 self_ref.revision_cache.take();
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	388 self_ref.nodemap_queries.store(0, Ordering::Relaxed);
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	389 drop(self_ref);
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	390
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	391 Self::with_core_write(slf, \|_self_ref, mut irl\| {
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	392 irl.clear_cache();
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	393 Ok(slf.py().None())
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	394 })
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	395 }
c72d8df0c080 rust-pyo3-revlog: clear_cache Rapha?l Gom?s <rgomes@octobus.net> parents: 52831 diff changeset	396
52817 5950957af8a3 rust-pyo3-revlog: issnapshot Rapha?l Gom?s <rgomes@octobus.net> parents: 52816 diff changeset	397 fn issnapshot(slf: &Bound<'_, Self>, rev: PyRevision) -> PyResult<bool> {
5950957af8a3 rust-pyo3-revlog: issnapshot Rapha?l Gom?s <rgomes@octobus.net> parents: 52816 diff changeset	398 Self::_index_issnapshot(slf, rev)
5950957af8a3 rust-pyo3-revlog: issnapshot Rapha?l Gom?s <rgomes@octobus.net> parents: 52816 diff changeset	399 }
5950957af8a3 rust-pyo3-revlog: issnapshot Rapha?l Gom?s <rgomes@octobus.net> parents: 52816 diff changeset	400
52818 acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	401 #[pyo3(signature = (rev, stoprev=None))]
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	402 fn _deltachain(
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	403 slf: &Bound<'_, Self>,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	404 py: Python<'_>,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	405 rev: PyRevision,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	406 stoprev: Option<PyRevision>,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	407 ) -> PyResult<Py<PyTuple>> {
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	408 Self::with_index_read(slf, \|idx\| {
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	409 let using_general_delta = idx.uses_generaldelta();
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	410 Self::_index_deltachain(
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	411 slf,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	412 py,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	413 rev,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	414 stoprev,
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	415 Some(using_general_delta.into()),
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	416 )
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	417 })
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	418 }
acd76143f868 rust-pyo3-revlog: _deltachain Rapha?l Gom?s <rgomes@octobus.net> parents: 52817 diff changeset	419
52819 88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	420 fn compress(
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	421 slf: &Bound<'_, Self>,
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	422 py: Python<'_>,
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	423 data: &Bound<'_, PyAny>,
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	424 ) -> PyResult<Py<PyTuple>> {
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	425 Self::with_core_read(slf, \|_self_ref, irl\| {
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	426 // Safety: we only hold on to the data for as long as `_buf`
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	427 // is alive
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	428 let (_buf, data) = unsafe { take_buffer_with_slice(data)? };
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	429 let compressed =
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	430 irl.compress(&data).map_err(revlog_error_from_msg)?;
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	431 let compressed = compressed.as_deref();
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	432 let header = if compressed.is_some() {
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	433 PyBytes::new(py, &b""[..])
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	434 } else {
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	435 PyBytes::new(py, &b"u"[..])
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	436 };
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	437 Ok(PyTuple::new(
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	438 py,
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	439 &[header, PyBytes::new(py, compressed.unwrap_or(&data))],
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	440 )?
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	441 .unbind())
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	442 })
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	443 }
88a69ebbba3b rust-pyo3-revlog: compress Rapha?l Gom?s <rgomes@octobus.net> parents: 52818 diff changeset	444
52820 503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	445 #[pyo3(signature = (tr, header, new_index_file_path=None))]
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	446 fn split_inline(
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	447 slf: &Bound<'_, Self>,
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	448 py: Python<'_>,
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	449 tr: PyObject,
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	450 header: i32,
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	451 new_index_file_path: Option<&Bound<'_, PyBytes>>,
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	452 ) -> PyResult<Py<PyBytes>> {
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	453 // Also unused in Python, TODO clean this up.
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	454 let _ = tr;
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	455
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	456 Self::with_core_write(slf, \|_self_ref, mut irl\| {
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	457 let new_index_file_path = new_index_file_path
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	458 .map(\|path\| get_path_from_bytes(path.as_bytes()).to_owned());
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	459 let header = IndexHeader::parse(&header.to_be_bytes())
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	460 .expect("invalid header bytes");
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	461 let old_path = irl
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	462 .split_inline(header, new_index_file_path)
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	463 .map_err(revlog_error_from_msg)?;
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	464 Ok(PyBytes::new(py, &get_bytes_from_path(old_path)).unbind())
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	465 })
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	466 }
503752cc7177 rust-pyo3-revlog: split_inline Rapha?l Gom?s <rgomes@octobus.net> parents: 52819 diff changeset	467
52821 d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	468 fn get_segment_for_revs(
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	469 slf: &Bound<'_, Self>,
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	470 py: Python<'_>,
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	471 startrev: PyRevision,
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	472 endrev: PyRevision,
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	473 ) -> PyResult<Py<PyTuple>> {
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	474 Self::with_core_read(slf, \|_self_ref, irl\| {
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	475 // Here both revisions only come from revlog code, so we assume
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	476 // them to be valid.
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	477 // Panics will alert the offending programmer if not.
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	478 let (offset, data) = irl
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	479 .get_segment_for_revs(Revision(startrev.0), Revision(endrev.0))
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	480 .map_err(revlog_error_from_msg)?;
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	481 let data = PyBytes::new(py, &data);
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	482 Ok(PyTuple::new(
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	483 py,
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	484 &[offset.into_py_any(py)?, data.into_py_any(py)?],
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	485 )?
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	486 .unbind())
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	487 })
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	488 }
d90a78ca0bdd rust-pyo3-revlog: get_segment_for_revs Rapha?l Gom?s <rgomes@octobus.net> parents: 52820 diff changeset	489
52823 9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	490 fn raw_text(
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	491 slf: &Bound<'_, Self>,
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	492 py: Python<'_>,
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	493 _node: PyObject,
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	494 rev: PyRevision,
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	495 ) -> PyResult<Py<PyBytes>> {
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	496 Self::with_core_read(slf, \|_self_ref, irl\| {
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	497 let mut py_bytes = PyBytes::new(py, &[]).unbind();
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	498 irl.raw_text(Revision(rev.0), \|size, f\| {
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	499 py_bytes = with_pybytes_buffer(py, size, f)?;
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	500 Ok(())
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	501 })
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	502 .map_err(revlog_error_from_msg)?;
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	503 Ok(py_bytes)
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	504 })
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	505 }
9435a212a773 rust-pyo3-revlog: raw_text Rapha?l Gom?s <rgomes@octobus.net> parents: 52821 diff changeset	506
52828 e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	507 #[allow(clippy::too_many_arguments)]
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	508 #[pyo3(signature = (
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	509 transaction,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	510 entry,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	511 data,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	512 _link,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	513 offset,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	514 _sidedata,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	515 _sidedata_offset,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	516 index_end,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	517 data_end,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	518 _sidedata_end
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	519 ))]
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	520 fn write_entry(
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	521 slf: &Bound<'_, Self>,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	522 py: Python<'_>,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	523 transaction: PyObject,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	524 entry: &Bound<'_, PyBytes>,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	525 data: &Bound<'_, PyTuple>,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	526 // TODO remove and also from Python
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	527 _link: PyObject,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	528 offset: usize,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	529 // Other underscore args are for revlog-v2, which is unimplemented
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	530 _sidedata: PyObject,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	531 _sidedata_offset: u64,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	532 index_end: Option<u64>,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	533 data_end: Option<u64>,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	534 _sidedata_end: Option<u64>,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	535 ) -> PyResult<Py<PyTuple>> {
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	536 Self::with_core_write(slf, \|_self_ref, mut irl\| {
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	537 let transaction = PyTransaction::new(transaction);
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	538 let header = data.get_borrowed_item(0)?;
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	539 let header = header.downcast::<PyBytes>()?;
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	540 let data = data.get_borrowed_item(1)?;
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	541 let data = data.downcast::<PyBytes>()?;
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	542 let (idx_pos, data_pos) = irl
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	543 .write_entry(
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	544 transaction,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	545 entry.as_bytes(),
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	546 (header.as_bytes(), data.as_bytes()),
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	547 offset,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	548 index_end,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	549 data_end,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	550 )
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	551 .map_err(revlog_error_from_msg)?;
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	552 let tuple = PyTuple::new(
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	553 py,
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	554 [idx_pos.into_py_any(py)?, data_pos.into_py_any(py)?],
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	555 )?;
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	556 Ok(tuple.unbind())
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	557 })
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	558 }
e49794d16657 rust-pyo3-revlog: write_entry Rapha?l Gom?s <rgomes@octobus.net> parents: 52826 diff changeset	559
52825 1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	560 fn delay(
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	561 slf: &Bound<'_, Self>,
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	562 py: Python<'_>,
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	563 ) -> PyResult<Option<Py<PyBytes>>> {
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	564 Self::with_core_write(slf, \|_self_ref, mut irl\| {
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	565 let path = irl.delay().map_err(revlog_error_from_msg)?;
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	566 Ok(path
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	567 .map(\|p\| PyBytes::new(py, &get_bytes_from_path(p)).unbind()))
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	568 })
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	569 }
1cd75e5d6347 rust-pyo3-revlog: delay Rapha?l Gom?s <rgomes@octobus.net> parents: 52824 diff changeset	570
52826 9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	571 fn finalize_pending(
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	572 slf: &Bound<'_, Self>,
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	573 py: Python<'_>,
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	574 ) -> PyResult<Py<PyBytes>> {
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	575 Self::with_core_write(slf, \|_self_ref, mut irl\| {
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	576 let path =
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	577 irl.finalize_pending().map_err(revlog_error_from_msg)?;
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	578 Ok(PyBytes::new(py, &get_bytes_from_path(path)).unbind())
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	579 })
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	580 }
9dcba9b379cb rust-pyo3-revlog: finalize_pending Rapha?l Gom?s <rgomes@octobus.net> parents: 52825 diff changeset	581
52824 14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	582 fn _chunk(
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	583 slf: &Bound<'_, Self>,
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	584 py: Python<'_>,
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	585 rev: PyRevision,
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	586 ) -> PyResult<Py<PyBytes>> {
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	587 Self::with_core_read(slf, \|_self_ref, irl\| {
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	588 let chunk = irl
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	589 .chunk_for_rev(Revision(rev.0))
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	590 .map_err(revlog_error_from_msg)?;
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	591 Ok(PyBytes::new(py, &chunk).unbind())
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	592 })
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	593 }
14824d621c2c rust-pyo3-revlog: _chunk Rapha?l Gom?s <rgomes@octobus.net> parents: 52823 diff changeset	594
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	595 fn reading(slf: &Bound<'_, Self>) -> PyResult<ReadingContextManager> {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	596 Ok(ReadingContextManager {
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	597 inner_revlog: slf.clone().unbind(),
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	598 })
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	599 }
6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	600
52829 c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	601 #[pyo3(signature = (transaction, data_end=None, sidedata_end=None))]
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	602 fn writing(
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	603 slf: &Bound<'_, Self>,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	604 transaction: PyObject,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	605 data_end: Option<usize>,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	606 sidedata_end: Option<usize>,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	607 ) -> PyResult<WritingContextManager> {
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	608 // Only useful in revlog v2
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	609 let _ = sidedata_end;
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	610 Ok(WritingContextManager {
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	611 inner_revlog: slf.clone().unbind(),
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	612 transaction: RwLock::new(PyTransaction::new(transaction)),
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	613 data_end,
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	614 })
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	615 }
c82f64877055 rust-pyo3-revlog: writing Rapha?l Gom?s <rgomes@octobus.net> parents: 52828 diff changeset	616
52786 4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	617 //
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	618 // -- forwarded index methods --
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	619 //
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	620
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	621 fn _index_get_rev(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	622 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	623 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	624 ) -> PyResult<Option<PyRevision>> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	625 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	626
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	627 // 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	628 // 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	629 // 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	630 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	631 let idx = &irl.index;
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	632
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	633 let prev_queries =
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	634 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	635 // 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	636 // 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	637 // 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	638 // (automation file that changes every
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	639 // 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	640 // all measured purposes so far.
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	641 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	642 return Ok(idx
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	643 .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	644 .ok()
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	645 .map(Into::into));
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	646 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	647
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	648 let opt =
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	649 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	650 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	651
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	652 let rust_rev =
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	653 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	654 Ok(rust_rev.map(Into::into))
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	655 })
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	656 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	657
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	658 /// 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	659 /// node is not found.
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	660 ///
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	661 /// 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	662 /// will catch and rewrap with it
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	663 fn _index_rev(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	664 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	665 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	666 ) -> PyResult<PyRevision> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	667 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	668 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	669
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	670 /// 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	671 fn _index_has_node(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	672 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	673 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	674 ) -> PyResult<bool> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	675 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	676 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	677
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	678 /// 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	679 fn _index_shortest(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	680 slf: &Bound<'_, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	681 node: &Bound<'_, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	682 ) -> PyResult<usize> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	683 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	684 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	685 Ok(Some(l)) => Ok(l),
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	686 Ok(None) => Err(revlog_error_bare()),
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	687 Err(e) => Err(nodemap_error(e)),
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	688 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	689 })
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	690 }
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	691
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	692 fn _index_partialmatch<'py>(
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	693 slf: &Bound<'py, Self>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	694 node: &Bound<'py, PyBytes>,
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	695 ) -> PyResult<Option<Bound<'py, PyBytes>>> {
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	696 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	697 Ok(nt
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	698 .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	699 .map_err(nodemap_error)?
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	700 .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	701 })
4e34e8fd46d4 rust-pyo3-revlog: nodemap based index methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52784 diff changeset	702 }
52787 e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	703
52789 34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	704 /// append an index entry
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	705 fn _index_append(
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	706 slf: &Bound<'_, Self>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	707 tup: &Bound<'_, PyTuple>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	708 ) -> PyResult<()> {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	709 // 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	710 // proper errors
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	711 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	712 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	713
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	714 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	715 let rev = idx.len() as BaseRevision;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	716 // 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	717 // index
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	718 let rev = Revision(rev);
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	719 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	720 .map_err(revlog_error_from_msg)?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	721
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	722 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	723 Ok(())
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	724 })
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	725 }
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	726
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	727 /// 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	728 ///
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	729 /// 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	730 /// 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	731 /// 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	732 fn _index___delitem__(
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	733 slf: &Bound<'_, Self>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	734 arg: &Bound<'_, PyAny>,
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	735 ) -> PyResult<()> {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	736 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	737 UncheckedRevision(rev)
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	738 } else {
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	739 // 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	740 // 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	741 // `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	742 // 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	743 // 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	744 // with the C index).
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	745 let start = arg.getattr("start")?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	746 UncheckedRevision(start.extract()?)
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	747 };
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	748
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	749 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	750 // 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	751 // `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	752 // `PySlice_GetIndicesEx`
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	753 // (Python integration tests will tell us)
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	754 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	755 nodemap_error(NodeMapError::RevisionNotInIndex(start))
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	756 })?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	757 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	758 nt.invalidate_all();
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	759 Self::fill_nodemap(idx, nt)?;
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	760 Ok(())
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	761 })
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	762 }
34f44aa5e844 rust-pyo3-index: first mutating methods Georges Racinet <georges.racinet@cloudcrane.io> parents: 52788 diff changeset	763
52800 ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	764 /// return the gca set of the given revs
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	765 #[pyo3(signature = (*revs))]
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	766 fn _index_ancestors(
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	767 slf: &Bound<'_, Self>,
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	768 revs: &Bound<'_, PyTuple>,
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	769 ) -> PyResult<PyObject> {
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	770 Self::with_index_read(slf, \|idx\| {
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	771 let revs: Vec<_> = rev_pyiter_collect(revs, idx)?;
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	772 Ok(PyList::new(
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	773 slf.py(),
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	774 idx.ancestors(&revs)
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	775 .map_err(graph_error)?
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	776 .into_iter()
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	777 .map(PyRevision::from),
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	778 )?
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	779 .into_any()
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	780 .unbind())
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	781 })
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	782 }
ebcbd2b7a3b6 rust-pyo3-revlog: _index_ancestors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52799 diff changeset	783
52801 419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	784 /// return the heads of the common ancestors of the given revs
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	785 #[pyo3(signature = (*revs))]
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	786 fn _index_commonancestorsheads(
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	787 slf: &Bound<'_, Self>,
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	788 revs: &Bound<'_, PyTuple>,
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	789 ) -> PyResult<Py<PyList>> {
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	790 Self::with_index_read(slf, \|idx\| {
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	791 let revs: Vec<_> = rev_pyiter_collect(revs, idx)?;
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	792 revs_py_list(
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	793 slf.py(),
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	794 idx.common_ancestor_heads(&revs).map_err(graph_error)?,
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	795 )
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	796 })
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	797 }
419e60ff3c44 rust-pyo3-revlog: _index_commonancestorsheads Georges Racinet <georges.racinet@cloudcrane.io> parents: 52800 diff changeset	798
52802 8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	799 /// Clear the index caches and inner py_class data.
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	800 /// It is Python's responsibility to call `update_nodemap_data` again.
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	801 fn _index_clearcaches(slf: &Bound<'_, Self>) -> PyResult<()> {
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	802 Self::with_index_write(slf, \|idx\| {
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	803 idx.clear_caches();
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	804 Ok(())
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	805 })?;
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	806
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	807 let mut self_ref = slf.borrow_mut();
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	808 self_ref.nt.write().map_err(map_lock_error)?.take();
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	809 self_ref.docket.take();
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	810 self_ref.nodemap_mmap.take();
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	811 self_ref.head_revs_py_list.take();
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	812 self_ref.head_node_ids_py_list.take();
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	813 Ok(())
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	814 }
8cb1a561c5ac rust-pyo3-revlog: _index_clearcaches Georges Racinet <georges.racinet@cloudcrane.io> parents: 52801 diff changeset	815
52804 6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	816 /// return the raw binary string representing a revision
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	817 fn _index_entry_binary(
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	818 slf: &Bound<'_, Self>,
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	819 rev: PyRevision,
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	820 ) -> PyResult<Py<PyBytes>> {
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	821 let rev: UncheckedRevision = rev.into();
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	822 Self::with_index_read(slf, \|idx\| {
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	823 idx.check_revision(rev)
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	824 .and_then(\|r\| idx.entry_binary(r))
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	825 .map(\|rust_bytes\| PyBytes::new(slf.py(), rust_bytes).unbind())
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	826 .ok_or_else(\|\| rev_not_in_index(rev))
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	827 })
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	828 }
6c6cfb89a4f1 rust-pyo3-revlog: _index_entry_binary Georges Racinet <georges.racinet@cloudcrane.io> parents: 52803 diff changeset	829
52803 3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	830 /// return a binary packed version of the header
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	831 fn _index_pack_header(
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	832 slf: &Bound<'_, Self>,
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	833 header: i32,
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	834 ) -> PyResult<Py<PyBytes>> {
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	835 let packed =
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	836 Self::with_index_read(slf, \|idx\| Ok(idx.pack_header(header)))?;
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	837 Ok(PyBytes::new(slf.py(), &packed).unbind())
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	838 }
3bfd80a277be rust-pyo3-revlog: _index_pack_header Georges Racinet <georges.racinet@cloudcrane.io> parents: 52802 diff changeset	839
52808 6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	840 /// compute phases
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	841 fn _index_computephasesmapsets(
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	842 slf: &Bound<'_, Self>,
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	843 py: Python<'_>,
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	844 roots: &Bound<'_, PyDict>,
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	845 ) -> PyResult<Py<PyTuple>> {
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	846 let (len, phase_maps) = Self::with_index_read(slf, \|idx\| {
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	847 let extracted_roots: PyResult<HashMap<Phase, Vec<Revision>>> =
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	848 roots
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	849 .iter()
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	850 .map(\|(phase, revs)\| {
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	851 let phase = Phase::try_from(phase.extract::<usize>()?)
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	852 .map_err(\|_\| revlog_error_bare())?;
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	853 let revs: Vec<Revision> =
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	854 rev_pyiter_collect(&revs, idx)?;
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	855 Ok((phase, revs))
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	856 })
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	857 .collect();
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	858 idx.compute_phases_map_sets(extracted_roots?)
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	859 .map_err(graph_error)
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	860 })?;
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	861 // Ugly hack, but temporary (!)
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	862 const IDX_TO_PHASE_NUM: [usize; 4] = [1, 2, 32, 96];
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	863 let py_phase_maps = PyDict::new(py);
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	864 for (i, roots) in phase_maps.into_iter().enumerate() {
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	865 py_phase_maps.set_item(
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	866 IDX_TO_PHASE_NUM[i],
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	867 revs_py_set(py, roots)?.into_any(),
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	868 )?;
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	869 }
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	870 Ok((len, py_phase_maps).into_pyobject(py)?.unbind())
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	871 }
6222326bd13e rust-pyo3-revlog: _index_computephasesmapsets Georges Racinet <georges.racinet@cloudcrane.io> parents: 52806 diff changeset	872
52796 670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	873 /// reachableroots
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	874 #[pyo3(signature = (*args))]
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	875 fn _index_reachableroots2(
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	876 slf: &Bound<'_, Self>,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	877 py: Python<'_>,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	878 args: &Bound<'_, PyTuple>,
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	879 ) -> PyResult<Py<PyList>> {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	880 // 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	881 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	882 let heads = args.get_item(1)?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	883 let roots = args.get_item(2)?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	884 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	885
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	886 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	887 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	888 PyIndexError::new_err("head out of range")
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	889 })?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	890 let roots: Result<_, _> = roots
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	891 .try_iter()?
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	892 .map(\|r\| {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	893 r.and_then(\|o\| match o.extract::<PyRevision>() {
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	894 Ok(r) => Ok(UncheckedRevision(r.0)),
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	895 Err(e) => Err(e),
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	896 })
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	897 })
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	898 .collect();
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	899 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	900 .map_err(graph_error)
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	901 })?;
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	902
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	903 revs_py_list(py, as_set)
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	904 }
670ebb2f975a rust-pyo3-revlog: reachable roots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52795 diff changeset	905
52795 adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	906 #[pyo3(signature = (*args))]
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	907 fn _index_headrevs(
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	908 slf: &Bound<'_, Self>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	909 py: Python<'_>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	910 args: &Bound<'_, PyTuple>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	911 ) -> PyResult<Py<PyList>> {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	912 let (filtered_revs, stop_rev) = match args.len() {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	913 0 => Ok((None, None)),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	914 1 => Ok((Some(args.get_item(0)?), None)),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	915 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	916 _ => Err(PyTypeError::new_err("too many arguments")),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	917 }?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	918 let stop_rev = stop_rev
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	919 .map(\|o\| o.extract::<Option<i32>>())
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	920 .transpose()?
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	921 .flatten();
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	922 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	923
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	924 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	925 let stop_rev = stop_rev
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	926 // should this not just be the normal checking?
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	927 .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	928 .map(Revision);
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	929
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	930 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	931 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	932 idx.head_revs_advanced(
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	933 &filtered_revs,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	934 stop_rev,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	935 stop_rev.is_none(),
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	936 )
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	937 } else if stop_rev.is_some() {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	938 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	939 } else {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	940 idx.head_revs_shortcut()
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	941 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	942 .map_err(graph_error)?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	943 Ok((from_core, stop_rev))
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	944 })?;
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	945
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	946 if stop_rev.is_some() {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	947 // we don't cache result for now
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	948 let new_heads =
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	949 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	950
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	951 revs_py_list(py, new_heads)
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	952 } else {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	953 if let Some(new_heads) = from_core {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	954 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	955 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	956
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	957 Ok(slf
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	958 .borrow()
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	959 .head_revs_py_list
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	960 .as_ref()
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	961 .expect("head revs should be cached")
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	962 .clone_ref(py))
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	963 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	964 }
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	965
52810 09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	966 /// get head nodeids
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	967 fn _index_head_node_ids(
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	968 slf: &Bound<'_, Self>,
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	969 py: Python<'_>,
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	970 ) -> PyResult<Py<PyList>> {
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	971 let (head_revs, head_nodes) = Self::with_index_read(slf, \|idx\| {
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	972 // We don't use the shortcut here, as it's actually slower to loop
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	973 // through the cached `PyList` than to re-do the whole
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	974 // conversion for large lists, which are the performance
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	975 // sensitive ones anyway.
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	976 let head_revs = idx.head_revs().map_err(graph_error)?;
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	977 let head_nodes = PyList::new(
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	978 py,
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	979 head_revs.iter().map(\|r\| {
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	980 PyBytes::new(
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	981 py,
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	982 idx.node(*r)
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	983 .expect("rev should have been in the index")
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	984 .as_bytes(),
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	985 )
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	986 .unbind()
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	987 }),
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	988 )?
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	989 .unbind();
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	990 Ok((head_revs, head_nodes))
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	991 })?;
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	992
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	993 Self::cache_new_heads_py_list(slf, head_revs)?;
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	994 // TODO discussion with Alphare: in hg-cpython,
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	995 // `cache_new_heads_node_ids_py_list` reconverts `head_nodes`,
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	996 // to store it in the cache attr that is not actually used.
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	997 // Should we drop the idea of this cache definition or actually
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	998 // use it? Perhaps in a later move for perf assessment?
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	999 Ok(head_nodes)
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	1000 }
09544af536ef rust-pyo3-revlog: _index_node_ids Georges Racinet <georges.racinet@cloudcrane.io> parents: 52809 diff changeset	1001
52809 6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1002 /// get diff in head revisions
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1003 fn _index_headrevsdiff(
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1004 slf: &Bound<'_, Self>,
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1005 py: Python<'_>,
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1006 begin: PyRevision,
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1007 end: PyRevision,
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1008 ) -> PyResult<Py<PyTuple>> {
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1009 let begin: BaseRevision = begin.0 - 1;
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1010 let end: BaseRevision = end.0 - 1;
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1011 let (removed, added) = Self::with_index_read(slf, \|idx\| {
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1012 idx.head_revs_diff(
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1013 check_revision(idx, begin)?,
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1014 check_revision(idx, end)?,
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1015 )
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1016 .map_err(graph_error)
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1017 })?;
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1018 let py_removed = revs_py_list(py, removed)?;
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1019 let py_added = revs_py_list(py, added)?;
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1020 Ok((py_removed, py_added).into_pyobject(py)?.unbind())
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1021 }
6bd11e3e05d6 rust-pyo3-revlog: _index_headrevsdiff Georges Racinet <georges.racinet@cloudcrane.io> parents: 52808 diff changeset	1022
52797 f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1023 /// True if the object is a snapshot
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1024 fn _index_issnapshot(
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1025 slf: &Bound<'_, Self>,
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1026 rev: PyRevision,
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1027 ) -> PyResult<bool> {
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1028 let rev: UncheckedRevision = rev.into();
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1029 let rev = Self::with_index_read(slf, \|idx\| {
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1030 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	1031 })?;
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1032 Self::with_core_read(slf, \|_self_ref, irl\| {
52797 f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1033 irl.is_snapshot(rev)
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1034 .map_err(\|e\| PyValueError::new_err(e.to_string()))
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1035 })
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1036 }
f76911bfbfb1 rust-pyo3-revlog: issnapshot Georges Racinet <georges.racinet@cloudcrane.io> parents: 52796 diff changeset	1037
52799 798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1038 /// Gather snapshot data in a cache dict
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1039 fn _index_findsnapshots(
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1040 slf: &Bound<'_, Self>,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1041 cache: &Bound<'_, PyDict>,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1042 start_rev: PyRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1043 end_rev: PyRevision,
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1044 ) -> PyResult<()> {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1045 let mut cache = PySnapshotsCache(cache);
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1046 Self::with_index_read(slf, \|idx\| {
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1047 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	1048 .map_err(\|_\| revlog_error_bare())
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1049 })?;
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1050 Ok(())
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1051 }
798355e46d03 rust-pyo3-revlog: _index_findsnapshots Georges Racinet <georges.racinet@cloudcrane.io> parents: 52798 diff changeset	1052
52805 a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1053 /// determine revisions with deltas to reconstruct fulltext
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1054 #[pyo3(signature = (rev, stop_rev, using_general_delta))]
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1055 fn _index_deltachain(
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1056 slf: &Bound<'_, Self>,
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1057 py: Python<'_>,
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1058 rev: PyRevision,
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1059 stop_rev: Option<PyRevision>,
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1060 using_general_delta: Option<u32>,
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1061 ) -> PyResult<Py<PyTuple>> {
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1062 let using_general_delta = using_general_delta.map(\|i\| i != 0);
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1063 let rev: UncheckedRevision = rev.into();
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1064 let stop_rev: Option<UncheckedRevision> = stop_rev.map(Into::into);
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1065
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1066 let (chain, stopped) = Self::with_index_read(slf, \|idx\| {
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1067 let rev = idx.check_revision(rev).ok_or_else(\|\| {
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1068 nodemap_error(NodeMapError::RevisionNotInIndex(rev))
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1069 })?;
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1070 let stop_rev = stop_rev
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1071 .map(\|r\| {
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1072 idx.check_revision(r).ok_or_else(\|\| {
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1073 nodemap_error(NodeMapError::RevisionNotInIndex(
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1074 rev.into(),
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1075 ))
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1076 })
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1077 })
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1078 .transpose()?;
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1079 idx.delta_chain(rev, stop_rev, using_general_delta)
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1080 .map_err(\|e\| PyValueError::new_err(e.to_string()))
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1081 })?;
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1082
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1083 let py_chain = revs_py_list(py, chain)?.into_any();
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1084 let py_stopped =
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1085 PyBool::new(py, stopped).to_owned().unbind().into_any();
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1086 Ok((py_chain, py_stopped).into_pyobject(py)?.unbind())
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1087 }
a8debfd85d55 rust-pyo3-revlog: _index_deltachain Georges Racinet <georges.racinet@cloudcrane.io> parents: 52804 diff changeset	1088
52806 0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1089 /// slice planned chunk read to reach a density threshold
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1090 fn _index_slicechunktodensity(
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1091 slf: &Bound<'_, Self>,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1092 py: Python<'_>,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1093 revs: &Bound<'_, PyAny>,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1094 target_density: f64,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1095 min_gap_size: usize,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1096 ) -> PyResult<PyObject> {
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1097 let as_nested_vec =
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1098 Self::with_index_read(slf, \|idx\| {
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1099 let revs: Vec<_> = rev_pyiter_collect(revs, idx)?;
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1100 Ok(idx.slice_chunk_to_density(
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1101 &revs,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1102 target_density,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1103 min_gap_size,
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1104 ))
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1105 })?;
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1106 let res_len = as_nested_vec.len();
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1107
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1108 // cannot build the outer sequence from iterator, because
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1109 // `rev_py_list()` returns `Result<T>` instead of `T`.
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1110 let mut res = Vec::with_capacity(res_len);
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1111 for chunk in as_nested_vec {
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1112 res.push(revs_py_list(py, chunk)?.into_any());
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1113 }
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1114 // This is just to do the same as C, not sure why it does this
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1115 Ok(if res_len == 1 {
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1116 PyTuple::new(py, res)?.unbind().into_any()
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1117 } else {
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1118 PyList::new(py, res)?.unbind().into_any()
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1119 })
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1120 }
0a0ed46ef6d6 rust-pyo3-revlog: _index_slicechunktodensity Georges Racinet <georges.racinet@cloudcrane.io> parents: 52805 diff changeset	1121
52787 e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	1122 fn _index___len__(slf: &Bound<'_, Self>) -> PyResult<usize> {
e5f89bd1a5ee rust-pyo3-revlog: _index___len__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52786 diff changeset	1123 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	1124 }
52791 0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1125
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1126 fn _index___getitem__(
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1127 slf: &Bound<'_, Self>,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1128 py: Python<'_>,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1129 key: &Bound<'_, PyAny>,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1130 ) -> PyResult<PyObject> {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1131 Self::with_index_read(slf, \|idx\| {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1132 match key.extract::<BaseRevision>() {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1133 Ok(key_as_int) => {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1134 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	1135 RevisionDataParams::default()
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1136 } else {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1137 let rev = UncheckedRevision(key_as_int);
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1138 match idx.entry_as_params(rev) {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1139 Some(e) => e,
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1140 None => {
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1141 return Err(PyIndexError::new_err(
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1142 "revlog index out of range",
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1143 ));
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1144 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1145 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1146 };
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1147 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	1148 .into_any()
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1149 .unbind())
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1150 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1151 // 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	1152 _ => Self::_index_get_rev(slf, key.downcast::<PyBytes>()?)?
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1153 .map_or_else(
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1154 \|\| Ok(py.None()),
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1155 \|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	1156 ),
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1157 }
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1158 })
0ac956db7ea7 rust-pyo3-index: __getitem__ Georges Racinet <georges.racinet@cloudcrane.io> parents: 52789 diff changeset	1159 }
52798 88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1160
52811 e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1161 /// Returns the full nodemap bytes to be written as-is to disk
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1162 fn _index_nodemap_data_all(
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1163 slf: &Bound<'_, Self>,
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1164 py: Python<'_>,
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1165 ) -> PyResult<Py<PyBytes>> {
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1166 Self::with_index_nt_write(slf, \|idx, nt\| {
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1167 let old_nt = std::mem::take(nt);
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1168 let (readonly, bytes) = old_nt.into_readonly_and_added_bytes();
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1169
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1170 // If there's anything readonly, we need to build the data again
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1171 // from scratch
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1172 let bytes = if readonly.len() > 0 {
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1173 let mut nt =
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1174 CoreNodeTree::load_bytes(Box::<Vec<_>>::default(), 0);
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1175 Self::fill_nodemap(idx, &mut nt)?;
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1176
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1177 let (readonly, bytes) = nt.into_readonly_and_added_bytes();
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1178 assert_eq!(readonly.len(), 0);
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1179
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1180 bytes
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1181 } else {
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1182 bytes
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1183 };
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1184
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1185 let bytes = PyBytes::new(py, &bytes);
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1186 Ok(bytes.unbind())
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1187 })
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1188 }
e7ad174ae58f rust-pyo3-revlog: _index_nodemap_data_all Rapha?l Gom?s <rgomes@octobus.net> parents: 52810 diff changeset	1189
52812 6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1190 /// Returns the last saved docket along with the size of any changed data
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1191 /// (in number of blocks), and said data as bytes.
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1192 fn _index_nodemap_data_incremental(
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1193 slf: &Bound<'_, Self>,
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1194 py: Python<'_>,
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1195 ) -> PyResult<PyObject> {
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1196 let mut self_ref = slf.borrow_mut();
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1197 let docket = &mut self_ref.docket;
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1198 let docket = match docket.as_ref() {
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1199 Some(d) => d.clone_ref(py),
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1200 None => return Ok(py.None()),
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1201 };
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1202 drop(self_ref);
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1203
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1204 Self::with_core_write(slf, \|self_ref, irl\| {
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1205 let mut nt = self_ref
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1206 .get_nodetree(&irl.index)?
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1207 .write()
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1208 .map_err(map_lock_error)?;
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1209 let nt = nt.take().expect("nodetree should be set");
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1210 let masked_blocks = nt.masked_readonly_blocks();
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1211 let (_, data) = nt.into_readonly_and_added_bytes();
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1212 let changed = masked_blocks * std::mem::size_of::<Block>();
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1213
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1214 Ok(PyTuple::new(
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1215 py,
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1216 [
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1217 docket,
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1218 changed.into_py_any(py)?,
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1219 PyBytes::new(py, &data).into_py_any(py)?,
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1220 ],
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1221 )?
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1222 .unbind()
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1223 .into_any())
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1224 })
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1225 }
6fc1637f0986 rust-pyo3-revlog: _index_nodemap_data_incremental Rapha?l Gom?s <rgomes@octobus.net> parents: 52811 diff changeset	1226
52813 65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1227 /// Update the nodemap from the new (mmaped) data.
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1228 /// The docket is kept as a reference for later incremental calls.
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1229 fn _index_update_nodemap_data(
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1230 slf: &Bound<'_, Self>,
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1231 py: Python<'_>,
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1232 docket: &Bound<'_, PyAny>,
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1233 nm_data: &Bound<'_, PyAny>,
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1234 ) -> PyResult<PyObject> {
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1235 // Safety: we keep the buffer around inside the class as `nodemap_mmap`
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1236 let (buf, bytes) = unsafe { take_buffer_with_slice(nm_data)? };
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1237 let len = buf.item_count();
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1238 let data_tip =
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1239 docket.getattr("tip_rev")?.extract::<BaseRevision>()?.into();
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1240
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1241 let mut nt = CoreNodeTree::load_bytes(bytes, len);
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1242
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1243 Self::with_index_read(slf, \|idx\| {
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1244 let data_tip = idx.check_revision(data_tip).ok_or_else(\|\| {
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1245 nodemap_error(NodeMapError::RevisionNotInIndex(data_tip))
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1246 })?;
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1247 let current_tip = idx.len();
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1248
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1249 for r in (data_tip.0 + 1)..current_tip as BaseRevision {
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1250 let rev = Revision(r);
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1251 // in this case node() won't ever return None
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1252 nt.insert(idx, idx.node(rev).expect("node should exist"), rev)
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1253 .map_err(nodemap_error)?;
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1254 }
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1255
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1256 Ok(py.None())
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1257 })?;
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1258
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1259 let mut self_ref = slf.borrow_mut();
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1260 self_ref.docket.replace(docket.clone().unbind());
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1261 self_ref.nodemap_mmap = Some(buf);
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1262 self_ref.nt.write().map_err(map_lock_error)?.replace(nt);
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1263
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1264 Ok(py.None())
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1265 }
65df754d598b rust-pyo3-revlog: _index_update_nodemap_data Rapha?l Gom?s <rgomes@octobus.net> parents: 52812 diff changeset	1266
52798 88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1267 #[getter]
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1268 fn _index_entry_size(&self) -> usize {
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1269 INDEX_ENTRY_SIZE
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1270 }
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1271
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1272 #[getter]
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1273 fn _index_rust_ext_compat(&self) -> i32 {
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1274 1
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1275 }
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1276
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1277 #[getter]
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1278 fn _index_is_rust(&self) -> bool {
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1279 true
88d62995a65b rust-pyo3-revlog: constant properties Georges Racinet <georges.racinet@cloudcrane.io> parents: 52797 diff changeset	1280 }
52780 42b219a1404a rust-pyo3-revlog: InnerRevlog definition and constructor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52778 diff changeset	1281 }
52778 523ca3d225f5 rust-pyo3-revlog: config extraction functions Georges Racinet <georges.racinet@cloudcrane.io> parents: 52775 diff changeset	1282
52782 827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1283 impl InnerRevlog {
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1284 /// 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	1285 ///
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1286 /// 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	1287 /// 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	1288 /// 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	1289 /// 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	1290 /// 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	1291 /// later on.
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1292 ///
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1293 /// 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	1294 /// 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	1295 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	1296 slf: &Bound<'py, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1297 f: impl FnOnce(
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1298 &PyRef<'py, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1299 RwLockReadGuard<CoreInnerRevlog>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1300 ) -> PyResult<T>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1301 ) -> PyResult<T> {
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1302 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	1303 // 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	1304 // 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	1305 // 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	1306 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	1307 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	1308 f(&self_ref, guard)
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1309 }
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1310
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1311 /// 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	1312 ///
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1313 /// 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	1314 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	1315 slf: &Bound<'py, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1316 f: impl FnOnce(
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1317 &PyRef<'py, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1318 RwLockWriteGuard<CoreInnerRevlog>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1319 ) -> PyResult<T>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1320 ) -> PyResult<T> {
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1321 let self_ref = slf.borrow();
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1322 // 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	1323 // 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	1324 // 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	1325 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	1326 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	1327 f(&self_ref, guard)
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1328 }
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1329
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1330 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	1331 slf: &Bound<'_, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1332 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	1333 ) -> PyResult<T> {
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1334 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	1335 }
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1336
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1337 fn with_index_write<T>(
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1338 slf: &Bound<'_, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1339 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	1340 ) -> PyResult<T> {
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1341 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	1342 }
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1343
52784 5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1344 /// 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	1345 /// [`NodeTree`]
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1346 ///
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1347 /// 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	1348 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	1349 slf: &Bound<'_, Self>,
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1350 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	1351 ) -> PyResult<T> {
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1352 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	1353 let idx = &guard.index;
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1354 let nt =
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1355 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	1356 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	1357 f(idx, nt)
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1358 })
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1359 }
5e3e8876fd9e rust-pyo3-revlog: index and nodetree read accessor helpers Georges Racinet <georges.racinet@cloudcrane.io> parents: 52782 diff changeset	1360
52788 e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1361 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	1362 slf: &Bound<'_, Self>,
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1363 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	1364 ) -> PyResult<T> {
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1365 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	1366 let idx = &mut guard.index;
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1367 let mut nt = self_ref
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1368 .get_nodetree(idx)?
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1369 .write()
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1370 .map_err(map_lock_error)?;
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1371 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	1372 f(idx, nt)
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1373 })
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1374 }
e29e75e8328c rust-pyo3-revlog: index and notetree writing accessors Georges Racinet <georges.racinet@cloudcrane.io> parents: 52787 diff changeset	1375
52782 827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1376 /// 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	1377 /// [`Index`]
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1378 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1379 /// # Python exceptions
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1380 /// 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	1381 /// with `idx`.
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1382 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	1383 for r in 0..idx.len() {
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1384 let rev = Revision(r as BaseRevision);
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1385 // 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	1386 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	1387 .map_err(nodemap_error)?
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1388 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1389 Ok(())
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1390 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1391
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1392 /// Return a working NodeTree of this InnerRevlog
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1393 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1394 /// 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	1395 /// 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	1396 /// filled right away from the index.
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1397 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1398 /// 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	1399 /// 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	1400 /// 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	1401 ///
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1402 /// # Python exceptions
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1403 /// 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	1404 /// NodeTree is empty when it is called.
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1405 fn get_nodetree(
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1406 &self,
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1407 idx: &Index,
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1408 ) -> PyResult<&RwLock<Option<CoreNodeTree>>> {
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1409 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	1410 let readonly = Box::<Vec<_>>::default();
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1411 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	1412 Self::fill_nodemap(idx, &mut nt)?;
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1413 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	1414 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1415 Ok(&self.nt)
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1416 }
52795 adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1417
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1418 fn cache_new_heads_py_list(
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1419 slf: &Bound<'_, Self>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1420 new_heads: Vec<Revision>,
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1421 ) -> PyResult<Py<PyList>> {
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1422 let py = slf.py();
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1423 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	1424 slf.borrow_mut().head_revs_py_list =
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1425 Some(new_heads_py_list.clone_ref(py));
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1426 // TODO is returning really useful?
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1427 Ok(new_heads_py_list)
adf91dfe6c04 rust-pyo3-index: _index_headrevs Georges Racinet <georges.racinet@cloudcrane.io> parents: 52794 diff changeset	1428 }
52782 827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1429 }
827889802d11 rust-pyo3-revlog: nodemap filling accessor Georges Racinet <georges.racinet@cloudcrane.io> parents: 52780 diff changeset	1430
52792 acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1431 #[pyclass]
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1432 struct NodeTree {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1433 nt: RwLock<CoreNodeTree>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1434 index: SharedByPyObject<PySharedIndex>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1435 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1436
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1437 #[pymethods]
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1438 impl NodeTree {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1439 #[new]
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1440 // 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	1441 // `py_rust_index_to_graph`
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1442 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	1443 let py_irl = index_proxy.getattr("inner")?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1444 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	1445 let shareable_irl = &py_irl_ref.irl;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1446
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1447 // 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	1448 // internal reference.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1449 let index = unsafe {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1450 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	1451 };
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1452 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	1453
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1454 Ok(Self {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1455 nt: nt.into(),
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1456 index,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1457 })
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1458 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1459
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1460 /// Tell whether the NodeTree is still valid
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1461 ///
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1462 /// 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	1463 /// 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	1464 /// 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	1465 /// (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	1466 /// been meanwhile mutated.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1467 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	1468 // 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	1469 // we only check errors
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1470 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	1471 // 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	1472 // - the index has previously been mutably borrowed
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1473 // - there is currently a mutable borrow
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1474 // 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	1475 // the index to still be valid.
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1476 Ok(result.is_err())
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1477 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1478
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1479 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	1480 // 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	1481 // 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	1482 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	1483
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1484 let rev = check_revision(idx, rev)?;
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1485 if rev == NULL_REVISION {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1486 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	1487 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1488
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1489 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	1490 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	1491 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	1492 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1493
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1494 fn shortest(
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1495 &self,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1496 py: Python<'_>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1497 node: &Bound<'_, PyBytes>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1498 ) -> PyResult<usize> {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1499 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	1500 // 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	1501 // as returned type is Copy
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1502 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	1503 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	1504 .map_err(nodemap_error)?
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1505 .ok_or_else(revlog_error_bare)
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1506 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1507
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1508 /// 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	1509 ///
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1510 /// 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	1511 /// 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	1512 fn prefix_rev_lookup(
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1513 &self,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1514 py: Python<'_>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1515 node_prefix: &Bound<'_, PyBytes>,
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1516 ) -> PyResult<Option<PyRevision>> {
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1517 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	1518 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	1519 // 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	1520 // as returned type is Copy
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1521 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	1522 Ok(nt
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1523 .find_bin(idx, prefix)
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1524 .map_err(nodemap_error)?
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1525 .map(\|r\| r.into()))
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1526 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1527 }
acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1528
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1529 pub fn init_module<'py>(
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1530 py: Python<'py>,
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1531 package: &str,
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1532 ) -> PyResult<Bound<'py, PyModule>> {
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1533 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	1534 m.add_class::<InnerRevlog>()?;
52792 acae91fad6be rust-pyo3-revlog: standalone NodeTree class Georges Racinet <georges.racinet@cloudcrane.io> parents: 52791 diff changeset	1535 m.add_class::<NodeTree>()?;
52793 6a70e4931773 rust-pyo3-revlog: ReadingContextManager Georges Racinet <georges.racinet@cloudcrane.io> parents: 52792 diff changeset	1536 m.add_class::<ReadingContextManager>()?;
52775 264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1537 Ok(m)
264047bf4b9b rust-pyo3-revlog: new Python and Rust module Georges Racinet <georges.racinet@cloudcrane.io> parents: diff changeset	1538 }

Mercurial > public > mercurial-scm > hg

annotate rust/hg-pyo3/src/revlog/mod.rs @ 52832:c72d8df0c080