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)}
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)
}