index: simplify segment_resolve_prefix() loop to make both impls look close

This commit is contained in:
Yuya Nishihara 2023-01-26 15:19:00 +09:00
parent 5d675f13a2
commit b4c837fd4a

View file

@ -1334,9 +1334,9 @@ impl IndexSegment for ReadonlyIndex {
fn segment_resolve_prefix(&self, prefix: &HexPrefix) -> PrefixResolution<CommitId> {
let (bytes_prefix, min_bytes_prefix) = prefix.bytes_prefixes();
match self.commit_id_byte_prefix_to_lookup_pos(&min_bytes_prefix) {
None => PrefixResolution::NoMatch,
Some(lookup_pos) => {
let lookup_pos = self
.commit_id_byte_prefix_to_lookup_pos(&min_bytes_prefix)
.unwrap_or(self.num_local_commits);
let mut first_match = None;
for i in lookup_pos..self.num_local_commits {
let entry = self.lookup_entry(i);
@ -1356,8 +1356,6 @@ impl IndexSegment for ReadonlyIndex {
Some(id) => PrefixResolution::SingleMatch(id),
}
}
}
}
fn segment_generation_number(&self, local_pos: u32) -> u32 {
self.graph_entry(local_pos).generation_number()
@ -1440,16 +1438,11 @@ impl IndexSegment for MutableIndex {
fn segment_resolve_prefix(&self, prefix: &HexPrefix) -> PrefixResolution<CommitId> {
let (bytes_prefix, min_bytes_prefix) = prefix.bytes_prefixes();
let mut potential_range = self
let potential_range = self
.lookup
.range((Bound::Included(&min_bytes_prefix), Bound::Unbounded));
let mut first_match = None;
loop {
match potential_range.next() {
None => {
break;
}
Some((id, _pos)) => {
for (id, _pos) in potential_range {
if !id.as_bytes().starts_with(bytes_prefix.as_bytes()) {
break;
}
@ -1460,8 +1453,6 @@ impl IndexSegment for MutableIndex {
first_match = Some(id)
}
}
}
}
match first_match {
None => PrefixResolution::NoMatch,
Some(id) => PrefixResolution::SingleMatch(id.clone()),