facebook/sapling
1
1
Fork 0
mirror of https://github.com/facebook/sapling.git synced 2024-10-10 00:45:18 +03:00
Code Issues Projects Releases Wiki Activity

dag: extract prefix lookup logic from MemIdMap to CoreMemIdMap

Browse Source 
Summary: It will be reused in NameDag.

Reviewed By: sfilipco

Differential Revision: D27547583

fbshipit-source-id: da85fc7504d20877210e8ed1a97cbec18d06eede
This commit is contained in:
Jun Wu 2021-04-13 16:35:05 -07:00 committed by Facebook GitHub Bot
parent 2ab2ba667c
commit d450aa7bd4
1 changed files with 20 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
eden/scm/lib/dag/src/idmap/mem_idmap.rs
View File

@ -69,6 +69,25 @@ impl CoreMemIdMap {
self.id2name.get(&id).cloned()
}
pub fn lookup_vertexes_by_hex_prefix(
&self,
hex_prefix: &[u8],
limit: usize,
) -> Result<Vec<VertexName>> {
let start = VertexName::from_hex(hex_prefix)?;
let mut result = Vec::new();
for (vertex, _) in self.name2id.range(start..) {
if !vertex.to_hex().as_bytes().starts_with(hex_prefix) {
break;
}
result.push(vertex.clone());
if result.len() >= limit {
break;
}
}
Ok(result)
}
pub fn has_vertex_name(&self, name: &VertexName) -> bool {
self.name2id.contains_key(name)
}
@ -188,18 +207,7 @@ impl PrefixLookup for MemIdMap {
hex_prefix: &[u8],
limit: usize,
) -> Result<Vec<VertexName>> {
let start = VertexName::from_hex(hex_prefix)?;
let mut result = Vec::new();
for (vertex, _) in self.core.name2id.range(start..) {
if !vertex.to_hex().as_bytes().starts_with(hex_prefix) {
break;
}
result.push(vertex.clone());
if result.len() >= limit {
break;
}
}
Ok(result)
self.core.lookup_vertexes_by_hex_prefix(hex_prefix, limit)
}
}