Sebastian Thielrefactor (#266)b88f2534 years agoHistory
80 lines · 3.3 KB · Rust
RawDownload
use crate::{    pack,    store_impls::{compound, loose},}; /// Returned by [`compound::Store::try_find()`]#[derive(thiserror::Error, Debug)]#[allow(missing_docs)]pub enum Error {    #[error("An error occurred while obtaining an object from the loose object store")]    Loose(#[from] loose::find::Error),    #[error("An error occurred while obtaining an object from the packed object store")]    Pack(#[from] pack::data::decode_entry::Error),} #[derive(PartialEq, Eq, Debug, Hash, Ord, PartialOrd, Clone)]pub(crate) struct PackLocation {    pub bundle_index: usize,    pub entry_index: u32,} impl compound::Store {    /// Find an object as identified by [`ObjectId`][git_hash::ObjectId] and store its data in full in the provided `buffer`.    /// This will search the object in all contained object databases.    /// Use a `pack_cache` to accelerate pack access by reducing the amount of work duplication, or [`pack::cache::Never`] to disable any caching.    pub fn try_find<'a>(        &self,        id: impl AsRef<git_hash::oid>,        buffer: &'a mut Vec<u8>,        pack_cache: &mut impl pack::cache::DecodeEntry,    ) -> Result<Option<git_object::Data<'a>>, Error> {        let id = id.as_ref();        for bundle in &self.bundles {            if let Some(idx) = find_pack_index(bundle, id) {                let (object, _location) = bundle.get_object_by_index(idx, buffer, pack_cache)?;                return Ok(Some(object));            }        }        if self.loose.contains(id) {            return self.loose.try_find(id, buffer).map_err(Into::into);        }        Ok(None)    }     /// Internal-use function to look up a packed object index or loose object.    /// Used to avoid double-lookups in linked::Store::locate.    /// (The polonius borrow-checker would support this via the locate    /// function, so this can be [simplified](https://github.com/Byron/gitoxide/blob/0c5f4043da4615820cb180804a81c2d4fe75fe5e/git-odb/src/compound/locate.rs#L47)    /// once polonius is stable.)    pub(crate) fn internal_find_packed(&self, id: impl AsRef<git_hash::oid>) -> Option<PackLocation> {        let id = id.as_ref();        for (bundle_index, bundle) in self.bundles.iter().enumerate() {            if let Some(idx) = find_pack_index(bundle, id) {                return Some(PackLocation {                    bundle_index,                    entry_index: idx,                });            }        }        None    }     pub(crate) fn internal_get_packed_object_by_index<'a>(        &self,        pack_index: usize,        object_index: u32,        buffer: &'a mut Vec<u8>,        pack_cache: &mut impl pack::cache::DecodeEntry,    ) -> Result<(git_object::Data<'a>, pack::data::entry::Location), pack::data::decode_entry::Error> {        self.bundles[pack_index].get_object_by_index(object_index, buffer, pack_cache)    }} /// Special-use function to look up an object index. Used to avoid double-lookups in/// [compound::Store::find()][crate::store::compound::Store::find()]. (The polonius borrow-checker would support this via the 'find'/// function, so this can be [simplified](https://github.com/Byron/gitoxide/blob/0c5f4043da4615820cb180804a81c2d4fe75fe5e/git-odb/src/compound/locate.rs#L47)/// once polonius is stable.)fn find_pack_index(bundle: &git_pack::Bundle, id: &git_hash::oid) -> Option<u32> {    bundle.index.lookup(id)}