owning.rs
104 lines
| 2.5 KiB
| application/rls-services+xml
|
RustLexer
|
r49864 | use crate::{DirstateError, DirstateParents}; | ||
|
r48766 | use super::dirstate_map::DirstateMap; | ||
use std::ops::Deref; | ||||
|
r49864 | use ouroboros::self_referencing; | ||
|
r49863 | |||
|
r48766 | /// Keep a `DirstateMap<'on_disk>` next to the `on_disk` buffer that it | ||
/// borrows. | ||||
|
r49864 | #[self_referencing] | ||
|
r48766 | pub struct OwningDirstateMap { | ||
on_disk: Box<dyn Deref<Target = [u8]> + Send>, | ||||
|
r49864 | #[borrows(on_disk)] | ||
#[covariant] | ||||
map: DirstateMap<'this>, | ||||
|
r48766 | } | ||
impl OwningDirstateMap { | ||||
pub fn new_empty<OnDisk>(on_disk: OnDisk) -> Self | ||||
where | ||||
|
r49864 | OnDisk: Deref<Target = [u8]> + Send + 'static, | ||
|
r48766 | { | ||
let on_disk = Box::new(on_disk); | ||||
|
r49864 | OwningDirstateMapBuilder { | ||
on_disk, | ||||
|
r50825 | map_builder: |bytes| DirstateMap::empty(bytes), | ||
|
r49864 | } | ||
.build() | ||||
|
r48766 | } | ||
|
r49864 | pub fn new_v1<OnDisk>( | ||
on_disk: OnDisk, | ||||
|
r51140 | identity: Option<u64>, | ||
|
r49864 | ) -> Result<(Self, DirstateParents), DirstateError> | ||
where | ||||
OnDisk: Deref<Target = [u8]> + Send + 'static, | ||||
{ | ||||
let on_disk = Box::new(on_disk); | ||||
let mut parents = DirstateParents::NULL; | ||||
|
r48766 | |||
|
r49864 | Ok(( | ||
OwningDirstateMapTryBuilder { | ||||
on_disk, | ||||
map_builder: |bytes| { | ||||
r51147 | DirstateMap::new_v1(bytes, identity).map(|(dmap, p)| { | |||
|
r49864 | parents = p.unwrap_or(DirstateParents::NULL); | ||
dmap | ||||
}) | ||||
}, | ||||
} | ||||
.try_build()?, | ||||
parents, | ||||
)) | ||||
|
r48766 | } | ||
|
r49864 | pub fn new_v2<OnDisk>( | ||
on_disk: OnDisk, | ||||
data_size: usize, | ||||
metadata: &[u8], | ||||
|
r51138 | uuid: Vec<u8>, | ||
|
r51140 | identity: Option<u64>, | ||
|
r49864 | ) -> Result<Self, DirstateError> | ||
where | ||||
OnDisk: Deref<Target = [u8]> + Send + 'static, | ||||
{ | ||||
let on_disk = Box::new(on_disk); | ||||
OwningDirstateMapTryBuilder { | ||||
on_disk, | ||||
map_builder: |bytes| { | ||||
r51147 | DirstateMap::new_v2(bytes, data_size, metadata, uuid, identity) | |||
|
r49864 | }, | ||
} | ||||
.try_build() | ||||
|
r48766 | } | ||
|
r49864 | pub fn with_dmap_mut<R>( | ||
&mut self, | ||||
f: impl FnOnce(&mut DirstateMap) -> R, | ||||
) -> R { | ||||
self.with_map_mut(f) | ||||
} | ||||
pub fn get_map(&self) -> &DirstateMap { | ||||
self.borrow_map() | ||||
} | ||||
pub fn on_disk(&self) -> &[u8] { | ||||
self.borrow_on_disk() | ||||
|
r48766 | } | ||
|
r51138 | |||
pub fn old_uuid(&self) -> Option<&[u8]> { | ||||
self.get_map().old_uuid.as_deref() | ||||
} | ||||
|
r51140 | pub fn old_identity(&self) -> Option<u64> { | ||
self.get_map().identity | ||||
} | ||||
|
r51138 | pub fn old_data_size(&self) -> usize { | ||
self.get_map().old_data_size | ||||
} | ||||
|
r48766 | } | ||