-
-
Notifications
You must be signed in to change notification settings - Fork 33
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor levels & add level & tiered compactor tests
- Loading branch information
1 parent
16bb004
commit 87f9927
Showing
10 changed files
with
200 additions
and
141 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,78 @@ | ||
| use super::HiddenSet; | ||
| use crate::segment::Segment; | ||
| use serde::{Deserialize, Serialize}; | ||
| use std::{collections::HashMap, ops::DerefMut, sync::Arc}; | ||
|
|
||
| #[derive(Serialize, Deserialize)] | ||
| pub struct Level(Vec<String>); | ||
|
|
||
| impl std::ops::Deref for Level { | ||
| type Target = Vec<String>; | ||
|
|
||
| fn deref(&self) -> &Self::Target { | ||
| &self.0 | ||
| } | ||
| } | ||
|
|
||
| impl DerefMut for Level { | ||
| fn deref_mut(&mut self) -> &mut Self::Target { | ||
| &mut self.0 | ||
| } | ||
| } | ||
|
|
||
| impl Default for Level { | ||
| fn default() -> Self { | ||
| Self(Vec::with_capacity(20)) | ||
| } | ||
| } | ||
|
|
||
| pub struct ResolvedLevel(pub(crate) Vec<Arc<Segment>>); | ||
|
|
||
| impl std::ops::Deref for ResolvedLevel { | ||
| type Target = Vec<Arc<Segment>>; | ||
|
|
||
| fn deref(&self) -> &Self::Target { | ||
| &self.0 | ||
| } | ||
| } | ||
|
|
||
| impl DerefMut for ResolvedLevel { | ||
| fn deref_mut(&mut self) -> &mut Self::Target { | ||
| &mut self.0 | ||
| } | ||
| } | ||
|
|
||
| impl ResolvedLevel { | ||
| pub fn new( | ||
| level: &Level, | ||
| hidden_set: &HiddenSet, | ||
| segments: &HashMap<String, Arc<Segment>>, | ||
| ) -> Self { | ||
| let mut new_level = Vec::new(); | ||
|
|
||
| for segment_id in level.iter() { | ||
| if !hidden_set.contains(segment_id) { | ||
| new_level.push( | ||
| segments | ||
| .get(segment_id) | ||
| .cloned() | ||
| .expect("where's the segment at?"), | ||
| ); | ||
| } | ||
| } | ||
|
|
||
| Self(new_level) | ||
| } | ||
|
|
||
| pub fn get_overlapping_segments(&self, start: Vec<u8>, end: Vec<u8>) -> Vec<&String> { | ||
| use std::ops::Bound::Included; | ||
|
|
||
| let bounds = (Included(start), Included(end)); | ||
|
|
||
| self.0 | ||
| .iter() | ||
| .filter(|x| Segment::check_key_range_overlap(x, &bounds)) | ||
| .map(|x| &x.metadata.id) | ||
| .collect() | ||
| } | ||
| } |
Oops, something went wrong.